import asyncio
import datetime
from typing import AsyncGenerator
import typing

import strawberry
from selfprivacy_api.graphql import IsAuthenticated

from selfprivacy_api.jobs import Job, Jobs

@strawberry.type
class ApiJob:
    name: str
    description: str
    status: str
    status_text: typing.Optional[str]
    progress: typing.Optional[int]
    created_at: datetime.datetime
    updated_at: datetime.datetime
    finished_at: typing.Optional[datetime.datetime]
    error: typing.Optional[str]
    result: typing.Optional[str]

@strawberry.type
class JobSubscription:
    @strawberry.subscription(permission_classes=[IsAuthenticated])
    async def job_subscription(self) -> AsyncGenerator[typing.List[ApiJob], None]:
        is_updated = True
        def callback(jobs: typing.List[Job]):
            nonlocal is_updated
            is_updated = True
        Jobs().add_observer(callback)
        try:
            while True:
                if is_updated:
                    is_updated = False
                    yield [ ApiJob(
                        name=job.name,
                        description=job.description,
                        status=job.status.name,
                        status_text=job.status_text,
                        progress=job.progress,
                        created_at=job.created_at,
                        updated_at=job.updated_at,
                        finished_at=job.finished_at,
                        error=job.error,
                        result=job.result,
                    ) for job in Jobs().get_jobs() ]
        except GeneratorExit:
            Jobs().remove_observer(callback)
            return