import json
import argparse
from os import getenv
from dotenv import load_dotenv
from glut import GLUTAcademic, Term
from server import run as server_run

load_dotenv()

def fetch_scores(glut_academic: GLUTAcademic, terms: list[str]):
    all_scores:list[GLUTAcademic.Score] = []
    for term in terms:
        year, term_str = term.split('_')
        term_str = term_str.lower()

        if term_str == 'sp':
            term = Term.SPRING
        elif term_str == 'au':
            term = Term.AUTUMN
        else:
            raise ValueError(f'Invalid term: {term}, must be one of 2023_sp, 2024_au')

        print(f'Fetching scores for {year} {term_str}...')
        all_scores.extend(
            glut_academic.get_scores(int(year), term)
        )

    return json.dumps([score.as_dict() for score in all_scores], 
                      ensure_ascii=False, indent=4), all_scores

def main():
    parser = argparse.ArgumentParser(
        description='GLUT Academic Score Calculator',
    )
    parser.add_argument('-u', '--username', type=str, default=getenv('USERNAME', default=None))
    parser.add_argument('-p', '--password', type=str, default=getenv('PASSWORD', default=None))
    parser.add_argument('-s', '--server', action='store_true',
                        help='Run the server instead of fetching scores')
    parser.add_argument('-t', '--terms', type=str, nargs='+',
                        help='Year and term combinations (e.g., 2023_sp 2024_au)')
    parser.add_argument('-o', '--output', type=str, default='scores.json',
                        help='Output file name (default: scores.json)')
    parser.add_argument('-q', '--quiet', action='store_true', default=True,
                        help='Suppress terminal output')
    parser.add_argument('--disable-file-output', action='store_true',
                        help='Disable file output')
    parser.add_argument('--port', type=int, default=int(getenv('PORT', default='8000')),
                        help='Port number for the server')

    args = parser.parse_args()

    if not args.username or not args.password:
        raise ValueError("USERNAME and PASSWORD must be set in the .env file")

    if args.server:
        server_run(port=args.port)
    else:
        glut_acdemic = GLUTAcademic(int(args.username), args.password)
        glut_acdemic.login()

        res, origin_li = fetch_scores(glut_acdemic, args.terms)

        if not args.disable_file_output:
            with open('scores.json', 'w', encoding='utf-8') as f:
                f.write(res)

        print(f"aggregate_score: {glut_acdemic.calculate_scores(origin_li)}")
        if not args.quiet:
            print(res)

if __name__ == '__main__':
    main()