DIY_GIT_in_Python/ugit/data.py

from pathlib import Path

import hashlib

GIT_DIR = ".ugit"


def init():
    Path.mkdir(GIT_DIR)
    Path.mkdir(f"{GIT_DIR}/objects")


def update_ref(ref, oid):
    ref_path = f"{GIT_DIR}/{ref}"
    Path.mkdir(ref_path, exist_ok=True)
    with open(ref_path, "w") as f:
        f.write(oid)


def get_ref(ref):
    ref_path = f"{GIT_DIR}/{ref}"
    if Path.is_file(ref_path):
        with open(ref_path) as f:
            return f.read().strip()


def hash_object(data, type_="blob"):
    obj = type_.encode() + b"\x00" + data
    oid = hashlib.sha1(obj).hexdigest()
    with open(f"{GIT_DIR}/objects/{oid}", "wb") as out:
        out.write(obj)
    return oid


def get_object(oid, expected="blob"):
    with open(f"{GIT_DIR}/objects/{oid}", "rb") as f:
        obj = f.read()

    type_, _, content = obj.partition(b"\x00")
    type_ = type_.decode()

    if expected is not None:
        assert type_ == expected, f"Expected {expected}, got {type_}"
    return content