U [MhG@sddlmZmZmZmZmZmZddlmZddl m Z m Z m Z m Z ddlmZmZmZddlZddlZddlZddlmZddlmZddlZeZeeZGdd d eZeeed d d Z d-eeedddZ!d.eeee"dddZ#e$dedddedeefeeeedddZ%e&deedddZ'e&dddeefee ee eed d!d"Z(e&d#ddeefe ee eed$d%d&Z)e&d'd(d)Z*e e d*d+d,Z+dS)/) APIRouterDepends HTTPExceptionstatusHeaderRequest)Session)ListDictOptionalAny)datetimetimezone timedeltaN) BaseModel)get_dbc@s.eZdZUeed<eed<eeefed<dS)ActivityWatchEvent timestampdurationdataN)__name__ __module__ __qualname__str__annotations__floatr r rr./stateless_webhook.pyrs r) developer_idprovided_tokenreturncCsntdd}|d|dtj}t|}ddl }t |dd}| | d}||kS)z;Validate token without database lookup - uses master secretZ MASTER_SECRETz'your-master-secret-for-token-generation:rN0=)osgetenvr nowZyearhashlibZsha256encodeZ hexdigestbase64bytesfromhexZurlsafe_b64encodedecoderstrip)rrZ master_secretZ token_inputZ token_hashr)Z token_bytesZexpected_tokenrrrvalidate_stateless_tokens r.)app_name window_titler cs|sdS||r|nd}tfdddDr:dStfdddDrTd Stfd dd Drnd Stfd ddDrdStfdddDrdStfdddDrdSdS)z5Categorize application based on name and window titleotherr/c3s|]}|kVqdSNr.0browserapp_name_lowerrr ,sz)categorize_application..)chromefirefoxsafariedgeZoperaZbraver6c3s|]}|kVqdSr3rr5Zider7rrr90s) vscodez visual studiopycharmintellijZsublimeZatomZvimZemacsz notepad++cursorcode developmentc3s|]}|kVqdSr3r)r5dbr7rrr94s)ZdatagripZpgadminZmysqlZdbeaverZnavicatZ sqlserverZoracledatabasec3s|]}|kVqdSr3r)r5Zprodr7rrr98s) ZwordZexcelZ powerpointZoutlookZteamsZslackZdiscordZzoomZnotionZobsidianZpostman productivityc3s|]}|kVqdSr3r)r5Zmediar7rrr9<s)ZspotifyZyoutubeZvlcz media playerZnetflixZtwitch entertainmentc3s|]}|kVqdSr3r)r5systemr7rrr9@s) ZexplorerfinderZterminalcmdZ powershellz task managerlockZdwmZwinlogonrI)lowerany)r0r1window_title_lowerrr7rcategorize_application#s"rP)r1r0urlr c s ddd|dd}|s|s|S|r(|nd|r8|nd}tfdddDrd|kr|d}t|d kr|d }|d }||d d |kr|d|n|d|d|d|SntfdddDr|rzd dlm}||} | j dddksdkrH|d| j p,ddd|d|WSdddd d!d"g} tfd#d| Dr|d$d%|d|WSWnt k rYnXd|kr|dd } || d&d'| d|S|r| d(dnd)} || d|r| d*|n| d|S)+z@Extract project information from window title, app name, and URLNWork) project_name project_type file_pathrQdetailed_activityr/c3s|]}|kVqdSr3rr>r7rrr9Vsz'extract_project_info..)rBr?rCr@rAz - rZ Development./zCoding: z in )rSrTrUrVc3s|]}|kVqdSr3rr4r7rrr9fs)r:r;r=r<)urlparsezwww.Z localhostz 127.0.0.1z localhost:Z3000zWeb DevelopmentzLocal Development: )rSrTrVz github.comzstackoverflow.comzdocs.zapi.z developer.zconsole.c3s|]}|kVqdSr3r)r5Z work_domain)domainrrr9xsz Web Researchz Research: z Web Browsingz Browsing: z.exeUnknownz: ) rMrNsplitlenstripupdateZ urllib.parser[ZnetlocreplaceZport Exception) r1r0rQ project_inforOpartsfilenameprojectr[ZparsedZ work_domainsZ page_titleZclean_app_namer)r8r\rextract_project_infoEsz       rhz/activitywatch/webhook.z Developer-ID)alias)requestr authorizationrEcs|dstddd|dd}t||sJtd|tdddtd |z|Id H}d }|D]\}}t |t rx|D]t} t | t sq| d } | d d } | di} | r| dkrqzt | dd} | dp| dd}| dd}| dd }|r$|dkr(Wqt||}t|||}d dlm}|||j|k|j| k|j|k|j| k}|rWq||||||d|| | |d|d|dd }|||d7}Wqtk r}ztd|WYqW5d }~XYqXqqx|td|d |d!d"|d#|t t j!"d$WStj#k rntd%d&dYnRtk r}z2|$td'|td(d)t%|dW5d }~XYnXd S)*zCReceive ActivityWatch data via webhook - stateless token validationzBearer izInvalid authorization header)Z status_codeZdetail rXzInvalid token for developer zInvalid developer or tokenz$Received ActivityWatch webhook from NrrrrZ+00:00ZappZ applicationr]titler/rQActivityRecordrUrSrTrV) rapplication_namer1rQrUcategoryrrrSrTrVzError processing event: zSuccessfully processed z activities from successz Processed z activities)rmessagerrizInvalid JSON dataz(Error processing ActivityWatch webhook: izProcessing error: )& startswithrr^r.loggerZwarninginfojsonitems isinstancelistdictgetr fromisoformatrbrPrhmodelsrrqueryfilterrrrsrfirstaddrcerrorZcommitr&rutc isoformatZJSONDecodeErrorZrollbackr)rjrrkrEtokenZ webhook_dataZprocessed_activitiesZ bucket_nameZ bucket_dataZeventZ timestamp_strrrrr0r1rQrtrdrrexistingZactivity_recorderrr'receive_activitywatch_webhook_statelesss                    rz/activitywatch/validate-token)rrcs"t||}||ttjdS)z1Test endpoint to validate a token (for debugging))rZ token_validr)r.r r&rrr)rrZis_validrrrvalidate_token_endpoint s  rz//activitywatch/developer-summary/{developer_id})r start_dateend_daterEc s|rt|dd}nttjjddddd}|rJt|dd}n ttj}ddlm}|| |j |k|j |k|j |k }|s|| | dddiiddSt|}|| | d|dS) zFGet activity summary for a developer (no user authentication required)rnrorZhourZminutesecondZ microsecondrqstartend)total_activities total_time categoriesprojects)r date_rangesummary)r rrbr&rrrrrrrrrallrprocess_developer_activities) rrrrErrrr activitiesrrrrget_developer_summary_statelesss,     rz/activitywatch/team-summary)rrrEcs|rt|dd}nttjjddddd}|rJt|dd}n ttj}ddlm}ddlm }| ||j  |j d|j|k|j|k}g}d} d} |D]\} | sq| | |j | k|j|k|j|k} t| } | | d7} | | d 7} | d d }d | kr@| d dd }d |}|| || d q||d|t|| | ddd| ddS)z;Get team-wide summary without requiring user authenticationrnrorr)distinctrqNrr_rl)rnamerr.2fh)rtotal_time_formattedr)r team_dataZtotal_developersZ team_totals)r rrbr&rrZ sqlalchemyrrrrrrrZisnotrrrrpr^joinappendrr_)rrrErrrrrZ developer_idsrZtotal_team_timeZtotal_team_activitiesZdev_idrrZdeveloper_nameZ name_partsrrrget_team_summary_statelessCsX           rz/healthcsddttjddS)zHealth check endpointZhealthyZ statelessz:ActivityWatch webhook endpoint is running (stateless mode))rmoderrv)r r&rrrrrrr health_checks r)rr c Cs|sddiidddStdd|D}i}i}d}|D]}|jpDd}|jpNd}|j}||krlddd||<||dd 7<||d |7<||krdd|jpd d ||<||dd 7<||d |7<d d d ddddd} | |d} ||| 7}q8|dkr||dnd} |D]} | d ddd| d<q*|D]} | d ddd| d<qRt|||ddd||dddt| d ||dS)z*Process activities into summary statisticsr)rrrr working_hoursproductivity_percentagecss|] }|jVqdSr3)r)r5activityrrrr9sz/process_developer_activities..r2r])countrrrXrrR)rrtypeg?g333333?g?g?g)rDrFrGr6r2rIrHdrrrZduration_formatted)rrrrZworking_hours_formattedrrr) sumrtrSrrTrvaluesr_round)rrrrrrrtrgrZcategory_weightsZweightrZ category_dataZ project_datarrrrs`      r)r/)N),ZfastapirrrrrrZsqlalchemy.ormrtypingr r r r r rrrzr'r$ZpydanticrrFrZloggingZrouterZ getLoggerrrxrrboolr.rPr~rhZpostrrrrrrrrrrrsf     "R q +F