revanced-static-api/app/generator.py
2023-10-29 22:56:38 +01:00

145 lines
3.7 KiB
Python

import json
from os.path import join
from app import api
from app.utils import get_repository_name, write_json, read_json, create_if_not_exists
from abc import abstractmethod
class Api:
_api: api.Api
def __init__(self, name: str, api: api.Api = api.GitHubApi()):
self.name = name
self._api = api
@abstractmethod
def generate(self, config, path):
"""
Generates the api based on the config to the path.
Args:
config (dict): The config for the api
path (str): The path where the api should be generated
"""
raise NotImplementedError
class ReleaseApi(Api):
def __init__(self, api):
super().__init__("release", api)
def generate(self, config, path):
path = join(path, "release")
repositories = config["repositories"]
for repository in repositories:
release = self._api.get_release(repository)
repository_name = get_repository_name(repository)
tag = release["tag"]
release_path = join(path, repository_name)
release_json = json.dumps(release)
create_if_not_exists(release_path)
write_json(release_json, join(
release_path, f"{tag}.json"), overwrite=False)
write_json(
release_json, join(release_path, "latest.json")
) # Overwrite the latest release
# At last join the current tag to an index file
index_path = join(path, f"{repository_name}.json")
index = read_json(index_path, [])
if tag not in index: # TODO: Check if there a better way to do this
index.append(tag) # Add the current tag to the index
write_json(index, index_path)
class ContributorApi(Api):
def __init__(self, api):
super().__init__("contributor", api)
def generate(self, config, path):
path = join(path, "contributor")
create_if_not_exists(path)
repositories = config["repositories"]
for repository in repositories:
repository_name = get_repository_name(repository)
contributors = self._api.get_contributor(repository)
contributors_path = join(path, f"{repository_name}.json")
write_json(contributors, contributors_path)
class SocialApi(Api):
def __init__(self, api):
super().__init__("social", api)
def generate(self, config, path):
new_social = config["socials"]
social_path = join(path, f"social.json")
write_json(new_social, social_path)
class TeamApi(Api):
def __init__(self, api):
super().__init__("team", api)
def generate(self, config, path):
organization = config["organization"]
team = self._api.get_members(organization)
team_path = join(path, f"team.json")
write_json(team, team_path)
class DonationApi(Api):
def __init__(self, api):
super().__init__("donation", api)
def generate(self, config, path):
donation = config["links"]
donation_path = join(path, f"donation.json")
write_json(donation, donation_path)
class ApiProvider:
_apis: list[Api]
def __init__(self, apis: list[Api]):
self._apis = apis
def get(self, name: str) -> Api | None:
for api in self._apis:
if api.name == name:
return api
return None
class DefaultApiProvider(ApiProvider):
def __init__(self):
self._api = api.GitHubApi() # Use GitHub as default api
super().__init__([
ReleaseApi(self._api),
ContributorApi(self._api),
SocialApi(self._api),
TeamApi(self._api),
DonationApi(self._api)
])