GVjJ|UdZddlmZddlZddlZddlmZddlmZddl m Z e e jjjZddgZiZd ed <d#d Zd$d Zd%d Zd&dZd'dZd(dZd)dZd*dZd+dZd,dZd-dZd.dZd/dZd0dZd1d2dZ dZ!d3dZ"d4dZ#d5dZ$d6dZ%d7d8dZ&d9d Z' d: d;d!Z(d"Z)y).json``). The first action on an account opens a browser once; sign in with THAT account (work vs business) and the token caches + refreshes silently forever after. No API key, no per-call billing. Runs IN-PROCESS via the Agent SDK MCP server (key ``"gmail"``; Claude sees ``mcp__gmail__``). Google client libs are imported lazily so importing this module never hard-fails when they're missing. Config (``.env`` / env): GMAIL_ACCOUNTS comma-separated labels to offer, default "work,business" GMAIL_DEFAULT label used when a tool call omits ``account``, default the first in GMAIL_ACCOUNTS ("work") GOOGLE_OAUTH_JSON shared desktop OAuth client (default google_oauth.json) Set ``MCP_GMAIL=0`` to disable. Enabling ALSO requires the Gmail API turned on in the same Google Cloud project as Calendar and the ``gmail.*`` scopes added to that project's OAuth consent screen (see README). ) annotationsN) EmailMessage) getaddresses)Pathz,https://www.googleapis.com/auth/gmail.modifyz*https://www.googleapis.com/auth/gmail.sendzdict[str, tuple]_SVCctjjdd}|jdDcgc]1}|j s|j j 3c}Scc}w)NGMAIL_ACCOUNTSz work,business,)osenvirongetsplitstriplower)rawls A/Users/ahmed/devFolder/Ultron/claude-voice/voice/gmail_control.py_labelsr2sL **..)? ;C'*yy~ C~!AGGIOO ~ CC Cs A- A-ctjjddjj }t }||vr|S|r|dSdS)N GMAIL_DEFAULTrwork)r r r rrr)dlabelss r_default_labelr7sK +11399;A YFV 1Bv&)B6BcPttjjddz S)NGOOGLE_OAUTH_JSONzgoogle_oauth.json)_ROOTr r r rr _oauth_pathr!=s 2::>>"57JK KKrctd|dz S)N google_gmail_z.json)r)labels r _token_pathr%As ]5'/ //rcztjjdddk(rytj S)zWhether the Gmail MCP should be exposed to Claude. On when not disabled and the shared OAuth client exists (individual accounts consent on first use). MCP_GMAIL10F)r r r r!existsr rrenabledr+Es/ zz~~k3'3. =   !!rcdd|dgiS)z2Wrap a plain string as an MCP text content result.contenttext)typer.r msgs r_textr2Ns 56 77rc|s tS|jj}tD]}||k(s ||vs||vs|cStS)zvNormalise a spoken account label to a configured one (default if unknown/empty). 'my work'/'work email' -> 'work'.)rrrr)r$rcfgs r_resolver5SsS  Ay !8sax18J  rc|tvr t|Sddlm}ddlm}ddlm}ddlm}t|}d}|jr|jt|t}|r |js|r/|jr#|j r|j#|nYt%d|d|j'd |j)tt+t}|j-d }|j/|j1|d d |d }|j3j5dj7j9d|} || ft|<t|S)uReturn (gmail_service, email_address) for one account, running the one-time browser consent if that account has no cached token. Blocking — call via asyncio.to_thread so the voice loop never stalls.r) Credentials)InstalledAppFlow)Request)buildNz& [gmail] first-time sign-in for the 'u4' account — a browser will open; sign in with the z# Google account and approve Jarvis.)portgmailv1F) credentialscache_discoveryme)userId emailAddress)rgoogle.oauth2.credentialsr7google_auth_oauthlib.flowr8google.auth.transport.requestsr9googleapiclient.discoveryr:r%r*from_authorized_user_filestr_SCOPESvalidexpired refresh_tokenrefreshprintupperfrom_client_secrets_filer!run_local_server write_textto_jsonusers getProfileexecuter ) r$r7r8r9r: token_pathcredsflowsvcaddrs r_servicer\_sC }E{5:6/U#J E55c*owO  U]]u':': MM') $ :5'B99>H01 2$<<KM"G-D))q)1Eemmo. 5% HC 99; ! ! ! . 6 6 8 < < D+DK ;rcdtj|jjS)N)base64urlsafe_b64encodeas_bytesdecoder0s r_b64rbs"  # #CLLN 3 : : <>EgiGGJ3 43q1T73C 4 -eWJtfBGGwjb ) *E IIK " & &C 7 ' 99@  EE)R $ $Y 3%% 2&t, 3%r$q&/**T!V_4EFAy)*$wi 9 :  99U  5s% Ec|g}d}|r|jd}|jdd}|jdijd}|dk(r'|r%tj|j ddS|d k(r)|r'|s%tj|j dd}|j |jd gxsg|r|rdd l}|jd d|Sy )zEDepth-first search for a text/plain part; fall back to any text/html.rrmimeTyperhdata text/plainutf-8replace text/htmlpartsNz<[^>]+>z(no readable body))popr r^urlsafe_b64decoderaextendresub)rstackhtmlpmimerrs r _walk_bodyrs IE D  IIaLuuZ$uuVR $$V, < D++D188)L L ; 4++D188)LD QUU7B'-2.  vvj"d++ rc t|\}}|jjjd|dj }|jdijdg}t |jdi}t |dkDr|dddz}dt|d d t|d d t|d dt|dd| S)Nr@fullrArrrrwiu …(truncated)zFrom: rz To: rdz Subject: rfz Date: rz )r\rTrnr rVrlenry)r$rrZr[rrxrhs r_readrsIC ""$3v"FNNPA i  B/A aeeIr* +D 4y4ET{//T!V_%VDDM?;Q *+8DFO3DD PQrct|\}}|jjjd|dgdj }|jdijdg}t |d}t |d} | j jd sd | z} t |d } t |d d z| zj} t} || d<| | d<d} |r|j ht|gDchc]\}}|r|j c}}z}g}tt |dt |dgD]K\}}|s |j |vs|j|j |j|Mdj|} | r| | d<| r | | d<| | d <| j||jjjdt!| |jddj d|| rd| dndzd|d|dzScc}}w)zReply in-thread to a message id, preserving threading headers. reply_all=True keeps every other recipient (original To + Cc, minus us).r@r)rrdrerf Message-ID Referencesrrrwrrfzre:zRe: rr rdrre, z In-Reply-TothreadId)rrrgz Replied to rirjz on your rkrl)r\rTrnr rVryr startswithrrraddrrrmrorb)r$rrh reply_allrZr[origrxrpsubjmsg_idrefsr1rr_aseenkeep_ns r_replyrsEIC 99;   ! % %J' & ()0   B##Ir2A aB 9 D ::< " "5 )} !\ "F L !C '& 0 7 7 9D .CCIC N B ~!-rd!3!:!304q!78"#!3!::!44=$q$-"@AEBQWWYd*# ABYYt_ CI #M LOODIIK#YDHHZ,@ A CCJ79" 2E"Q-2 >% 4&3 45#!:sI i@c|xsdjddjdd}tj|dt| dzzzS)z|jd}||j|jdgxsg|r=yyw)zCYield every MIME part of a message, depth-first (payload included).rrN)rrr )rrrs r _walk_partsr sC IE  IIaL QUU7B'-2. s AAAct|jdgxsgdj}|jdr|j dr|ddS|S)z9The part's Content-ID with the stripped.rwz Content-ID<>)ryr rrendswith)partvs r_cidrsM TXXi $ *L9??AAll3'AJJsO1Qr7BBrc dd\}}i}g}t|jdixsiD]}|jddxsd}|jdr,|jdixsi}|jd} |jdxsd} t|} | r|jd r| } | s|jd rt |jd xsd t krn|j jjjd |jdd|d jjd} | r2|tjt| jd|| <d| rI|j|jd dxsd| |t |jd xsd dd|dk(r"| r |t| jddz }|dk(s| s|t| jddz }||||dS)uSplit one full-format message into its text body, HTML body, inline (cid:) images and real attachments. Bodies are NOT truncated — the card shows the whole thing, unlike ``_read`` which is a TTS summary.)rrrrrz multipart/rhrfilenamezimage/ attachmentIdsizerr@rrA messageIdr)rb64)att_idrrrpathrrrr) body_text body_htmlinline attachments)rr rrint _INLINE_MAXrTrnrrVr^ b64encoderrar) rZrr.rrattsrrrhrfnamecidrs r _parts_ofrsJD$ FD y"-3 4uuZ$* ??< ( uuVR &Bxxj!'R1g 4??8,C488N3(-A.+=iik**,88:>>155r?N+?--4WYss6{'+&,&6&6uSz&B&I&I&KMs   KK488NB#?#E2%*D!$TXXf%5%:!;RI J  < D E$K&&w : :D [ T E$K&&w : :D758DF !!rct|\}}|jjjd|xsd|j }|j dgxsgDcgc]}|d c}Scc}w)zAMessage ids matching a Gmail query, newest first (empty = inbox).r@r{r|rnr)r\rTrnrrVr )r$rrrZ_addrrrs r search_idsr?sz%JC ))+   % %u* u & >>Egi WWZ4:: ;:AdG: ;; ;s) A8c t|\}} |jjjd|dj }g}|jdgxsgD]}|jdijdgxsg}|j|jd d t|d t|d t|d t|dt|dt|jdxsddz d|jdxsgvdt|||jdtd|Dd}|jd ||||dS#t $r|jj jd|dj }|jjjd|jd|dj }YwxYw)zA whole THREAD as structured data for the HUD card. `ident` is a thread id OR any message id inside it (Gmail ids are interchangeable-looking, so a message id is resolved to its threadId on the fly). Oldest first.r@rrminimalrrnrrwrrrrdrerfr internalDaterg@@SENTlabelIds)rfromrprrrqdatetsoutgoingc |dS)Nrr )xs rzthread..`sAdGr)keyc32K|]}|ds |dyw)rqNr ).0rs r zthread..as?$Q!I,AiL$s  z (no subject)) thread_id account_addrrqrn) r\rTthreadsr rV Exceptionrnrryrrsortnext) r$identrZr[trmsgsrxrqs rthreadrGsICO IIK   ! % %E& & 229') D UU:r " (b ( EE)R $ $Y 3 9r eeD"oFO44=Q Ay)46?aeeN+0q1F:155#4#:;  Q  ) II#I$?$?PGtU+TD 22' O IIK " & &E) ' 55E**BHHcXddlm}t|\}}|jj j j d||j}|j|xs||} | jt|j ddt| S)zDownload one attachment to ~/.jarvis/attachments// and return its local path. The filename is sanitised (no path traversal).r) mail_viewr@rrr) voicerr\rTrnrr rV dest_path write_bytesrrH) r$rrrrrrZrrrs r attachmentr fs %JC **,00v& 1 229')   y2FH =DU155,-. t9rc  ddl ddlm}m}dj t xsd}d|dt d}|d d |zd ztttttd d fd }|dd|ztttdd fd }|dd|zttdd fd }|dd|ztttdd fd }|ddid d}|dd|||||gS)!z5Build and return the in-process ``gmail`` MCP server.rN)toolcreate_sdk_mcp_serverrrz `account` picks the mailbox (z); omit it for 'z'. gmail_sendz1Send an email from one of Ahmed's Gmail accounts.z8 `to`/`cc` are comma-separated addresses; `cc` optional.)rprqrhrraccountc Kt|jddj}|s tdS j t t |jd|t|jddt|jddt|jddjd{}t|S7#t$r}td|cYd}~Sd}~wwxYww) Nrprz gmail_send failed: no recipient.rrqrhrrzgmail_send failed: )rHr rr2 to_threadrsr5r)argsrpouteasyncios rr z build_server..gmail_send~s $# $ * * ,;< < 4))x 34bDHHY+,c$((622F.GDHHT2&'--/11C:  1  4.qc23 3 4sG7C?BC C CC?C C<$ C71C<2C?7C<<C? gmail_searchzSearch a Gmail account using Gmail search syntax (e.g. 'from:elie is:unread', 'subject:invoice newer_than:7d', empty = inbox). Returns a list with a message id for each; pass an id to gmail_read.)rrrc vK tdtt|jdxsdd}j t t |jdt|jdd|d{}t|S7#t$r}td|cYd}~Sd}~wwxYww) Nrr rrrzgmail_search failed: ) maxminrr rrr5rHr2r)rrrrrs rrz"build_server..gmail_searchs 63s488G#4#:;R@AE))$((9"56DHHWb)*E33C: 3 6045 5 6sGB9A>BBBB9B B6 B1+B6,B91B66B9 gmail_readz@Read the full body of one message by its id (from gmail_search).)rrcJKt|jddj}|s tdS j t t |jd|d{}t|S7#t$r}td|cYd}~Sd}~wwxYww)Nrrz!gmail_read failed: no message id.rzgmail_read failed: )rHr rr2rrr5rrrrrrs rrz build_server..gmail_reads$((4$%++-<= = 4))x 34c;;C: ; 4.qc23 3 4sF7B#2A?-A=.A?<B#=A?? B  BB B#B  B# gmail_replyzReply in-thread to a message by its id (from gmail_search), keeping the subject and conversation threading. Goes to the original sender.)rrhrc ~Kt|jddj}|s tdS j t t |jd|t|jddd{}t|S7#t$r}td|cYd}~Sd}~wwxYww)Nrrz"gmail_reply failed: no message id.rrhzgmail_reply failed: )rHr rr2rrr5rrs rrz!build_server..gmail_replys $((4$%++-=> > 5)))!45sDHHVR()++C: + 5/s34 4 5sG7B=A BBBB=B B:" B5/B:0B=5B::B=gmail_accountszDList the Gmail accounts Jarvis can use and which one is the default.c  Kg}tD]S}t|j}|tk(rdnd}|rdnd}|j d|d|d|rd|zndUt d j |xsd Sw) Ndefaultrz signed inznot signed in yetz- z (rju — rzNo accounts configured.)rr%r*rrr2r)rrrauthedmarkstates rr z$build_server..gmail_accountssA ^**,F !^%5 592D#)K/BE LL2aS5'T7T>r*JK L  TYYu%B)BCCsBBr<z1.0.0)ruversiontools)rdictreturnr() rclaude_agent_sdkr r rrrrHr) r r accts acct_helpr rrrr rs @r build_serverr-ts\< IIgi *FE1%9#%&b*I , = I F Gc MO 4 O 4 . $ $## 6 8 6 8 6 , L  !c24 44 4 -  (+SS!IK 5 K 5  DD ! <[ r)r) list[str])r)rH)r)r)r$rHr)r)r)bool)r1rHr)r()r$z str | Noner)rH)r$rH)r1rr)rH) r$rHrprHrqrHrhrHrrrHr)rH)rwrrurHr)rH)r$rHrrHrrr)rH)rr(r)rH)r$rHrrHr)rH)F) r$rHrrHrhrHrr/r)rH)rrHr)bytes)rr()rr(r)rH)rr(r)r())r$rHrrHrrr)r.)r$rHrrHr)r()r) r$rHrrHrrHrrHrrHr)rH)*__doc__ __future__rr^r email.messager email.utilsrpathlibr__file__resolveparentrrIr__annotations__rrr!r%r+r2r5r\rbrsryrrrrrrrrrrrr r-r rrr;s8# &$ X ''.. 30  D C L0"8 J= 7( &Q(5b 5 /C $!N<2@!#  '* Zr