from github import Github, Auth
from ..config import GITHUB_TOKEN, REPO_NAME, REPO_OWNER

# Initialize GitHub client
github_client = Github(auth=Auth.Token(GITHUB_TOKEN))

def create_or_update_file(file_path, content, commit_message = 'Default Commit Message'):
    try:
        repo = github_client.get_repo(f'{REPO_OWNER}/{REPO_NAME}')
        
        try:
            # Check if the file already exists
            contents = repo.get_contents(file_path)
            # If it exists, update the file
            repo.update_file(contents.path, commit_message, content, contents.sha)
            return f"File '{file_path}' updated successfully."
        except Exception as e:
            # If the file does not exist, create a new file
            repo.create_file(file_path, commit_message, content)
            return f"File '{file_path}' created successfully."
    except Exception as e:
        return f"Error creating or updating file: {e}"

def delete_file(file_path, commit_message='Delete file'):
    try:
        repo = github_client.get_repo(f'{REPO_OWNER}/{REPO_NAME}')
        # Check if the file exists
        contents = repo.get_contents(file_path)
        # Delete the file
        repo.delete_file(contents.path, commit_message, contents.sha)
        return f"File '{file_path}' deleted successfully."
    except Exception as e:
        return f"Error deleting file: {e}"

def list_files(file_path):
    try:
        repo = github_client.get_repo(f'{REPO_OWNER}/{REPO_NAME}')
        contents = repo.get_contents(file_path)
        files = [content.path for content in contents if content.type == 'file']
        return files
    except Exception as e:
        return f"Error listing files: {e}"