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()
    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)
        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() ]