jetsung / 【脚本】2FA 二维码生成 Markdown

#!/usr/bin/env python3 #============================================================ # File: totp2md.py # Description: 将 TOTP 二维码转换为 Markdown 表格 # URL: https://s.fx4.cn/ # ORIGIN: https://gist.asfd.cn/jetsung/totp2md/raw/HEAD/totp2md.py # Author: Jetsung Chan <[email protected]> # Version: 0.1.0 # CreatedAt: 2025-08-18 # UpdatedAt: 2025-08-18 #============================================================ import os import re import subprocess import urllib.parse from pathlib import Path import argparse def decode_qr_image(file_path): """使用 zbarimg 读取二维码内容""" try: result = subprocess.run( ['zbarimg', str(file_path)], stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True, check=True ) lines = result.stdout.strip().splitlines() if lines: qr_text = lines[0].strip() if qr_text.startswith("QR-Code:"): qr_text = qr_text[len("QR-Code:"):] return qr_text.strip() except subprocess.CalledProcessError: pass # zbarimg 无法识别 except FileNotFoundError: print("错误: 找不到 zbarimg，请确保已安装 ZBar 并加入 PATH。") exit(1) return None def parse_otpauth_uri(uri): """解析 otpauth://totp/ URI""" sitename = "" username = "" secret = "" issuer = "" try: from urllib.parse import urlparse, parse_qs parsed = urlparse(uri) query = parse_qs(parsed.query, keep_blank_values=True) secret = query.get('secret', [None])[0] or "" label = parsed.path.lstrip('/') # e.g., Google:[email protected] label = urllib.parse.unquote(label) issuer = query.get('issuer', [None])[0] or "" if issuer: issuer = urllib.parse.unquote(issuer) if ':' in label: parts = label.split(':', 1) sitename = parts[0].strip() username = parts[1].strip() else: username = label if not issuer: issuer = sitename or "" except Exception as e: print(f"解析 otpauth 失败: {e}") return sitename, username, secret, issuer def parse_ms_msa_uri(uri): """解析 ms-msa:// URI，提取 uaid 和 code""" username = "" secret = "" try: query_string = uri.split('//', 1)[1] # 使用 parse_qs 解析 query = urllib.parse.parse_qs(query_string, keep_blank_values=True) username = query.get('uaid', [None])[0] or "" secret = query.get('code', [None])[0] or "" # 返回值：sitename, username, secret, issuer return "", username, secret, "Microsoft" except Exception as e: print(f"解析 ms-msa 失败: {e}") return "", "", "", "Microsoft" def main(input_dir, output_file): file_dir = Path(input_dir) if not file_dir.exists(): print(f"目录不存在: {file_dir}") exit(1) # 获取所有 .svg 和 .png 文件 image_files = [] image_files.extend(file_dir.rglob("*.svg")) image_files.extend(file_dir.rglob("*.png")) # 按路径排序 image_files.sort(key=lambda x: str(x).lower()) # 写入 Markdown 表头 with open(output_file, 'w', encoding='utf-8') as f: f.write("| SiteName | Username | Secret | Text | Mark |\n") f.write("|:---|:---|:---|:---|:---|\n") for file_path in image_files: qr_text = decode_qr_image(file_path) if not qr_text: print(f"无法读取二维码: {file_path}") continue # 转义 Markdown 中的 | text_display = qr_text.replace('|', r'\|') sitename = "" username = "" secret = "" issuer = "" if qr_text.startswith("otpauth://totp/"): sitename, username, secret, issuer = parse_otpauth_uri(qr_text) if not issuer: issuer = sitename or file_path.stem elif qr_text.startswith("ms-msa://"): sitename, username, secret, issuer = parse_ms_msa_uri(qr_text) else: # 其他类型 sitename = file_path.stem username = "" secret = "" issuer = file_path.stem # 构造 SiteName 列：[issuer](相对路径) relative_path = file_path.as_posix() issuer_str = f"[{issuer}]({relative_path})" # 输出调试信息 print(f"filename: {file_path}") print(f"sitename: {issuer}") print(f"username: {username}") print(f"secret: {secret}") print(f"text: {qr_text}") print() # 写入表格行 f.write(f"| {issuer_str} | {username} | {secret} | `{text_display}` | |\n") print(f"✅ Markdown 表格已生成: {output_file}") if __name__ == "__main__": parser = argparse.ArgumentParser(description="从 TOTP 二维码图片生成 Markdown 表格") parser.add_argument( '-i', '--input', default="TOTP", help="二维码图片目录 (默认: TOTP)" ) parser.add_argument( '-o', '--output', default="qr_codes.md", help="输出 Markdown 文件路径 (默认: qr_codes.md)" ) args = parser.parse_args() main(args.input, args.output) ### # # 依赖 zbarimg # Windows:https://zbar.sourceforge.net/ # Linux:https://github.com/mchehab/zbar # ###

#!/usr/bin/env bash #============================================================ # File: totp2md.sh # Description: 将 TOTP 二维码转换为 Markdown 表格 # URL: https://s.fx4.cn/ # ORIGIN: https://gist.asfd.cn/jetsung/totp2md/raw/HEAD/totp2md.sh # Author: Jetsung Chan <[email protected]> # Version: 0.1.0 # CreatedAt: 2025-08-18 # UpdatedAt: 2025-08-18 #============================================================ if [[ -n "${DEBUG:-}" ]]; then set -eux else set -euo pipefail fi TOTP_DIR="${1:-TOTP}" OUTPUT_FILE="${2:-qr_codes.md}" # Initialize Markdown table header echo "| SiteName | Username | Secret | Text | Mark |" > "$OUTPUT_FILE" echo "|:---|:---|:---|:---|:---|" >> "$OUTPUT_FILE" # Find all SVG and PNG files in the specified directory find "$TOTP_DIR" -type f \( -name "*.svg" -o -name "*.png" \) | while read -r file; do qr_text=$(zbarimg "$file" | head -n 1 2>/dev/null) qr_text=${qr_text/QR-Code:/} # qr_text=$(qrtool decode "$file" 2>/dev/null) # qr_text=$(totp-qr --uri "$file" 2>/dev/null) filename="$file" sitename="" username="" secret="" text="$qr_text" issuer="" # # Handle otpauth URLs if [[ "$qr_text" =~ ^otpauth://totp/ ]]; then label=$(echo "$qr_text" | grep -oP '(?<=totp/).*?(?=\?)') label=$(printf '%b' "${label//%/\\x}") issuer=$(echo "$qr_text" | grep -oP 'issuer=\K[^&]*' | head -1) issuer=$(printf '%b' "${issuer//%/\\x}") if [ -n "$label" ]; then if [[ "$label" =~ ^([^:]+):(.+)$ ]]; then sitename="${BASH_REMATCH[1]}" username="${BASH_REMATCH[2]}" else username="$label" fi fi if [ -z "$issuer" ]; then issuer="$sitename" fi secret=$(echo "$qr_text" | grep -oP 'secret=\K[^&]*' | head -1) elif [[ "$qr_text" =~ ^ms-msa:// ]]; then username=$(echo "$qr_text" | grep -oP 'uaid=\K[^&]*' | head -1) secret=$(echo "$qr_text" | grep -oP 'code=\K[^&]*' | head -1) else sitename="$filename" username="" secret="" fi text=${text//|/\\|} if [ -z "$issuer" ]; then issuer=$(basename "$filename") fi issuer_str="[$issuer]($filename)" echo "filename: $filename" echo "sitename: $issuer" echo "username: $username" echo "secret: $secret" echo "text: $text" echo # shellcheck disable=SC2006 echo "| $issuer_str | $username | $secret | `$text` | |" >> "$OUTPUT_FILE" done echo "Markdown table generated in $OUTPUT_FILE" ### # # 参数1: TOTP 目录 # 参数2: 输出文件 # # 示例: # ./totp2md.sh TOTP qr_codes.md # ###