U [Mh'f@sddlmZmZmZmZmZddlmZddlm Z m Z m Z ddl m Z m Z mZddlZddlZddlmZddlZddlZddlZddlmZddlmZddlZdd lmZeZGd d d eZGd d d eZGdddeZ GdddeZ!dGe"e"e"dddZ#e"e"e$dddZ%e"e e"dddZ&e"e e"dddZ'e"e e"dddZ(e"e$dd d!Z)e"e e"dd"d#Z*e"e e"dd$d%Z+e"e e"ee"d&d'd(Z,e-d)ed*d+d,ed*eefee"e"ed-d.d/Z.e-d0ed*d1d,eefe e"ed2d3d4Z/e0d5ed*d1d,eefe"ed6d7d8Z1e0d9ddeeeefe"e e"e e"ej2ed:d;d<Z3e0d=ddeeeefe e"e e"ej2ed>d?d@Z4e0dAdBdCZ5e ej6e dDdEdFZ7dS)H) APIRouterDepends HTTPExceptionstatusHeader)Session)ListDictOptional)datetimetimezone timedeltaN) BaseModel)get_db)get_current_user)urlparsec@s>eZdZUeed<eed<eed<eed<eed<eed<dS)DeveloperActivityapplication_name window_titleduration timestampZ bucket_name developer_idN)__name__ __module__ __qualname__str__annotations__floatrr./multi_developer_api.pyrs rc@s*eZdZUeed<eeed<eed<dS)ActivityDataPayloadr activitiesrN)rrrrrrrrrrrr s  r c@sBeZdZUdZeeed<eed<dZeeed<dZe ed<dS)DeveloperRegistrationNrnameemailTactive) rrrrr rrr$r%boolrrrrr"s r"c@sVeZdZUeed<eed<dZeeed<eed<dZee ed<dZ ee ed<dS) DeveloperInforr#Nr$r% last_syncrrecent_activities) rrrrrr$r r&r(r r)intrrrrr'$s r')app_namerreturncs||}tfdddDr*dStfdddDrDdStfddd Dr^d Stfd dd Drxd StfdddDrdStfdddDsȈdrtfdddDrdSdS)z5Categorize application based on name and window titlec3s|]}|kVqdSNr.0browserZapp_name_lowerrr 3sz)categorize_application..ZchromeZfirefoxZsafariZedgeZoperaZbraver1c3s|]}|kVqdSr.rr0Zider2rrr37s) vscode visual studiopycharmintellijsublimeatomvimemacs notepad++cursorcode developmentc3s|]}|kVqdSr.r)r0dbr2rrr3;sZdatagripZpgadminZmysqlZdbeaverZnavicatZ sqlserverZoracledatabasec3s|]}|kVqdSr.r)r0Zprodr2rrr3?s) wordexcel powerpointoutlookteamsslackdiscordzoomZnotionZobsidianZpostman productivityc3s|]}|kVqdSr.rr0Zmediar2rrr3Cs)spotifyyoutubevlc media playernetflixZtwitch entertainmentc3s|]}|kVqdSr.r)r0systemr2rrr3Gs)explorerfinderZterminalcmdZ powershellz task managerlockdwmwinlogonZcsrssZlsassservicesZsvchost searchhostnotepad.exec3s|]}|kVqdSr.r)r0Zsys_exer2rrr3Hs)rYrVrZr[r]r^rUother)loweranyendswith)r,rwindow_title_lowerrr2rcategorize_application-s&re)rr,r-c sdddddddd}||}tfdddDrt||d<d|kr|dd}d ||d <|dr|d d |dd 7<nd ||d <t|}|d |d <|d|d<ntfdddDrzt||d<|drTd|dkr4|ddd\}}d|d||d <||d <nd|d|d <|d|d <nd||d <t||d <d|d<ntfdddDr t ||d<|drd|d|d <d|d|d <n0| dd dd}d|d||d <||d <d |d<nt ||d!ks& dr||d"<| drF| ddn|}|r||krt |dkrd#|d||d <nd#||d <||d <d$|d<ntfd%dd&Drt||d<|drd'|d|d <|d|d <nd'||d <||d <d(|d<n~tfd)dd*DrPd+||d <| dd|d <d,|d<n>tfd-dd.Drd/||d <| dd|d <d0|d<|d s|d1||d <| dd|d <d2|d<|S)3z;Extract detailed information from window title and app nameN)url file_pathdatabase_connectionspecific_processdetailed_activity project_name project_typec3s|]}|kVqdSr.rr/r2rrr3]sz(extract_detailed_info..r4rf - rz Browsing: rjz ()rkrlc3s|]}|kVqdSr.rr5r2rrr3ms) r6r7r8r9r:r;r<r=r>r?rg/zCoding: z in Z Developmentc3s|]}|kVqdSr.r)r0Zdb_toolr2rrr3~srCrhzDatabase: Connected to zDB: r_r+Z64z Database: ZDatabaserUrizSystem: ZSystemc3s|]}|kVqdSr.r)r0Zofficer2rrr3s)rErFrGrHzOffice: Z Productivityc3s|]}|kVqdSr.r)r0Zcommr2rrr3s)rIrJrKrLZskypezCommunication: Z Communicationc3s|]}|kVqdSr.rrNr2rrr3s)rOrPrQrRrSzMedia: Z Entertainmentz: Work)rarbextract_url_from_browser_titlesplitstripextract_project_from_browserextract_file_pathrsplitextract_project_from_ide_titleextract_database_inforeplacererclenextract_office_document) rr,inford page_titleZ project_infoprojectfilenameZclean_app_namerr2rextract_detailed_infoMs           "      r)rr-c Csdddg}|D]}t||tj}|rd|ks>d|kr|}|dsXd|}z&t|}|jd|j|jWS|YSXq|d }d d d d d dddd}| }| D]\} }| |kr|SqqdS)z1Enhanced URL extraction from browser window titlezN^(.+?)\s*-\s*(Google Chrome|Mozilla Firefox|Microsoft Edge|Safari|Opera|Brave)zhttps?://[^\s\)]+z www\.[^\s\)]+ZhttpZwwwzhttps://z://rpzhttps://claude.aizhttps://chat.openai.comzhttps://github.comzhttps://stackoverflow.comzhttps://google.comzhttps://youtube.comzhttps://gmail.comzhttps://linkedin.com)ZclaudeZchatgptZgithubZ stackoverflowZgooglerPZgmailZlinkedinN) research IGNORECASEgroup startswithrZschemeZnetlocpathrtraitems) rZbrowser_patternspatternmatchrfZparsedr~Z title_to_urlZ page_lowerkeyrrrrrs>  rrcCsddddddg}|D]}t||tj}|rt|dkr|d}t|dkrh|dnd }|d kr|Sd |krt|d dkr|d |S|Sq|dSqd S)z'Extract file path from IDE window titlez&([A-Za-z]:\\[^|<>:*?"]+\.[a-zA-Z0-9]+)z(/[^|<>:*?"]+\.[a-zA-Z0-9]+)zF^([^-]+\.[a-zA-Z0-9]+)\s*-\s*([^-]+)\s*-\s*(Cursor|Code|Visual Studio)zI^([^-]+\.[a-zA-Z0-9]+)\s*-\s*(Cursor|Code|Visual Studio|PyCharm|IntelliJ)z<^([^-]+)\s*-\s*(Cursor|Code|Visual Studio|PyCharm|IntelliJ)$z^([^\\/:*?"<>|]+\.[a-zA-Z0-9]+)rpr+)r?r@r7r8r9.roN) rrrr{groupsrrtrars)rpatternsrrZ first_partZ second_partrrrrvs( "  rvcCs*d}t||tj}|r&|dSdS)z*Extract project name from IDE window titlez:^(.+?)\s*-\s*(Cursor|Code|Visual Studio|PyCharm|IntelliJ)$rpN)rrrrrt)rZ ide_patternrrrrrxs rxcCsd|krz|dd}d|krPtd|}|r<|dnd}d|ddStd |}|rp|dd dS|d dSd d dS) z5Extract project information from browser window titlermrz localhost:zlocalhost:(\d+)rpunknownzWeb Development)rkrlz([a-zA-Z0-9.-]+\.[a-zA-Z]{2,})z Web Researchz Web BrowsingZBrowser)rsrtrrr)rr~Z port_matchZportZ url_matchrrrrus&  rucCs4dddg}|D] }t||}|r|SqdS)z Extract database connection infoz @([^@\s]+)z([a-zA-Z0-9_]+@[a-zA-Z0-9.-]+)z&([a-zA-Z0-9_]+)\s*-\s*([a-zA-Z0-9.-]+)N)rrr)rZ db_patternsrrrrrry&s rycCsJddddddg}|}|D]}||kr||d}q6q|rF|SdS) z3Extract document name from Office application titlez - Wordz - Excelz - PowerPointz - Microsoft Wordz - Microsoft Excelz - Microsoft PowerPointr+N)rzrt)rZoffice_suffixestitlesuffixrrrr|5s r|)r#r$rBr-c Csddl}ddlm}|}ddd|D}d|}|rt||dd}|d|}n| j }|d|}|} d } | t j t j j|kr| d| }| d 7} q|S) z2Generate a unique developer ID from name and emailrN)r r+css&|]}|s|r|ndVqdS)r+N)isalnumisspace)r0crrrr3Hsz/generate_unique_developer_id.._rp)hashlibr rartjoinrsZsha256encodeZ hexdigestnowyearquerymodels Developerfilterrfirst) r#r$rBrr Z clean_nameZ email_hashZbase_idrZ original_idZcounterrrrgenerate_unique_developer_idAs     rz/receive-activity-data.z Developer-ID)alias)payloadr authorizationrBc s|dstddd|dd}|tjtjj|ktjj|ktjj dk }|sftdddg}|j D]}zt |jd d }t|j|j} tj|j|j|j|j|t|j|j| d | d | d | d| d| d| dd } || || Wqptk r:} ztd| WYqpW5d} ~ XYqpXqpt tj|_z.|ddt|d|j t||dWStk r} z"|!tddt"| dW5d} ~ XYnXdS)z3Receive activity data from remote developer systemszBearer zInvalid authorization headerZ status_codeZdetail rpTzInvalid developer or tokenZ+00:00rfrgrhrirjrkrl) rrrrrcategoryrfrgrhrirjrkrlzError processing activity: NsuccesszStored z activities for )rmessageZactivities_countrizDatabase error: )#rrrsrrrrr api_tokenr%rr!r fromisoformatrrzrrrActivityRecordrreaddappend Exceptionprintrr utcr(commitr{r#Zrollbackr) rrrrBtoken developerZstored_activitiesactivityrZ detailed_infoZactivity_recorderrrreceive_activity_data]s\          rz/register-developerz Admin-Token)developer_info admin_tokenrBcsddl}|dd}||kr(tddd|tjtjj|jk}|rXtdd d|jspt |j |j ||_t d }tj|j|j |j |d d }|||||d |j|d|j ddS)z%Register a new developer (admin only)rN ADMIN_TOKEN!timesheet-admin-2025-secure-tokenrAdmin access requiredrizDeveloper already registered T)rr#r$rr%rz Developer z registered successfully)rrrr)osgetenvrrrrrrrrr#r$secretsZ token_urlsaferrZrefresh)rrrBrexpected_admin_tokenexistingrrrrrregister_developers6          rz /developers)rrBc sddl}|dd}||kr(tddd|tj}g}|D]f}|tjtjj |j ktjj t t jtdd k}||j |j|j|j|j||jd q@|t|d S) z+List all registered developers (admin only)rNrrrrr)Zhours)rr#r$r%r(r)r) developersZ total_count)rrrrrrallrrrZ created_atr rr rr countrr#r$r%r(rr{)rrBrrrresultZdevr)rrrlist_developerss*     rz!/developer-summary/{developer_id})r start_dateend_date current_userrBc s|rt|dd}nttjjddddd}|rJt|dd}n ttj}|tj tjj |k }|st ddd|tj  tj j |ktj j|ktj j|k}|s||j||dddiid d St|} ||j||d| d S) z-Get activity summary for a specific developerrrrZhourZminutesecondZ microsecondizDeveloper not foundrstartend)total_activities total_time categoriesprojects)rZdeveloper_name date_rangesummary)r rrzrr rrrrrrrrrrrr# isoformatprocess_developer_activities) rrrrrBrrrr!rrrrget_developer_summarys8          rz/team-dashboard)rrrrBc s0|rt|dd}nttjjddddd}|rJt|dd}n ttj}|tj tjj dk }g}d}d} |D]v} |tj  tj j | j ktj j|ktj j|k } t| } || d7}| | d7} || j | j| j| j| dq||d |t|||d d d | d dS)zGet team-wide dashboard datarrrrTrr)rr#r$r(rr.2fh)rtotal_time_formattedr)r team_dataZtotal_developersZ team_totals)r rrzrr rrrrrr%rrrrrrr#r$r(rr{) rrrrBrrrrZtotal_team_timeZtotal_team_activitiesrr!rrrrget_team_dashboard4sD         rz/healthcsdttjddS)z+Health check endpoint for developer systemsZhealthyzTimesheet server is running)rrr)r rr rrrrrr health_checknsr)r!r-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|] }|jVqdSr.)r)r0rrrrr3sz/process_developer_activities..r`ZUnknown)rrrrprrq)rrtypeg?g333333?g?g?g)rArDrMr1r`rUrTdrrrZduration_formatted)rrrrZworking_hours_formattedrrr) sumrrkrrlgetvaluesr{round)r!rrrrrrrrZcategory_weightsZweightrZ category_dataZ project_datarrrrws`      r)r+)8ZfastapirrrrrZsqlalchemy.ormrtypingrr r r r r rrZpydanticrrZschemasZcrudrDrZauthrrZ urllib.parserZrouterrr r"r'rredictrrrrvrxruryr|rZpostrrrrZUserrrrrrrrrrs       j*   L 0 "59