#!/usr/bin/env python3

from typing import Union
import hashlib
import sqlite3

from fastapi import Depends, FastAPI
from fastapi.middleware.cors import CORSMiddleware

import uvicorn

from aci.db import get_db
from aci.models import CodeSubmission
from aci import utils

app = FastAPI()

app.add_middleware(
    CORSMiddleware,
    allow_origin_regex=r"moz-extension://.*",
    allow_methods=["GET"],
    allow_headers="*",
)

@app.post('/api/codes')
def submit_code(date: str, time: str, activity: str, space: str, submission: CodeSubmission, db: sqlite3.Cursor=Depends(get_db)):
    # TODO: validate parameters meet expected form.
    db.execute('INSERT INTO code_results (date, time, space, activity, code, result) VALUES (?,?,?,?,?,?)', (date, time, space, activity, submission.code, submission.success))
    return {'ok': True}

@app.get('/api/codes')
def current_codes(db: sqlite3.Cursor=Depends(get_db)):
    result = db.execute('SELECT date, time, space, activity, code, result FROM code_results WHERE date = ?', (utils.today(),))
    codes = result.fetchall()
    code_scores = {}
    code_info = {}
    for date, time, space, activity, code, result in codes:
        k = str((date, time, space, activity))
        code_info[k] = (date, time, space, activity)
        if k not in code_scores:
            code_scores[k] = {}
        if code not in code_scores[k]:
            code_scores[k][code] = 0
        code_scores[k][code] += 1 if result else -1
    ret = []
    for k, v in code_scores.items():
        date, time, space, activity = code_info[k]
        codes = list(map(lambda c: {'code': c[0], 'score': c[1]}, v.items()))
        codes.sort(key=lambda c: -c['score'])
        ret.append({
            'date': date,
            'time': time,
            'space': space,
            'activity': activity,
            'codes': codes,
        })
    return {
        'activities': ret * 30,
    }

if __name__ == '__main__':
    uvicorn.run(app)