U %hG@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 rr4/var/www/html/timesheet/backend/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 nowyearhashlibsha256encode hexdigestbase64bytesfromhexurlsafe_b64encodedecoderstrip)rrZ master_secretZ token_inputZ token_hashr, token_bytesZexpected_tokenrrrvalidate_stateless_tokens r3)app_name window_titler cs|sdS||r|nd}tfdddDr:dStfdddDrTd Stfd dd Drnd Stfd ddDrdStfdddDrdStfdddDrdSdS)z5Categorize application based on name and window titleotherr4c3s|]}|kVqdSNr.0browserapp_name_lowerrr ,sz)categorize_application..)chromefirefoxsafariedgeZoperaZbraver;c3s|]}|kVqdSr8rr:Zider<rrr>0s) vscodez visual studiopycharmintellijZsublimeZatomZvimZemacsz notepad++cursorcode developmentc3s|]}|kVqdSr8r)r:dbr<rrr>4s)ZdatagripZpgadminmysqlZdbeaverZnavicatZ sqlserveroracledatabasec3s|]}|kVqdSr8r)r:prodr<rrr>8s) wordexcelZ powerpointZoutlookZteamsZslackZdiscordZzoomZnotionZobsidianZpostman productivityc3s|]}|kVqdSr8r)r:Zmediar<rrr><s)ZspotifyZyoutubeZvlcz media playerZnetflixZtwitch entertainmentc3s|]}|kVqdSr8r)r:systemr<rrr>@s) ZexplorerfinderZterminalcmdZ powershellz task managerlockZdwmZwinlogonrS)lowerany)r5r6window_title_lowerrr<rcategorize_application#s"rZ)r6r5urlr 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_pathr[detailed_activityr4c3s|]}|kVqdSr8rrCr<rrr>Vsz'extract_project_info..)rGrDrHrErFz - rZ Development./zCoding: z in )r]r^r_r`c3s|]}|kVqdSr8rr9r<rrr>fs)r?r@rBrA)urlparsezwww. localhostz 127.0.0.1z localhost:Z3000zWeb DevelopmentzLocal Development: )r]r^r`z github.comzstackoverflow.comzdocs.zapi.z developer.zconsole.c3s|]}|kVqdSr8r)r:Z work_domain)domainrrr>xsz Web Researchz Research: z Web Browsingz Browsing: z.exeUnknownz: ) rWrXsplitlenstripupdate urllib.parserenetlocreplaceport Exception) r6r5r[ project_inforYpartsfilenameprojectreparsedZ work_domainsZ page_titleZclean_app_namer)r=rgrextract_project_infoEsz       rwz/activitywatch/webhook.z Developer-ID)alias)requestr authorizationrJcs|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) status_codedetail rbzInvalid token for developer zInvalid developer or tokenz$Received ActivityWatch webhook from NrrrrZ+00:00app applicationrhtitler4r[ActivityRecordr_r]r^r`) rapplication_namer6r[r_categoryrrr]r^r`zError processing event: zSuccessfully processed z activities from Zsuccessz Processed z activities)rmessagerrizInvalid JSON dataz(Error processing ActivityWatch webhook: izProcessing error: )& startswithrrir3loggerwarninginfojsonitems isinstancelistdictgetr fromisoformatrorZrwmodelsrqueryfilterrrrrfirstaddrqerrorcommitr&rutc isoformatJSONDecodeErrorrollbackr)ryrrzrJtokenZ webhook_dataZprocessed_activitiesZ bucket_nameZ bucket_dataeventZ timestamp_strrrrr5r6r[rrrrexistingZactivity_recorderrr'receive_activitywatch_webhook_statelesss                    rz/activitywatch/validate-token)rrcs"t||}||ttjdS)z1Test endpoint to validate a token (for debugging))rZ token_validr)r3r r&rrr)rris_validrrrvalidate_token_endpoint s  rz//activitywatch/developer-summary/{developer_id})r start_dateend_daterJc 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)rrrhourminutesecond microsecondrstartend)total_activities total_time categoriesprojects)r date_rangesummary)r rror&rrrrrrrrallrprocess_developer_activities) rrrrJrrr activitiesrrrrget_developer_summary_statelesss,     rz/activitywatch/team-summary)rrrJcs|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 authenticationrrrr)distinctrNrr_r})rnamerr.2fh)rtotal_time_formattedr)r team_dataZtotal_developersZ team_totals)r rror&rr sqlalchemyrrrrrrisnotrrrrrijoinappendrrj)rrrJrrrrZ 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))rmoderr)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|] }|jVqdSr8)r)r:activityrrrr>sz/process_developer_activities..r7rh)countrrrbrr\)rrtypeg?g333333?g?g?g)rIrMrQr;r7rSrRdrrrduration_formatted)rrrrZworking_hours_formattedrrr) sumrr]rr^rvaluesrjround)rrrrrrrrurZcategory_weightsweightrZ category_dataZ project_datarrrrs`      r)r4)N),fastapirrrrrrsqlalchemy.ormrtypingr r r r r rrrr(r$pydanticrrMrloggingrouter getLoggerrrrrboolr3rZrrwpostrrrrrrrrrrrsf     "R q +F