U [Mh7@sddlmZmZmZmZmZmZddlmZddl m Z m Z m Z m Z ddlmZmZmZddlZddlmZddlZddlZddlZddlmZddlZeZeeZGdd d eZGd d d eZGd d d eZ d'e!e!e!dddZ"d(e!e!e!e#dddZ$e%dedddedeefee!e!edddZ&e%dedddedeefee!e!edddZ'e(dedd deefe!e!ed!d"d#Z)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./activitywatch_webhook.pyrs rc@s2eZdZUeed<eed<eed<eeed<dS)ActivityWatchBucketidtypehostnameZeventsN)rrrrrr rrrrrrs rc@s2eZdZUeeed<eeefed<eed<dS)ActivityWatchWebhookZbucketsZ timeperiodr!N)rrrr rrr rrrrrr"s  r")app_name window_titlereturncs|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..)chromefirefoxsafariedgeZoperaZbraver+c3s|]}|kVqdSr(rr*Zider,rrr.+s) vscodez visual studiopycharmintellijZsublimeZatomZvimZemacsz notepad++cursorcodeZ developmentc3s|]}|kVqdSr(r)r*dbr,rrr./s)ZdatagripZpgadminZmysqlZdbeaverZnavicatZ sqlserverZoracledatabasec3s|]}|kVqdSr(r)r*Zprodr,rrr.3s) ZwordZexcelZ powerpointZoutlookZteamsZslackZdiscordZzoomZnotionZobsidianZpostmanZ productivityc3s|]}|kVqdSr(r)r*Zmediar,rrr.7s)ZspotifyZyoutubeZvlcz media playerZnetflixZtwitchZ entertainmentc3s|]}|kVqdSr(r)r*systemr,rrr.;s) ZexplorerfinderZterminalcmdZ powershellz task managerlockZdwmZwinlogonr;)lowerany)r$r%window_title_lowerrr,rcategorize_applications"rB)r%r$urlr&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 URLNZWork) project_name project_type file_pathrCdetailed_activityr#c3s|]}|kVqdSr(rr3r,rrr.Qsz'extract_project_info..)r7r4r8r5r6z - rZ Development./zCoding: z in )rDrErFrGc3s|]}|kVqdSr(rr)r,rrr.bs)r/r0r2r1)urlparsezwww.Z localhostz 127.0.0.1z localhost:Z3000zWeb DevelopmentzLocal Development: )rDrErGz github.comzstackoverflow.comzdocs.zapi.z developer.zconsole.c3s|]}|kVqdSr(r)r*Z work_domain)domainrrr.usz Web Researchz Research: z Web Browsingz Browsing: z.exeUnknownz: ) r?r@splitlenstripupdateZ urllib.parserLZnetlocreplaceZport Exception) r%r$rC project_inforApartsfilenameZprojectrLZparsedZ work_domainsZ page_titleZclean_app_namer)r-rMrextract_project_info@sz       rXz/activitywatch/webhook.z Developer-ID)aliasrequest developer_id authorizationr9cs|dstddd|dd}|tjtjj|ktjj|ktjj dk }|sftdddz| Id H}t d |d }|D]\}} t| tr| D]v} t| tsq| d } | d d } | di} | r| dkrqzt| dd}| dp| dd}| dd}| dd }|rB|dkrFWqt||}t|||}|tjtjj|ktjj|ktjj|ktjj| k }|rWqtj|||||d|| ||d|d|dd }|||d7}Wqtk r"}zt d|WYqW5d }~XYqXqqtt j!|_"|#t d|d|j$d d!|d"||j$d#WSt j%k rtd$d%dYnRtk r}z2|&t d&|td'd(t'|dW5d }~XYnXd S))z&Receive ActivityWatch data via webhookzBearer zInvalid authorization headerZ status_codeZdetail rITzInvalid developer or tokenNz$Received ActivityWatch webhook from rrrrZz+00:00ZappZ applicationrNtitler#rCrFrDrErG) r\application_namer%rCrFcategoryrrrDrErGzError processing event: zSuccessfully processed z activities from successz Processed z activities)rmessager\developer_nameizInvalid JSON dataz(Error processing ActivityWatch webhook: izProcessing error: )( startswithrrOquerymodels Developerfilterr\ api_tokenactivefirstjsonloggerinfoitems isinstancelistdictgetr Z fromisoformatrSrBrXZActivityRecordrrdraddrTerrornowrutcZ last_syncZcommitnameZJSONDecodeErrorZrollbackr)r[r\r]r9token developerZ webhook_dataZprocessed_activitiesZ bucket_nameZ bucket_dataZeventZ timestamp_strrrrr$r%rCrerUexistingZactivity_recorderrrreceive_activitywatch_webhooks                          rz/activitywatch/synccst||||IdHS)z0Alternative sync endpoint for ActivityWatch dataN)rrZrrrreceive_activitywatch_sync s rz$/activitywatch/config/{developer_id}z Admin-Token)r\ admin_tokenr9csddl}|dd}||kr(tddd|tjtjj|ktjjdk }|s`td d dddl}|d |d d }d|j d|d|jd|j d|d|jd|j d}||j |dddddS)z2Generate ActivityWatch config.toml for a developerrNZ ADMIN_TOKENz!timesheet-admin-2025-secure-tokenr^zAdmin access requiredr_TizDeveloper not foundZ SERVER_URLZ FRONTEND_URLz!https://your-timesheet-server.comz"# ActivityWatch Configuration for as # Place this file in your ActivityWatch config directory # Windows: %APPDATA%\activitywatch\aw-server\config.toml # Linux: ~/.config/activitywatch/aw-server/config.toml # macOS: ~/Library/Application Support/activitywatch/aw-server/config.toml [server] host = "127.0.0.1" port = 5600 # Team timesheet integration [integrations.timesheet] enabled = true webhook_url = "z//api/v1/activitywatch/webhook" developer_id = "z" api_token = "z" sync_interval = 1800 # 30 minutes in seconds # Webhook configuration [webhooks] enabled = true [[webhooks.endpoints]] url = "zo/api/v1/activitywatch/webhook" method = "POST" interval = 1800 # 30 minutes headers = { "Developer-ID" = "z!", "Authorization" = "Bearer a", "Content-Type" = "application/json" } # Include these buckets in sync include_buckets = [ "aw-watcher-window", "aw-watcher-web-chrome", "aw-watcher-web-firefox", "aw-watcher-afk" ] # Privacy settings [privacy] include_window_titles = true include_urls = true exclude_patterns = [ "*password*", "*login*", "*private*", "*confidential*" ] # Logging [logging] level = "INFO" z-%APPDATA%\activitywatch\aw-server\config.tomlz-~/.config/activitywatch/aw-server/config.tomlzA~/Library/Application Support/activitywatch/aw-server/config.toml)ZwindowsZlinuxZmacos)r\rh config_tomlZinstallation_instructions) osgetenvrrjrkrlrmr\rorpr}rn)r\rr9rZexpected_admin_tokenrZ server_urlrrrrget_activitywatch_configsB         9rz/activitywatch/test-webhookcsddttjdS)z'Test endpoint for ActivityWatch webhookZhealthyz)ActivityWatch webhook endpoint is working)rrgr)r r{rr|Z isoformatrrrrtest_webhook_endpointwsr)r#)N)+ZfastapirrrrrrZsqlalchemy.ormrtypingr r r r r rrrqZpydanticrrkZschemasZcrudr:rZloggingZrouterZ getLoggerrrrrrr"rrBrwrXZpostrrrxrrrrrrsZ     "T w   ^