from pygit2 import * from hurry.filesize import size, alternative from datetime import datetime import mimetypes from const import * from config import * mimetypes.add_type("text/gemini", ".gmi") mimetypes.add_type("text/gemini", ".gemini") def convert_filesize(bytes: int) -> str: return size(bytes, system=alternative) class GitGmiRepo: def __init__(self, name: str, path: str): self.name = name self.path = path try: self.repo = Repository(path) except GitError: raise FileNotFoundError(f"Error: no such repo: {name}") def generate_header(self): header = ( f"# {self.name}\n" f"=> {CGI_PATH} {GIT_GMI_SITE_TITLE}\n" f"=> {CGI_PATH}{self.name}/summary summary\n" f"=> {CGI_PATH}{self.name}/tree/{MAIN_BRANCH}/ tree\n" f"=> {CGI_PATH}{self.name}/log log\n\n" ) return header def view_summary(self) -> str: response = f"{STATUS_SUCCESS} {META_GEMINI}\n" + self.generate_header() # show 3 recent commits recent_commits = self.get_commit_log()[:3] for cmt in recent_commits: time = str(datetime.utcfromtimestamp(cmt["time"])) + " UTC" response += ( f"### {cmt['short_id']} - {cmt['author']} - {time}\n" f"{cmt['msg'].splitlines()[0]}\n\n" ) # TODO: link to commit view # find and display readme(.*) tree = self.get_tree(MAIN_BRANCH) trls = self.list_tree(tree) found_readme = False for item in trls: if ( item["type"] == "file" and item["name"].lower().split(".")[0] == ("readme") and not found_readme ): found_readme = True response += ( f"## {item['name']} | {convert_filesize(item['size'])}\n" f"{item['blob'].data.decode('utf-8')}" ) if not found_readme: response += "## No readme found." return response def get_commit_log(self) -> list: # returns the commit log in a human-readable way. repo = self.repo commits = list(repo.walk(repo[repo.head.target].id, GIT_SORT_TIME)) log = [ { "id": str(cmt.id), # hex SHA-1 hash "short_id": str(cmt.short_id), # short version of the above "author": cmt.author.name, # author's display name "time": cmt.commit_time, # unix timestamp "msg": cmt.message, # full commit message } for cmt in commits ] return log # reverse chronical order def view_log(self) -> str: response = f"{STATUS_SUCCESS} {META_GEMINI}\n" + self.generate_header() log = self.get_commit_log() for cmt in log: time = str(datetime.utcfromtimestamp(cmt["time"])) + " UTC" response += ( f"## {cmt['short_id']} - {cmt['author']} - {time}\n" f"=> tree/{cmt['short_id']}/ view tree\n" f"{cmt['msg']}\n\n" ) return response @classmethod def parse_recursive_tree(cls, tree: Tree) -> list: # recursively replace all Trees with a list of Blobs inside it, # bundled with the Tree's name as a tuple, # e.g. [('src', [blob0, blob1]), otherblob]. tree_list = list(tree) for idx, item in enumerate(tree_list): if isinstance(item, Tree): tree_list[idx] = (item.name, cls.parse_recursive_tree(tree_list[idx])) return tree_list def get_tree(self, commit_str: str) -> list: # returns a recursive list of Blob objects try: commit = self.repo.revparse_single(commit_str) # top level tree; may contain sub-trees return self.parse_recursive_tree(commit.tree) except ValueError: raise FileNotFoundError(f"Error: no such tree: {commit_str}") return None @staticmethod def list_tree(tree_list: list, location=[]) -> list: # tree_list is the output of parse_recursive_tree(); # location is which dir you are viewing, represented path-like # in a list, e.g. ['src', 'static', 'css'] => 'src/static/css', # which this method will cd into and display to the visitor. # when there is no such dir, raises FileNotFoundError. trls = tree_list for loc in location: found = False for item in trls: if isinstance(item, tuple) and item[0] == loc: trls = item[1] found = True break if not found: raise FileNotFoundError( f"Error: no such directory: {'/'.join(location)}" ) contents = [] for item in trls: if isinstance(item, tuple): # was originally a Tree; structure: ('dir_name', [list_of_blobs]) contents.append( { "type": "dir", "name": item[0], "size": len(item[1]), # number of objects in dir } ) elif isinstance(item, Blob): contents.append( { "type": "file", "name": item.name, "blob": item, "size": item.size, # size in bytes } ) return contents def view_tree(self, branch: str, location=[]) -> str: # actual Gemini response # consists of a header and a body response = f"{STATUS_SUCCESS} {META_GEMINI}\n" + self.generate_header() tree = self.get_tree(branch) contents = self.list_tree(tree, location) for item in contents: if item["type"] == "dir": response += ( f"=> {item['name']}/ {item['name']}/ | {item['size']} items\n" ) elif item["type"] == "file": response += f"=> {item['name']} {item['name']} | {convert_filesize(item['size'])}\n" return response def get_blob(self, commit_str: str, location=[]) -> Blob: # returns a specific Blob object # location: just like that of list_tree, but the last element # is the filename try: tree = self.get_tree(commit_str) trls = self.list_tree(tree, location[:-1]) for item in trls: if item["type"] == "file" and item["name"] == location[-1]: return item["blob"] raise FileNotFoundError(f"Error: no such file: {'/'.join(location)}") except FileNotFoundError: raise FileNotFoundError(f"Error: No such tree: {'/'.join(location[:-1])}") def view_blob(self, branch: str, location=[]) -> str: blob = self.get_blob(branch, location) response = ( f"{STATUS_SUCCESS} {META_GEMINI}\n" + self.generate_header() + f"## {self.name}/{'/'.join(location)} | {convert_filesize(blob.size)}\n\n" f"=> {blob.name}?raw view raw\n\n" f"```\n" ) response += blob.data.decode("utf-8") + "\n```" return response def view_raw_blob(self, branch: str, location=[]) -> str: blob = self.get_blob(branch, location) guessed_mimetype = mimetypes.guess_type(blob.name)[0] or "text/plain" response = f"{STATUS_SUCCESS} {guessed_mimetype}\n" response += blob.data.decode("utf-8") return response