45 lines
974 B
Python
45 lines
974 B
Python
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
|