dotcode-decoder 1.0.0__py3-none-any.whl

dotcode/codes.py

@@ -0,0 +1,247 @@
+"""
+DotCode encoding/decoding logic
+"""
+
+from .rs import DOT_TO_VAL, rs_correct, solve_ND_NC, mask_step, P
+
+def decode_codesets(vals):
+    msg = []
+    cs = "C"
+    i = 0
+    shift_to = None
+    shift_n = 0
+    upper_A = False
+    upper_B = False
+    
+    while i < len(vals):
+        v = vals[i]
+        
+        if v == 107:
+            break
+        
+        if v == 108:
+            msg.append("<FNC2>")
+            i += 1
+            continue
+        
+        if v == 109:
+            msg.append("<FNC3>")
+            i += 1
+            continue
+        
+        if v == 110:
+            upper_A = True
+            i += 1
+            continue
+        
+        if v == 111:
+            upper_B = True
+            i += 1
+            continue
+        
+        if v == 112:
+            return None
+        
+        if v > 112:
+            break
+        
+        eff = shift_to if shift_n > 0 and shift_to else cs
+        
+        if eff == "C" and v == 100:
+            break
+        
+        if eff in ("A", "B") and v >= 107 and msg:
+            break
+        
+        def tick():
+            nonlocal shift_n, shift_to
+            if shift_n > 0:
+                shift_n -= 1
+            if shift_n == 0:
+                shift_to = None
+        
+        def uc(ch):
+            nonlocal upper_A, upper_B
+            if upper_A or upper_B:
+                ch = chr((ord(ch) + 128) % 256)
+                upper_A = False
+                upper_B = False
+            return ch
+        
+        if eff == "C":
+            if v == 101:
+                cs = "A"
+                shift_to = None
+                shift_n = 0
+            elif v == 102:
+                shift_to = "B"
+                shift_n = 1
+            elif v == 103:
+                shift_to = "B"
+                shift_n = 2
+            elif v == 104:
+                shift_to = "B"
+                shift_n = 3
+            elif v == 105:
+                shift_to = "B"
+                shift_n = 4
+            elif v == 106:
+                cs = "B"
+                shift_to = None
+                shift_n = 0
+            elif 0 <= v <= 99:
+                msg.append(f"{v:02d}")
+                tick()
+        
+        elif eff == "A":
+            if v == 101:
+                shift_to = "B"
+                shift_n = 1
+            elif v == 102:
+                cs = "B"
+                shift_to = None
+                shift_n = 0
+            elif v == 103:
+                shift_to = "C"
+                shift_n = 2
+            elif v == 104:
+                shift_to = "C"
+                shift_n = 3
+            elif v == 105:
+                shift_to = "C"
+                shift_n = 4
+            elif v == 106:
+                cs = "C"
+                shift_to = None
+                shift_n = 0
+            elif 0 <= v <= 63:
+                msg.append(uc(chr(32 + v)))
+                tick()
+            elif 64 <= v <= 95:
+                msg.append(uc(chr(v - 64)))
+                tick()
+            else:
+                tick()
+        
+        elif eff == "B":
+            if v == 101:
+                shift_to = "A"
+                shift_n = 1
+            elif v == 102:
+                cs = "A"
+                shift_to = None
+                shift_n = 0
+            elif v == 103:
+                shift_to = "C"
+                shift_n = 2
+            elif v == 104:
+                shift_to = "C"
+                shift_n = 3
+            elif v == 105:
+                shift_to = "C"
+                shift_n = 4
+            elif v == 106:
+                cs = "C"
+                shift_to = None
+                shift_n = 0
+            elif 0 <= v <= 95:
+                msg.append(uc(chr(32 + v)))
+                tick()
+            elif v == 96:
+                msg.append("\r\n")
+                tick()
+            elif v == 97:
+                msg.append("\t")
+                tick()
+            elif v == 98:
+                msg.append(chr(28))
+                tick()
+            elif v == 99:
+                msg.append(chr(29))
+                tick()
+            elif v == 100:
+                msg.append(chr(30))
+                tick()
+            else:
+                tick()
+        
+        i += 1
+    
+    result = "".join(msg)
+    
+    if len(result) >= 12:
+        for pad in ["27", "28", "29", "00", "99", "14"]:
+            if result.endswith(pad):
+                result = result[:-2]
+                break
+    
+    return result if result else None
+
+def decode_bits(bitstr):
+    s = "".join(bitstr.split())
+    
+    if not all(c in "01" for c in s) or len(s) < 20:
+        return None
+    
+    ones = s.count("1") / len(s) * 100
+    if ones > 78 or ones < 8:
+        return None
+    
+    mask_id = (int(s[0]) << 1) | int(s[1])
+    cw_bits = [s[2+k*9:2+(k+1)*9] for k in range((len(s)-2)//9)]
+    vals = []
+    era = []
+    
+    for idx, b in enumerate(cw_bits):
+        v = None
+        if len(b) == 9 and b.count("1") == 5:
+            v = DOT_TO_VAL.get(b)
+        
+        if v is None:
+            vals.append(None)
+            era.append(idx)
+        else:
+            vals.append(v)
+    
+    NW = len(vals)
+    ND, NC = solve_ND_NC(NW)
+    
+    if ND is None:
+        return None
+    
+    if all(v is not None for v in vals[:ND]):
+        step = mask_step(mask_id)
+        dv = [(vals[i] - step * i) % P for i in range(ND)]
+        msg = decode_codesets(dv)
+        
+        if msg and len(msg) >= 4:
+            if not any(c in msg for c in ['/', '\\', '\x00', '\xff', ')', '(', '<']):
+                return msg
+    
+    res = rs_correct(mask_id, vals, ND, NC, era)
+    
+    if res is None:
+        return None
+    
+    corrected, _ = res
+    step = mask_step(mask_id)
+    dv = [(corrected[i] - step * i) % P for i in range(ND)]
+    msg = decode_codesets(dv)
+    
+    if msg and len(msg) >= 4:
+        if not any(c in msg for c in ['/', '\\', '\x00', '\xff', ')', '(', '<']):
+            return msg
+    
+    return None
+
+def is_valid_message(msg):
+    if msg is None or len(msg) < 4:
+        return False
+    
+    if msg.isdigit() and len(msg) > 20:
+        return False
+    
+    if any(c in msg for c in ['/', '\\', '\x00', '\xff', ')', '(', '<']):
+        return False
+    
+    return True