tPjE*dZddlZddlZddlmZddlmZmZddl m Z m Z ddl m Z GddeZGd d eZGd d eZGd deZGddeZdedefdZGddZy)zN BitBucket API client for fetching .prompt files from BitBucket repositories. N)Mapping)Final TypedDict) NotRequiredReadOnly) HTTPHandlerc:eZdZUeeeed<eeeed<y)BitBucketSrcEntrypathtypeN__name__ __module__ __qualname__rrstr__annotations__t/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/integrations/bitbucket/bitbucket_client.pyr r " ;s# $$ ;s# $$rr c*eZdZUeeeeed<y)BitBucketSrcListingvaluesN)rrrrrlistr rrrrrrs [&7!89 ::rrc:eZdZUeeeed<eeeed<y)BitBucketBranchnamer Nr rrrrrrrrc*eZdZUeeeeed<y)BitBucketBranchListingrN)rrrrrrrrrrrrrs [o!67 88rrcPeZdZUeedzed<eedzed<eedzed<y)BitBucketFileMetadataN content_typecontent_length last_modified)rrrrrrrrrr!r!!s/3:&&S4Z((C$J''rr! file_pathreturncd|vsd|vrtd|d|jd}|D]}|dk(s td|ddjd|DS) z7Reject path traversal and URL-encode each path segment.#?zInvalid file path z!: contains URL special characters/z..z: path traversal detectedc3^K|]%}tjj|d'yw))safeN)urllibparsequote).0parts r z&_sanitize_file_path../s$H$FLL&&t"&5Hs+-) ValueErrorsplitjoin)r%partsr2s r_sanitize_file_pathr8'sz i3)+-i]:[\]]??3'EZ 4<1)>WXY YZ 88H%H HHrceZdZdZdeeeffdZdededzfdZdded ede efd Z deeeffd Z de fd Z de efd ZdededzfdZdZy)BitBucketClienta Client for interacting with BitBucket API to fetch .prompt files. Supports: - Authentication with access tokens - Fetching file contents from repositories - Team-based access control through BitBucket permissions - Branch-specific file fetching configc |jd|_|jd|_|jd|_|jdd|_|jdd|_|jdd |_|jd |_t|j|j|jgs td d d d |_ |j dk(rk|jr_|jd|j}tj|jj}d||jd<nd|j|jd<t|_y)a8 Initialize the BitBucket client. Args: config: Dictionary containing: - workspace: BitBucket workspace name - repository: Repository name - access_token: BitBucket access token (or app password) - branch: Branch to fetch from (default: main) - base_url: Custom BitBucket API base URL (optional) - auth_method: Authentication method ('token' or 'basic', default: 'token') - username: Username for basic auth (optional) workspace repository access_tokenbranchmainr,zhttps://api.bitbucket.org/2.0 auth_methodtokenusernamez4workspace, repository, and access_token are requiredzapplication/json)Acceptz Content-Typebasic:zBasic AuthorizationzBearer N)getr=r>r?r@base_urlrBrDallr4headersbase64 b64encodeencodedecoder http_handler)selfr; credentialsencoded_credentialss r__init__zBitBucketClient.__init__=sL K0 **\2"JJ~6jj62  2'FG !::mW= :. DNNDOOT5F5FGHST T).    w &4==$(MM?!D4E4E3F!GK)/)9)9+:L:L:N)O)V)V)X .45H4I,JDLL )/6d6G6G5H,IDLL )(Mrr%r&Nc t|}|jd|jd|jd|jd| } |j j ||j}|j|jj ddjdr |jS tj|jjdS#t$r|jcYSwxYw#t$r}t!|d rt!|j"d r|j"j$d k(rYd }~y |j"j$d k(r)td|d|jd|jd|j"j$dk(r tdtd|d|td|d|d }~wwxYw)z Fetch the content of a file from the BitBucket repository. Args: file_path: Path to the file in the repository Returns: File content as string, or None if file not found /repositories/r*/src/rL content-typer,ztext/zutf-8response status_codeNzAccess denied to file '3'. Check your BitBucket permissions for workspace '' and repository ''.IAuthentication failed. Check your BitBucket access token and permissions.zFailed to fetch file '': zError fetching file ')r8rJr=r>r@rQrIrLraise_for_status startswithtextrM b64decodecontentrP Exceptionhasattrr[r\)rRr% safe_pathurlr[es rget_file_contentz BitBucketClient.get_file_contenths/y9  nT^^4DAdooEVV[\`\g\g[hhijsitu K"//33C3NH  % % '##NB7BB7K}}$)!++H,<,<=DDWMM )#==() Kq*%'!**m*L::))S0ZZ++s2#1)A-C=4-C""C:7C=9C::C== G;GBGGdirectory_pathfile_extensionc |r t|nd}|jd|jd|jd|jd| } |j j ||j}|j|j}g}|j dgD]L}|j ddk(s|j d d} | j|s<|j| N|S#t$r} t| d rt| jd r| jjd k(rgcYd } ~ S| jjdk(r)td|d|jd|jd| jjdk(r tdtd|d| td|d| d } ~ wwxYw)a List files in a directory with a specific extension. Args: directory_path: Directory path in the repository (empty for root) file_extension: File extension to filter by (default: .prompt) Returns: List of file paths r,rWr*rXrYrr commit_filer r[r\r]Nr^zAccess denied to directory 'r_r`rarbrczFailed to list files in 'rdzError listing files in ')r8rJr=r>r@rQrIrLrejsonendswithappendrjrkr[r\) rRrprqsafe_dirrmr[datafilesitemr%rns r list_fileszBitBucketClient.list_filessBP-n=UW nT^^4DAdooEVV[\`\g\g[hhijrist S"//33C3NH  % % '/7}}D&(E2. 088F#}4 $ 4I )).9 Y/  0 L Sq*%'!**m*L::))S0IZZ++s2#6~6FFyz~{I{IzJJ\]a]l]l\mmopZZ++s2#$opp#&??OsSTRU$VWW":>:J#aS QRR Ss2 A2C5<#C5 C55 G >rQrIrLrertrj)rRrmr[rns rget_repository_infoz#BitBucketClient.get_repository_infos nT^^4DAdooEVW C"//33C3NH  % % '==? " C=aSAB B CsAA11 B :BB cD |jy#t$rYywxYw)z Test the connection to the BitBucket repository. Returns: True if connection is successful, False otherwise TF)r~rjrRs rtest_connectionzBitBucketClient.test_connections(   $ $ &  s  cF|jd|jd|jd} |jj ||j }|j |j}|j dgS#t$r}td|d}~wwxYw)z Get list of branches in the repository. Returns: List of branch information dictionaries rWr*z/refs/branchesrYrzFailed to get branches: Nr})rRrmr[rxrns r get_brancheszBitBucketClient.get_branchess nT^^4DAdooEVVde <"//33C3NH  % % '2:--/D88Hb) ) <6qc:; ; r@rLcopyrQrIrerjrkr[r\)rRr%rlrmrLr[rns rget_file_metadataz!BitBucketClient.get_file_metadatasR/y9  nT^^4DAdooEVV[\`\g\g[hhijsitu W!\\..0G*GG "//33C3IH  % % '!) 0 0 4 4^ D"*"2"2"6"67G"H!)!1!1!5!5o!F   Wq*%'!**m*L::))S0"CI;cRSQT UVV"CI;cRSQT UVV WsBC"" E+;E +"E  EcRt|dr|jjyy)z)Close the HTTP handler to free resources.rQN)rkrQclosers rrzBitBucketClient.closes# 4 (    # # % )r)r,z.prompt)rrr__doc__rrobjectrUrorr{r~boolrrrr!rrrrrr:r:2s)*wsF{3)*V)K#)K#*)KV+S+S3+SW[\_W`+SZCWS&[%9C  rs|##3>% % ;);%i% 9Y9(I( I3I3Ic&c&r