updates
This commit is contained in:
Iliyan Angelov
Binary file not shown.
Binary file not shown.
Binary file not shown.
@@ -0,0 +1,227 @@
|
||||
import random
|
||||
import time
|
||||
from rich.text import Text
|
||||
from rich.style import Style
|
||||
|
||||
# -----------------------------
|
||||
# Celebration Effects
|
||||
# -----------------------------
|
||||
|
||||
|
||||
def show_confetti(console):
|
||||
# Characters to use as confetti
|
||||
chars = ["*", "o", "+", "~"]
|
||||
width = console.size.width
|
||||
height = console.size.height
|
||||
|
||||
frames = 10
|
||||
for _ in range(frames):
|
||||
console.clear()
|
||||
for __ in range(random.randint(50, 100)): # number of confetti pieces
|
||||
x = random.randint(0, max(0, width - 1))
|
||||
y = random.randint(0, max(0, height - 2))
|
||||
char = random.choice(chars)
|
||||
color = random.choice(["red", "green", "yellow", "blue", "magenta", "cyan"])
|
||||
console.print(
|
||||
Text(char, style=Style(color=color)),
|
||||
end="",
|
||||
style=color,
|
||||
justify="left",
|
||||
overflow="ignore",
|
||||
no_wrap=True,
|
||||
soft_wrap=False,
|
||||
)
|
||||
console.file.write(f"\x1b[{y};{x}H") # Move cursor to position
|
||||
console.file.flush()
|
||||
time.sleep(0.3)
|
||||
console.clear()
|
||||
console.print(
|
||||
"The confetti has settled! Congrats on a clean scan!", style="bold green"
|
||||
)
|
||||
|
||||
|
||||
def show_trophy(console):
|
||||
"""Displays a celebratory trophy with sparkles."""
|
||||
trophy = (
|
||||
r"""
|
||||
___________
|
||||
'._==_==_=_.'
|
||||
.-\: /-.
|
||||
| (|:. |) |
|
||||
'-|:. |-'
|
||||
\::. /
|
||||
'::. .'
|
||||
) (
|
||||
_.' '._
|
||||
`"""
|
||||
""""`
|
||||
"""
|
||||
)
|
||||
for _ in range(5): # Trophy animation
|
||||
console.clear()
|
||||
sparkles = random.choice(
|
||||
[":sparkles:", ":glowing_star:", ":dizzy:", ":party_popper:"]
|
||||
)
|
||||
console.print(trophy, style="bold yellow")
|
||||
console.print(
|
||||
f"{sparkles} Scan Complete! No vulnerabilities found! {sparkles}",
|
||||
style="bold green",
|
||||
justify="center",
|
||||
)
|
||||
time.sleep(0.5)
|
||||
console.print("Your code is SAFE and SOUND! :trophy:", style="bold yellow")
|
||||
|
||||
|
||||
def show_balloons(console):
|
||||
"""Displays celebratory balloons popping."""
|
||||
balloons = [":balloon:", ":party_popper:", ":sparkles:", ":collision:"]
|
||||
width = console.size.width
|
||||
|
||||
for _ in range(10): # Number of balloons
|
||||
console.clear()
|
||||
for __ in range(random.randint(5, 10)): # Balloons per frame
|
||||
x = random.randint(0, width - 1)
|
||||
balloon = random.choice(balloons)
|
||||
console.print(
|
||||
Text(balloon, style=Style(color="yellow")), end="", overflow="ignore"
|
||||
)
|
||||
console.file.write(f"\x1b[{random.randint(1, 10)};{x}H")
|
||||
console.file.flush()
|
||||
time.sleep(0.5)
|
||||
console.print(
|
||||
":balloon: POP! :party_popper: No vulnerabilities detected!", style="bold green"
|
||||
)
|
||||
|
||||
|
||||
def show_victory_parade(console):
|
||||
"""Displays a victory parade of emojis."""
|
||||
parade = [
|
||||
":party_popper:",
|
||||
":confetti_ball:",
|
||||
":trophy:",
|
||||
":partying_face:",
|
||||
":sparkles:",
|
||||
":laptop_computer:",
|
||||
":locked:",
|
||||
":white_heavy_check_mark:",
|
||||
]
|
||||
width = console.size.width
|
||||
|
||||
for _ in range(20): # Duration of parade
|
||||
console.clear()
|
||||
line = " ".join(random.choices(parade, k=width // 2))
|
||||
console.print(line, style="bold green", justify="center")
|
||||
time.sleep(0.2)
|
||||
|
||||
console.print(
|
||||
"The parade is over. Your code is safe! :trophy:", style="bold yellow"
|
||||
)
|
||||
|
||||
|
||||
def show_confetti_rain(console):
|
||||
"""Displays a colorful confetti rain effect."""
|
||||
colors = ["red", "green", "yellow", "blue", "magenta", "cyan", "white"]
|
||||
width = console.size.width
|
||||
|
||||
for _ in range(10): # Number of confetti frames
|
||||
console.clear()
|
||||
for __ in range(100): # Confetti pieces per frame
|
||||
x = random.randint(0, width - 1)
|
||||
char = random.choice(["*", "+", "~", ":sparkles:", "o"])
|
||||
color = random.choice(colors)
|
||||
console.print(
|
||||
Text(char, style=Style(color=color)), end="", overflow="ignore"
|
||||
)
|
||||
console.file.write(f"\x1b[{random.randint(1, 10)};{x}H")
|
||||
console.file.flush()
|
||||
time.sleep(0.3)
|
||||
|
||||
console.print(
|
||||
":party_popper: Confetti celebration complete! You're vulnerability-free! :party_popper:",
|
||||
style="bold cyan",
|
||||
)
|
||||
|
||||
|
||||
def show_fireworks_display(console):
|
||||
"""Displays a celebratory fireworks animation."""
|
||||
fireworks = [
|
||||
":collision:",
|
||||
":sparkles:",
|
||||
":glowing_star:",
|
||||
":fireworks:",
|
||||
":sparkler:",
|
||||
]
|
||||
width = console.size.width
|
||||
|
||||
for _ in range(15): # Number of fireworks
|
||||
x = random.randint(5, width - 5)
|
||||
y = random.randint(2, 8)
|
||||
firework = random.choice(fireworks)
|
||||
color = random.choice(["red", "yellow", "green", "blue", "magenta"])
|
||||
console.print(
|
||||
Text(firework, style=Style(color=color)), end="", overflow="ignore"
|
||||
)
|
||||
console.file.write(f"\x1b[{y};{x}H") # Position fireworks
|
||||
console.file.flush()
|
||||
time.sleep(0.3)
|
||||
|
||||
console.print(
|
||||
":fireworks: Fireworks display finished! Code is secure! :fireworks:",
|
||||
style="bold magenta",
|
||||
)
|
||||
|
||||
|
||||
def show_star_trail(console):
|
||||
"""Displays a shooting star trail effect."""
|
||||
stars = [":white_medium_star:", ":glowing_star:", ":sparkles:", ":dizzy:"]
|
||||
width = console.size.width
|
||||
|
||||
for _ in range(10): # Number of shooting stars
|
||||
console.clear()
|
||||
start_x = random.randint(0, width // 2)
|
||||
trail = "".join(random.choices(stars, k=10))
|
||||
console.print(f"{' ' * start_x}{trail}", style="bold yellow", justify="left")
|
||||
time.sleep(0.3)
|
||||
|
||||
console.print(
|
||||
":sparkles: Your code shines bright with no vulnerabilities! :sparkles:",
|
||||
style="bold cyan",
|
||||
)
|
||||
|
||||
|
||||
def show_celebration_wave(console):
|
||||
"""Displays a celebratory wave effect with emojis."""
|
||||
emojis = [
|
||||
":party_popper:",
|
||||
":confetti_ball:",
|
||||
":sparkles:",
|
||||
":partying_face:",
|
||||
":balloon:",
|
||||
]
|
||||
width = console.size.width
|
||||
wave = [random.choice(emojis) for _ in range(width)]
|
||||
|
||||
for _ in range(10): # Number of waves
|
||||
console.clear()
|
||||
line = "".join(wave)
|
||||
console.print(line, style="bold yellow", justify="center")
|
||||
wave.insert(0, wave.pop()) # Shift wave
|
||||
time.sleep(0.3)
|
||||
|
||||
console.print(
|
||||
":water_wave: Celebration wave ends! Your scan is clean! :glowing_star:",
|
||||
style="bold green",
|
||||
)
|
||||
|
||||
|
||||
# List of all celebratory effects
|
||||
CELEBRATION_EFFECTS = [
|
||||
show_confetti,
|
||||
show_trophy,
|
||||
show_balloons,
|
||||
show_victory_parade,
|
||||
show_confetti_rain,
|
||||
show_fireworks_display,
|
||||
show_star_trail,
|
||||
show_celebration_wave,
|
||||
]
|
||||
@@ -0,0 +1,284 @@
|
||||
import random
|
||||
import time
|
||||
import os
|
||||
from safety.scan.fun_mode.celebration_effects import CELEBRATION_EFFECTS
|
||||
|
||||
# -----------------------------
|
||||
# Data: ASCII Arts, Fortunes, EMOJIS, etc.
|
||||
# -----------------------------
|
||||
|
||||
ASCII_ARTS = {
|
||||
"ascii": [
|
||||
# Hedgehog guarding a shield
|
||||
r"""
|
||||
/\ /\
|
||||
{ `---' }
|
||||
{ O O }
|
||||
~~> V <~~
|
||||
\ \|/ /
|
||||
`-----'__
|
||||
/ \ `^\_
|
||||
{ }\ |\_\_ *Safely Protected from vulnerabilities!*
|
||||
| \_/ |/ / \_\_
|
||||
\__/ /(_E \__/
|
||||
( /
|
||||
MM
|
||||
""",
|
||||
# Cat with a shield
|
||||
r"""
|
||||
/\_/\
|
||||
=( °w° )= *Purrr... no vulnerabilities dare to cross!*
|
||||
( * )
|
||||
---( )---
|
||||
/ \
|
||||
/ ^ \
|
||||
( ( ) )
|
||||
\_) |_(_/
|
||||
__||___||__
|
||||
| |
|
||||
| SAFE |
|
||||
| & |
|
||||
| SECURE! |
|
||||
|___________|
|
||||
/ \
|
||||
| |
|
||||
|
||||
""",
|
||||
# Bunny with a shield
|
||||
r"""
|
||||
(\_/)
|
||||
( •_•) *Hop-hop! SafetyCLI ready, no vulns here!*
|
||||
( >:carrot:< )
|
||||
/ \
|
||||
/ | | \
|
||||
/ | | \
|
||||
/ |___| \
|
||||
( )
|
||||
\____|||_____/
|
||||
||||
|
||||
__||__
|
||||
| |
|
||||
| SAFE |
|
||||
| FROM |
|
||||
| BUGS |
|
||||
|______|
|
||||
|
||||
""",
|
||||
# Dog behind a shield
|
||||
r"""
|
||||
/ \__
|
||||
( o\____
|
||||
/ O
|
||||
/ (_______/ *Woof! Our shield is strong, no vulns inside!*
|
||||
/_____/
|
||||
|
||||
|
||||
""",
|
||||
]
|
||||
}
|
||||
|
||||
FORTUNES = [
|
||||
"Your dependencies are safer than a password-manager's vault.",
|
||||
"Your code sparkles with zero known vulnerabilities!",
|
||||
"All vulnerabilities fear your security prowess!",
|
||||
"Your build is as solid as a rock!",
|
||||
"Your code is a fortress; no bug can breach it.",
|
||||
"In the realm of code, you are the vigilant guardian.",
|
||||
"Each line you write fortifies the castle of your code.",
|
||||
"Your code is a well-oiled machine, impervious to rust.",
|
||||
"Your code is a symphony of security and efficiency.",
|
||||
"Your code is a beacon of safety in the digital ocean.",
|
||||
"Your code is a masterpiece, untouched by the hands of vulnerabilities.",
|
||||
"Your code stands tall, a citadel against cyber threats.",
|
||||
"Your code is a tapestry woven with threads of safety.",
|
||||
"Your code is a lighthouse, guiding ships away from the rocks of vulnerabilities.",
|
||||
"Your code is a garden where no weeds of bugs can grow.",
|
||||
"In the realm of software, your security measures are legendary.",
|
||||
]
|
||||
|
||||
|
||||
EMOJIS = [
|
||||
":dog_face:",
|
||||
":dog2:",
|
||||
":guide_dog:",
|
||||
":service_dog:",
|
||||
":poodle:",
|
||||
":wolf:",
|
||||
":fox_face:",
|
||||
":cat_face:",
|
||||
":cat2:",
|
||||
":cat2:",
|
||||
":lion_face:",
|
||||
":tiger_face:",
|
||||
":tiger2:",
|
||||
":leopard:",
|
||||
":horse_face:",
|
||||
":deer:",
|
||||
":deer:",
|
||||
":racehorse:",
|
||||
":unicorn_face:",
|
||||
":zebra:",
|
||||
":deer:",
|
||||
":bison:",
|
||||
":cow_face:",
|
||||
":ox:",
|
||||
":water_buffalo:",
|
||||
":cow2:",
|
||||
":ram:",
|
||||
":sheep:",
|
||||
":goat:",
|
||||
":dromedary_camel:",
|
||||
":two-hump_camel:",
|
||||
":llama:",
|
||||
":giraffe:",
|
||||
":elephant:",
|
||||
":mammoth:",
|
||||
":rhinoceros:",
|
||||
":hippopotamus:",
|
||||
":mouse_face:",
|
||||
":mouse2:",
|
||||
":rat:",
|
||||
":hamster:",
|
||||
":rabbit_face:",
|
||||
":rabbit2:",
|
||||
":chipmunk:",
|
||||
":beaver:",
|
||||
":hedgehog:",
|
||||
":bat:",
|
||||
":bear:",
|
||||
":polar_bear:",
|
||||
":koala:",
|
||||
":panda_face:",
|
||||
":otter:",
|
||||
":kangaroo:",
|
||||
":badger:",
|
||||
":turkey:",
|
||||
":chicken:",
|
||||
":rooster:",
|
||||
":baby_chick:",
|
||||
":hatched_chick:",
|
||||
":bird:",
|
||||
":penguin:",
|
||||
":dove:",
|
||||
":eagle:",
|
||||
":duck:",
|
||||
":swan:",
|
||||
":owl:",
|
||||
":dodo:",
|
||||
":flamingo:",
|
||||
":peacock:",
|
||||
":parrot:",
|
||||
":bird:",
|
||||
":goose:",
|
||||
":phoenix:",
|
||||
":frog:",
|
||||
":crocodile:",
|
||||
":turtle:",
|
||||
":lizard:",
|
||||
":dragon:",
|
||||
":sauropod:",
|
||||
":t-rex:",
|
||||
":whale:",
|
||||
":whale2:",
|
||||
":flipper:",
|
||||
":seal:",
|
||||
":fish:",
|
||||
":tropical_fish:",
|
||||
":blowfish:",
|
||||
":shark:",
|
||||
":octopus:",
|
||||
":jellyfish:",
|
||||
":crab:",
|
||||
":lobster:",
|
||||
":squid:",
|
||||
":snail:",
|
||||
":butterfly:",
|
||||
":bug:",
|
||||
":bee:",
|
||||
]
|
||||
|
||||
# -----------------------------
|
||||
# Helper functions (Effects)
|
||||
# -----------------------------
|
||||
|
||||
|
||||
def show_race(console):
|
||||
# Pick two different EMOJIS at random
|
||||
emoji1, emoji2 = random.sample(EMOJIS, 2)
|
||||
finish_line = 50
|
||||
pos1 = 0
|
||||
pos2 = 0
|
||||
|
||||
console.print("Ready... Set... Go!", style="bold cyan")
|
||||
time.sleep(1)
|
||||
console.clear()
|
||||
|
||||
while True:
|
||||
# Move contestants forward by random increments
|
||||
pos1 += random.randint(1, 3)
|
||||
pos2 += random.randint(1, 3)
|
||||
|
||||
console.clear()
|
||||
console.print("[green]Finish line[/green]" + " " * (finish_line - 10) + "|")
|
||||
line1 = " " * pos1 + emoji1
|
||||
line2 = " " * pos2 + emoji2
|
||||
console.print(f"{emoji1} lane: {line1}")
|
||||
console.print(f"{emoji2} lane: {line2}")
|
||||
|
||||
time.sleep(0.3)
|
||||
|
||||
finished1 = pos1 >= finish_line
|
||||
finished2 = pos2 >= finish_line
|
||||
|
||||
if finished1 and finished2:
|
||||
console.print(
|
||||
"It's a tie! Both reached the finish line at the same time!",
|
||||
style="bold magenta",
|
||||
)
|
||||
break
|
||||
elif finished1:
|
||||
console.print(
|
||||
f"The {emoji1} wins! Slow and steady (or maybe fast?), it prevailed!",
|
||||
style="bold green",
|
||||
)
|
||||
break
|
||||
elif finished2:
|
||||
console.print(
|
||||
f"The {emoji2} wins! Speed and agility triumphed!", style="bold green"
|
||||
)
|
||||
break
|
||||
|
||||
time.sleep(2)
|
||||
console.clear()
|
||||
console.print("Hope you enjoyed the race! :party_popper:", style="bold cyan")
|
||||
|
||||
|
||||
# -----------------------------
|
||||
# Main Easter Egg Dispatcher
|
||||
# -----------------------------
|
||||
|
||||
|
||||
def run_easter_egg(console, exit_code: int) -> None:
|
||||
"""
|
||||
Runs an easter egg based on the SAFETY_FUN_MODE environment variable.
|
||||
This function can be easily removed or commented out.
|
||||
"""
|
||||
egg_mode = os.getenv("SAFETY_FUN_MODE", "").strip().lower()
|
||||
|
||||
allowed_modes = {"ascii", "fx", "race", "fortune"}
|
||||
|
||||
if exit_code == 0 and egg_mode in allowed_modes:
|
||||
if egg_mode == "ascii":
|
||||
art = random.choice(ASCII_ARTS["ascii"])
|
||||
console.print(art, style="green")
|
||||
|
||||
elif egg_mode == "fx":
|
||||
effect = random.choice(CELEBRATION_EFFECTS)
|
||||
effect(console) # Run the randomly selected effect
|
||||
|
||||
elif egg_mode == "race":
|
||||
show_race(console)
|
||||
|
||||
elif egg_mode == "fortune":
|
||||
fortune_message = random.choice(FORTUNES)
|
||||
console.print(f"\n[italic cyan]{fortune_message}[/italic cyan]\n")
|
||||
Reference in New Issue
Block a user