moonq/q-tools
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

letsee about this update

Browse Source
This commit is contained in:
q
2025-03-11 20:12:14 +02:00
parent 083fb06416
commit 779f22f60b
1 changed files with 72 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files

131
shell/qaskpass
View File

@@ -5,24 +5,12 @@ import hashlib
import os import os
import random import random
import re import re
import string
import sys import sys
import termios import termios
import time
import tty import tty
__version__ = "20250310.a" __version__ = "20250311.a"
class getch:
def get(self):
fd = sys.stdin.fileno()
old_settings = termios.tcgetattr(fd)
try:
tty.setraw(sys.stdin.fileno())
ch = sys.stdin.read(1)
finally:
termios.tcsetattr(fd, termios.TCSADRAIN, old_settings)
return ch
class bc: class bc:
@@ -41,6 +29,8 @@ class bc:
C = "\033[96m" C = "\033[96m"
bold = "\033[1m" bold = "\033[1m"
z = "\033[0m" z = "\033[0m"
noblink = "\033[?25l"
blink = "\033[?25h"
def disable(self): def disable(self):
for x in "rgbcmyzkRGBCMY": for x in "rgbcmyzkRGBCMY":
@@ -56,8 +46,7 @@ class QAsk:
self.c.disable() self.c.disable()
self.user_input = "" self.user_input = ""
self.eq = "▁▂▃▄▅▆▇█▇▆▅▄▃▂" self.eq = "▁▂▃▄▅▆▇█▇▆▅▄▃▂"
self.limits = ((self.c.r, 2), (self.c.y, 4), (self.c.c, 8), (self.c.g, 14)) self.limits = ((self.c.r, 2), (self.c.y, 4), (self.c.B, 7), (self.c.c, 9), (self.c.g, 12), (self.c.G, 15))
self.ch = getch()
def get_opts(self): def get_opts(self):
parser = argparse.ArgumentParser( parser = argparse.ArgumentParser(
@@ -100,78 +89,102 @@ class QAsk:
return score return score
def animchar(self, pos): def animchar(self, pos):
colorpos = len(self.user_input) % self.opts.w """Returns one character, based on pwscore or pw hash"""
i = len(self.user_input) - pos i = len(self.user_input) - pos
if i < 0 and self.opts.measure: if self.opts.measure:
return self.c.r + " ▁ "[random.randint(0, 2)] if i < 0:
if pos == colorpos: return self.c.r + " ▁ "[random.randint(0, 2)]
clr = self.c.C rheight = random.randint(-1, 1)
else: clr = self.limits[0][0]
clr = self.c.r
score = self.pwscore() score = self.pwscore()
for limit in self.limits: for limit in self.limits:
if score > limit[1] - 1 + 2 * random.random(): if score > limit[1] - 1 + 2 * random.random():
clr = limit[0] clr = limit[0]
else: else:
break break
if self.opts.measure: return clr + self.eq[int(rheight + (i / self.opts.w)) % 14]
rheight = random.randint(-1, 1)
else: else: # no measure
rheight = random.randint(-4, 4) if len(self.user_input) == 0:
if len(self.user_input) == 0: rheight = random.randint(0, 1)
rheight = 0 height = 0
return clr + self.eq[int(rheight + (i / self.opts.w)) % 14] clr = self.limits[0][0]
else:
rheight = random.randint(-1, 1)
height = int(self.sha256[i * 2 % len(self.sha256)], 16) / 2
clr = self.limits[int(self.sha256[(1 + i * 2) % len(self.sha256)], 16) % len(self.limits)][0]
return clr + self.eq[int(rheight + height) % 14]
def pquit(self, s="", e=0): def pquit(self, s="", e=0):
print(self.c.blink, file=sys.stderr, end="")
print(self.c.z, file=sys.stderr) print(self.c.z, file=sys.stderr)
print(s, end="", file=sys.stdout if e == 0 else sys.stderr) print(s, end="", file=sys.stdout if e == 0 else sys.stderr)
sys.exit(e) sys.exit(e)
def ask(self): def check_sha256(self):
# 3= ctrl-c, 13=enter
# 127 = backspace
dot = "•" self.sha256 = hashlib.sha256(self.user_input.encode("utf-8")).hexdigest()
display = f"{dot*4} {dot*4}" if self.opts.expect_sha256:
if self.sha256 == self.opts.expect_sha256:
self.dot_color = self.c.G
self.dot = "♥"
self.enter_exitcode = 0
else:
self.enter_exitcode = 10
def get_chr(self):
fd = sys.stdin.fileno()
old_settings = termios.tcgetattr(fd)
try:
tty.setraw(sys.stdin.fileno())
ch = sys.stdin.read(1)
finally:
termios.tcsetattr(fd, termios.TCSADRAIN, old_settings)
return ch
def ask(self):
print(self.c.noblink, file=sys.stderr, end="")
self.dot = "•"
if self.opts.title: if self.opts.title:
print(f"{self.c.Y}{self.opts.title}{self.c.z}", file=sys.stderr) print(f"{self.c.Y}{self.opts.title}{self.c.z}", file=sys.stderr)
enter_exitcode = 0
while True: while True:
try: try:
self.enter_exitcode = 0
if self.opts.w > 0: if self.opts.w > 0:
dot_color = self.c.m self.dot_color = self.c.m
dot = "•" self.dot = "•"
if self.opts.expect_sha256: self.check_sha256()
enter_exitcode = 10 display = "".join(
if hashlib.sha256(self.user_input.encode("utf-8")).hexdigest() == self.opts.expect_sha256: [
dot_color = self.c.G f"\r{self.c.z}{self.dot_color}{self.dot*4} ",
dot = "♥" *[self.animchar(i) for i in range(self.opts.w)],
enter_exitcode = 0 f" {self.dot_color}{self.dot*4}{self.c.z}\r{self.c.k}",
display = dot_color + f"{dot*4} " ]
for i in range(self.opts.w): )
display += self.animchar(i) print(display, file=sys.stderr, end="")
display += dot_color + f" {dot*4}" + self.c.z
print("\r" + self.c.z + display + "\r" + self.c.k, file=sys.stderr, end="")
sys.stderr.flush() sys.stderr.flush()
key = self.ch.get() key = self.get_chr()
# ~ print(f'-{key}{key}{key}{key}{key}///{ord(key)}///{len(key)}-',file=sys.stderr)
if ord(key) == 3: # ctrl-c if ord(key) == 3: # ctrl-c
self.pquit(e=1) self.pquit(e=1)
if ord(key) == 13: # enter elif ord(key) == 13: # enter
self.pquit(self.user_input, e=enter_exitcode) self.pquit(self.user_input, e=self.enter_exitcode)
if ord(key) == 27: # esc (also starts control characters elif ord(key) == 27: # esc (also starts control characters
key = self.ch.get() key = self.get_chr()
if ord(key) == 27: if ord(key) == 27:
self.pquit(e=1) self.pquit(e=1)
continue continue
if ord(key) == 127: # backspace elif ord(key) == 127: # backspace
self.user_input = self.user_input[0:-1] self.user_input = self.user_input[0:-1]
else: else:
self.user_input += key if key in string.printable:
# ~ print(f'-{key}-',file=sys.stderr) self.user_input += key
# ~ print(f'-{ord(key)}-',file=sys.stderr)
except Exception as e: except Exception as e:
raise (e)
self.pquit(s=str(e), e=1) self.pquit(s=str(e), e=1)