مختصر روابط محلي

import argparse
import hashlib
import json
from pathlib import Path
from urllib.parse import urlparse


def make_code(url: str, length: int = 7) -> str:
    return hashlib.sha256(url.encode("utf-8")).hexdigest()[:length]


def is_url(value: str) -> bool:
    parsed = urlparse(value)
    return parsed.scheme in {"http", "https"} and bool(parsed.netloc)


def load_db(path: Path) -> dict[str, str]:
    if not path.exists():
        return {}
    return json.loads(path.read_text(encoding="utf-8"))


def save_db(path: Path, data: dict[str, str]) -> None:
    path.write_text(json.dumps(data, indent=2), encoding="utf-8")


def main() -> None:
    parser = argparse.ArgumentParser(description="Create local short URL codes.")
    parser.add_argument("url")
    parser.add_argument("--db", type=Path, default=Path("short-links.json"))
    args = parser.parse_args()

    if not is_url(args.url):
        raise SystemExit("Please provide an http or https URL.")

    data = load_db(args.db)
    code = make_code(args.url)
    data[code] = args.url
    save_db(args.db, data)
    print(f"{code} -> {args.url}")


if __name__ == "__main__":
    main()