zTjV/dZddlmZddlZddlZddlZddlmZeejjjZ e dz Z e dz Z dgZddZdd Zdd Zd Zdd Zd Zy)uShared Google Drive storage — in-process MCP so Jarvis (and any AI on the same credentials) can read/write/upload/remove files from one shared folder. This is the storage twin of the shared MEMORY: a single "Jarvis Drive" folder on Ahmed's Google account that every Jarvis instance / Claude can use to pass files around (a CRM export, a doc, a screenshot, notes). AUTH = SERVICE ACCOUNT (key in jarvis_drive_sa.json), NOT a user login. A service account has no Drive of its own, so it can ONLY see the folders Ahmed explicitly SHARES with it — the one "Jarvis Drive" folder. His personal Drive is not merely code-blocked, it is structurally invisible: the SA cannot list or reach it at all. The key also works on any machine (no per-user login, no token expiry), so this is genuinely shared across devices. MCP_DRIVE=0 disables the whole server. Tools (Claude sees them as mcp__drive__): drive_list list files in the shared folder drive_read read a text file's contents (by name or id) drive_write create/overwrite a text file from given content drive_upload upload a local file into the folder drive_download download a folder file to a local path drive_delete remove (trash) a file (by name or id) ) annotationsN)Pathzjarvis_drive_sa.jsonzjarvis_drive_folder.txtz%https://www.googleapis.com/auth/drivecrtjjdddk(rytj S)zHOn when the service-account key exists and it isn't explicitly disabled. MCP_DRIVE10F)osenvironget_SA_KEYexistsA/Users/ahmed/devFolder/Ultron/claude-voice/voice/drive_control.pyenabledr's) zz~~k3'3. >> rcdd|dgiS)Ncontenttext)typerr)msgs r_textr.s 56 77rcFtjjS)N) _FOLDER_FILE read_textstriprrr _folder_idr2s  ! ! # ) ) ++rcddlm}ddlm}|jj t tt}|dd|dS) zBuild a Drive v3 client authenticated as the service account. The SA can only reach folders shared with it, so this can never touch Ahmed's own files.r)service_account)build)scopesdrivev3F) credentialscache_discovery) google.oauth2rgoogleapiclient.discoveryr Credentialsfrom_service_account_filestrr _SCOPES)rrcredss r_servicer,6s@./  ' ' A A G W B &E $E5 IIrct|j}|syt} |jj |dj }||j dxsgvr|SdS#t $rYnwxYwd|d|d}|jj|dd j j d g}|r|d SdS) uRFind a file by id or name — but ONLY inside the Jarvis Drive folder. This is the code jail: with the full-Drive token, a bare id could point at any file in Ahmed's Drive, so we REFUSE anything whose parent isn't our folder. Every read/write/delete goes through here, so nothing outside the shared folder is ever touched.Nzid,name,mimeType,size,parents)fileIdfieldsparents'z)' in parents and trashed=false and name='r!zfiles(id,name,mimeType,size))qspacesr/filesr)r)rrr4r execute Exceptionlist)svc name_or_idreffidfr2r4s r_resolver=@s j/   !C  ,C IIKOO>  @@G  AEE),23q==    SE:3%q AA IIK  q#A  %giGR(8 58&$&sAA20A22 A>=A>c  ddl ddlm}m}ddlmm |ddid fd }|ddd tid fd }|d d ttd d fd }|ddttdd fd }|ddttdd fd }|ddd tid fd }|dd||||||gS)Nr)toolcreate_sdk_mcp_server)MediaFileUploadMediaIoBaseUpload drive_listzList the files in the shared Jarvis Drive folder (name, type, size, last modified). Use for 'what's in the drive/storage', 'list my shared files', 'what did I save there'.c Kd} j|d{}|s tdSg}|D]R}|jd}|rt|dzdnd}|j d|dd |d |d dd d |dd Ttt |ddj |zS7#t$r$}tdt|ddcYd}~Sd}~wwxYww)Nct}dtd}|jj|dddj j dg}|S)Nr1z' in parents and trashed=falser!zmodifiedTime descz)files(id,name,mimeType,size,modifiedTime))r2r3orderByr/r4)r,rr4r7r5r )r8r2r4s rworkz.build_server..drive_list..workbs^*CJL>!?@AIIK$$G-@B%')CC, Lrz!The shared Drive folder is empty.sizeiKBu—z- namez (z, modifiedTime z) [id id]z file(s) in Jarvis Drive:  zdrive_list failed: ) to_threadrr intappendlenjoinr6r))argsrGr4linesr<szeasyncios rrCz build_server..drive_list]s    ?!++D11E@AAEUU6]-/B b)U r!F)Ct2a6G6L5MN$$%dG9A/0 CJ<'BC99U+,- -2 ?.s1vds|n=> > ?sOC"B2B0B2C"BB2/C"0B22 C;CCC"CC" drive_readzRead a text file's contents from the shared folder. `file` = the file name or its id. Works for plain text / csv / json / markdown and Google Docs (exported as text).filec&Kfd} j|d{\}}| tdS|ddt|dkDrdndz}t|dd|S7?#t$r$}tdt |dd cYd}~Sd}~wwxYww) Nct}t|jdd}|sy|jdd}|jdr3|j j |ddj }n1|j j|d j }|t|tr|jd d fSt|fS) Nr\)Nz not foundmimeTypezapplication/vnd.google-appsrM text/plain)r.r`r.utf-8ignore) r,r=r startswithr4exportr5 get_media isinstancebytesdecoder))r8r<mimedatarVs rrGz.build_server..drive_read..work~s*Cdhhvr23A(55R(D<=yy{))T7\*;;B79yy{,,AdG,<DDF$T51{{7H5B B7:4yB Brz.drive_read: no such file in the shared folder.ipu …[truncated]r_rJz: zdrive_read failed: rP)rQrrTr6r))rVrGr<bodysniprYrZs` rr[z build_server..drive_readys B ?#--d33GAtyMNN;s4y47G"3RPDAfI;c$01 1 4  ?.s1vds|n=> > ?sNBA!AA!B*A!BA!! B*B BB BB drive_writeaCreate (or overwrite) a text file in the shared folder from content you provide. `name` = the file name (e.g. 'riyadh-leads.csv'), `content` = the full text to store. Use to hand a file to another AI/device, e.g. write a CRM or notes for Jarvis to pick up later.)rJrcKt|jddjt|jdds tdSfd} j |d{\}}t|dd|dS7#t $r$}td t|dd cYd}~Sd}~wwxYww) NrJr_rz%drive_write failed: need a file name.ct}tjdxsd}tjj d|d}t |}|rM|jdk(r9|jj|d|jd |dfStgd }|jj||d j}d |dfS) NrrarcFmimetype resumablerJrMr. media_bodyupdatedrJr0rmrvr/created) r, mimetypes guess_typeioBytesIOencoder=r r4updater5rcreate) r8rkmediaexistingmetaoutrBrrJs rrGz/build_server..drive_write..works*C''-a0@LD% 7>>'23d!E T*HHLL0D8 ""(4.U"%%,WY (4.00 jl^.drive_writes488FB'(..0dhhy"-.@A A ( @%//55ID#D64&(=cU"EF F6 @/At ~>? ? @sHACB3B4BCB B>B93B>4C9B>>C drive_uploadzUpload a LOCAL file (by path on this Mac) into the shared folder. `local_path` = absolute path; `name` = optional name to store it as (defaults to the file's own name). Use for 'put this file on the drive', 'upload X to storage'.) local_pathrJcdKtjjt|j ddj rtjj stddSt|j ddj xstjjfd} j|d{\}}t|dd|d S7#t$r$}td t|dd cYd}~Sd}~wwxYww) Nrr_z drive_upload failed: no file at .rJct}tjdxsd}|d}t|}|rM|j dk(r9|j j |d|jd|dfStgd }|j j||d j}d |dfS) Nrzapplication/octet-streamTrrrJrMrurwrxryuploaded) r,r{r|r=r r4rr5rr) r8rkrrrrrAlprJs rrGz0build_server..drive_upload..works*C''+A.L2LD#BFET*HHLL0D8 ""(4.U"%%,WY (4.00 jl^.drive_uploadsWW  DHH\2$> ? E E G H+;B6CD D488FB'(..0HBGG4D4DR4H ) A%//55ID#D64&(=cU"EF F6 A0Q?@ @ AsHCD0D"C>#D=D0>D D- D("D-#D0(D--D0drive_downloadzDownload a file from the shared folder to a LOCAL path on this Mac. `file` = name or id; `local_path` = where to save it. Use to pull a shared file down so a local tool (or Jarvis) can work with it.)r\rcKtjjtj ddj s t dSfd} j|d{}|s t dSt d|ddS7##t$r$}t d t|dd cYd}~Sd}~wwxYww) Nrr_z)drive_download failed: need a local_path.c@ddlm}t}t|j dd}|syt j d}|||jj|d}d}|s|j\}}|s|j|d S) Nr)MediaIoBaseDownloadr\r_wbrMrbFrJ) googleapiclient.httprr,r=r r}FileIOr4rg next_chunkclose) rr8r<bufdldone_rVdests rrGz2build_server..drive_download..works @*Cdhhvr23A))D$'C$S#))+*?*?qw*?*OPBD--/4 IIKV9 rz2drive_download: no such file in the shared folder.z Downloaded u → rzdrive_download failed: rP) r rrr)r rrrQr6)rVrGrJrYrrZs` @rrz$build_server..drive_downloads ww!!#dhh|R&@"A"G"G"IJDE E  C **400DQRR;tfE$q9: :1 C23q6$3<.AB B CsTAC  B4B5BC BC B C#C<C=C CC  drive_deletezRemove (move to trash) a file from the shared folder. `file` = name or id. Use for 'delete X from the drive', 'remove that file'.cKfd} j|d{}|s tdStd|dS7 #t$r$}tdt|ddcYd}~Sd}~wwxYww)Nct}t|jdd}|sy|jj |dddij |dS)Nr\r_rMtrashedT)r.rmrJ)r,r=r r4rr5)r8r<rVs rrGz0build_server..drive_delete..works\*Cdhhvr23A IIK  agY4E  F N N PV9 rz0drive_delete: no such file in the shared folder.zDeleted z from Jarvis Drive.zdrive_delete failed: rP)rQrr6r))rVrGrJrYrZs` rrz"build_server..drive_deletes~   A **400DOPP8D6)<=> >1 A0Q?@ @ AsMA2AAAA2AA2A A/ A*$A/%A2*A//A2r!z1.0.0)rJversiontools)rVdictreturnr)rZclaude_agent_sdkr?r@rrArBr)) r?r@rCr[rorrrrArBrZs @@@r build_serverrXs.<G , 4579? 9?0 , 028#@? @?. - K3 ' ) @ ) @6 . +c * , A , A2  Kc * , C , C4 . J 3-A A ! g:{L|- r)rbool)rr)rr)rr))r9r)rz dict | None)__doc__ __future__rr}r{r pathlibr__file__resolveparent_HEREr rr*rrrr,r=rrrrrsy,#   X ''.. ( (00  3 38,J'0sr