o Á¿iÝIã@s¦UddlZddlZddlZddlZddlmZddlmZddl m Z e  e ¡Z Gdd„dƒZejejeefgeeejeffZgZeeeefed<ded ejegeffd d „Zd ejeefd eeejeffd d„Zedƒd ejeefd eeejeffdd„ƒZedƒd ejeefd eeejeffdd„ƒZedƒd ejeefd eeejeffdd„ƒZedƒd ejeefd eeejeffdd„ƒZedƒd ejeefd eeejeffdd„ƒZedƒd ejeefd eeejeffdd „ƒZ ed!ƒd ejeefd eeejeffd"d#„ƒZ!ed$ƒd ejeefd eeejeffd%d&„ƒZ"ed'ƒd ejeefd eeejeffd(d)„ƒZ#ed*ƒd ejeefd eeejeffd+d,„ƒZ$ed-ƒd ejeefd eeejeffd.d/„ƒZ%ed0ƒd ejeefd eeejeffd1d2„ƒZ&ed3ƒd ejeefd eeejeffd4d5„ƒZ'ed6ƒd ejeefd eeejeffd7d8„ƒZ(dS)9éN)Úgithub_actions)Úgit)ÚGitTagc@s@eZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdS)ÚCITagz ci.stage.namez ci.job.idz ci.job.namez ci.job.urlzci.pipeline.idzci.pipeline.namezci.pipeline.numberzci.pipeline.urlzci.provider.namez ci.node.namezci.node.labelszci.workspace_pathz_dd.ci.env_varsN)Ú__name__Ú __module__Ú __qualname__Ú STAGE_NAMEÚJOB_IDÚJOB_NAMEÚJOB_URLÚ PIPELINE_IDÚ PIPELINE_NAMEÚPIPELINE_NUMBERÚ PIPELINE_URLÚ PROVIDER_NAMEÚ NODE_NAMEÚ NODE_LABELSÚWORKSPACE_PATHÚ _CI_ENV_VARS©rrúO/home/ubuntu/.local/lib/python3.10/site-packages/ddtrace/testing/internal/ci.pyrsrÚ PROVIDERSÚkeyÚreturncsdtdtf‡fdd„ }|S)z¼ Register a handler to extract tags from the environment for a given CI provider. The handler will be used if the environment contains an environment variable named by `key`. Úfrcst ˆ|f¡|S)N)rÚappend)r©rrrÚ decoratorBsz$register_provider..decorator)ÚTProviderFunction)rrrrrÚregister_provider;sr ÚenvcCs&tD]\}}||vr||ƒSqiS)z5Extract tags from CI provider environment variables.)r)r!rÚextractrrrÚ get_ci_tagsIs  ÿr#ÚAPPVEYORcCsd | d¡| d¡¡}| d¡dkr0d | d¡¡}| d¡}| d¡p)| d ¡}| d ¡}nd }}}}| d ¡}|rL| d ¡}|rL|d|7}tjdtj|tj|tj| d¡tj| d¡tj | d¡tj | d¡tj |tj |tj |tj|tj|tj| d¡tj| d¡iS)z&Extract CI tags from Appveyor environ.z.https://ci.appveyor.com/project/{0}/builds/{1}ÚAPPVEYOR_REPO_NAMEÚAPPVEYOR_BUILD_IDÚAPPVEYOR_REPO_PROVIDERÚgithubúhttps://github.com/{0}.gitÚAPPVEYOR_REPO_COMMITÚ&APPVEYOR_PULL_REQUEST_HEAD_REPO_BRANCHÚAPPVEYOR_REPO_BRANCHÚAPPVEYOR_REPO_TAG_NAMENÚAPPVEYOR_REPO_COMMIT_MESSAGEÚ%APPVEYOR_REPO_COMMIT_MESSAGE_EXTENDEDÚ ÚappveyorÚAPPVEYOR_BUILD_FOLDERÚAPPVEYOR_BUILD_NUMBERÚAPPVEYOR_REPO_COMMIT_AUTHORÚ!APPVEYOR_REPO_COMMIT_AUTHOR_EMAIL)ÚformatÚgetrrrÚREPOSITORY_URLÚ COMMIT_SHArr rrrr ÚBRANCHÚTAGÚCOMMIT_MESSAGEÚCOMMIT_AUTHOR_NAMEÚCOMMIT_AUTHOR_EMAIL)r!ÚurlÚ repositoryÚcommitÚbranchÚtagÚcommit_messageÚextendedrrrÚextract_appveyorRs:ÿ           òrFÚTF_BUILDcCsŽ| d¡r0| d¡r0| d¡r0d | d¡| d¡| d¡¡}|}|d | d¡| d¡¡}nd}}itjd “tj| d ¡“tj| d¡“tj| d ¡“tj| d¡“tj|“tj |“t j | d ¡ph| d ¡“t j | d¡pt| d¡“t j | d¡p…| d¡p…| d¡“t j| d¡“t j| d¡“t j| d¡“tj| d¡“tj| d¡“tj| d¡“tjtj| d¡| d¡| d¡dœdd“S)z-Extract CI tags from Azure pipelines environ.ÚSYSTEM_TEAMFOUNDATIONSERVERURIÚSYSTEM_TEAMPROJECTIDÚ BUILD_BUILDIDz!{0}{1}/_build/results?buildId={2}z&view=logs&j={0}&t={1}Ú SYSTEM_JOBIDÚSYSTEM_TASKINSTANCEIDNÚazurepipelinesÚBUILD_SOURCESDIRECTORYÚBUILD_DEFINITIONNAMEÚ&SYSTEM_PULLREQUEST_SOURCEREPOSITORYURIÚBUILD_REPOSITORY_URIÚ!SYSTEM_PULLREQUEST_SOURCECOMMITIDÚBUILD_SOURCEVERSIONÚSYSTEM_PULLREQUEST_SOURCEBRANCHÚBUILD_SOURCEBRANCHÚBUILD_SOURCEBRANCHNAMEÚBUILD_SOURCEVERSIONMESSAGEÚBUILD_REQUESTEDFORIDÚBUILD_REQUESTEDFOREMAILÚSYSTEM_STAGEDISPLAYNAMEÚSYSTEM_JOBDISPLAYNAME)rIrJrK©ú,ú:©Ú separators)r7r6rrrr rrrr rr8r9r:r<r=r>r r r rÚjsonÚdumps)r!Úbase_urlÚ pipeline_urlÚjob_urlrrrÚextract_azure_pipelinesxsjÿÿÿ þ ý ü ûúùø ÷ ÿô ó ò ñ ð ï îýúírfÚBITBUCKET_COMMITcCs¤d | d¡| d¡¡}tj| d¡tj| d¡tj| d¡p#| d¡tj| d¡tj|tj | d d ¡  d ¡p8d tj | d¡tj | d¡tj |tjd tj| d¡i S)z'Extract CI tags from Bitbucket environ.zÚCOMMIT_COMMITTER_NAMEÚCOMMIT_COMMITTER_EMAILrrarbrr)r!Únode_label_listÚ buildkite_agent_meta_data_prefixÚ env_variablerÚvaluerrrÚextract_buildkite¹sh  € ÿ þ ý ü û ú ù ø ÷ ö õ ô ó ò ñ ð ïþû ærÚCIRCLECIcCs¶tj| d¡tj| d¡tj| d¡tj| d¡tj| d¡tj| d¡tj | d¡tj d  | d¡¡tj | d ¡tj | d ¡tjd tj| d ¡tjtj| d¡| d¡d œddi S)z&Extract CI tags from CircleCI environ.Ú CIRCLE_BRANCHÚ CIRCLE_SHA1ÚCIRCLE_REPOSITORY_URLÚ CIRCLE_TAGÚCIRCLE_WORKFLOW_IDÚCIRCLE_PROJECT_REPONAMEÚCIRCLE_BUILD_NUMz0https://app.circleci.com/pipelines/workflows/{0}ÚCIRCLE_BUILD_URLÚ CIRCLE_JOBÚcircleciÚCIRCLE_WORKING_DIRECTORY)r–r˜r\r_)rr:r7r9r8r;rr rrrr6r r rrrrarb©r!rrrÚextract_circle_ciâs&          þûóržÚ CF_BUILD_IDcCs^| d¡}tj| d¡tj|tj| d¡tj| d¡tj| d¡tjdtj t j d|iddiS) z'Extract CI tags from Codefresh environ.rŸÚ CF_BRANCHÚCF_PIPELINE_NAMEÚ CF_BUILD_URLÚ CF_STEP_NAMEÚ codefreshr\r_) r7rr:rr rrr rrrarb)r!Úbuild_idrrrÚextract_codefreshüs     þùr¦Ú GITHUB_SHAcCs t |¡S)z0Extract CI tags from Github Actions environment.)rÚextract_github_actionsrrrrr¨s r¨Ú GITLAB_CIc Cs`| d¡}d}d}|r| d¡ d¡\}}| d¡}itj| d¡“tj| d¡“tj| d¡“tj| d ¡“tj | d ¡“tj | d ¡“tj | d ¡“tj | d ¡“tj | d¡“tj| d¡“tj| d¡“tj| d¡“tjd“tj| d¡“tj| d¡“tj|“tj|“tj|tjtj| d¡| d¡| d ¡dœddtj| d¡tj| d¡i¥S)z$Extract CI tags from Gitlab environ.ÚCI_COMMIT_AUTHORNz> z <ÚCI_COMMIT_TIMESTAMPÚCI_COMMIT_REF_NAMEÚ CI_COMMIT_SHAÚCI_REPOSITORY_URLÚ CI_COMMIT_TAGÚ CI_JOB_STAGEÚ CI_JOB_IDÚ CI_JOB_NAMEÚ CI_JOB_URLÚCI_PIPELINE_IDÚCI_PROJECT_PATHÚCI_PIPELINE_IIDÚCI_PIPELINE_URLÚgitlabÚCI_PROJECT_DIRÚCI_COMMIT_MESSAGEÚCI_PROJECT_URL)r»r´r±r\r_ÚCI_RUNNER_TAGSÚ CI_RUNNER_ID)r7rrÚsplitrr:r9r8r;rr r r r r rrrrrr<r=r>ÚCOMMIT_AUTHOR_DATErrarbrr)r!ÚauthorÚ author_nameÚ author_emailÚcommit_timestamprrrÚextract_gitlabsh   ÿ þ ý ü û ú ù ø ÷ ö õ ô ó ò ñðïýú  ärÄÚ JENKINS_URLcCs| dd¡}| d¡}|r|rt d t |¡¡d|¡}|r+d dd„| d¡Dƒ¡}g}| d¡}|r8| ¡}tj | d¡tj | d ¡tj | d | d ¡¡t j | d ¡t j|t j| d ¡t j| d¡t jdt j| d¡t jtjd| d¡iddt jtj|ddt j| d¡i S)z%Extract CI tags from Jenkins environ.Ú GIT_BRANCHror z/{0}ú/css |] }|rd|vr|VqdS)ú=Nr)Ú.0ÚvrrrÚ Fs€z"extract_jenkins..rÚ GIT_COMMITÚGIT_URLÚ GIT_URL_1Ú BUILD_TAGÚ BUILD_NUMBERÚ BUILD_URLÚjenkinsÚ WORKSPACEÚDD_CUSTOM_TRACE_IDr\r_r)r7ÚreÚsubr6rÚ normalize_refÚjoinr¾rr:r9r8rr rrrrrrrarbrr)r!rBÚnameÚnode_labels_listÚnode_labels_envrrrÚextract_jenkins>s6          ÿü ïrÜÚTEAMCITY_VERSIONcCs"tj| d¡tj| d¡tjdiS)z&Extract CI tags from Teamcity environ.rÑÚTEAMCITY_BUILDCONF_NAMEÚteamcity)rr r7r rrrrrÚextract_teamcity`s  ýràÚTRAVIScCsžtj| d¡p | d¡tj| d¡tjd | d¡¡tj| d¡tj| d¡tj | d¡tj | d¡tj | d ¡tj | d ¡tj d tj| d ¡tj| d ¡i S)z$Extract CI tags from Travis environ.ÚTRAVIS_PULL_REQUEST_BRANCHÚ TRAVIS_BRANCHÚ TRAVIS_COMMITr)ÚTRAVIS_REPO_SLUGÚ TRAVIS_TAGÚTRAVIS_JOB_WEB_URLÚTRAVIS_BUILD_IDÚTRAVIS_BUILD_NUMBERÚTRAVIS_BUILD_WEB_URLÚtravisciÚTRAVIS_BUILD_DIRÚTRAVIS_COMMIT_MESSAGE)rr:r7r9r8r6r;rr r rrrrrr<rrrrÚextract_travisjs         ôrîÚBITRISE_BUILD_SLUGc Cs| d¡p | d¡}| d¡p| d¡}| d¡r| d¡}n| d¡s)| d¡r6d | d¡| d¡¡}nd }tjd tj| d ¡tj| d ¡tj| d ¡tj| d¡tj| d¡t j | d¡t j |t j |t j | d¡t j|t j| d¡t j| d¡t j| d¡t j| d¡iS)z%Extract CI tags from Bitrise environ.ÚBITRISE_GIT_COMMITÚGIT_CLONE_COMMIT_HASHÚBITRISEIO_GIT_BRANCH_DESTÚBITRISE_GIT_BRANCHÚBITRISE_GIT_MESSAGEÚ GIT_CLONE_COMMIT_MESSAGE_SUBJECTÚGIT_CLONE_COMMIT_MESSAGE_BODYz{0}: {1}NÚbitriserïÚBITRISE_TRIGGERED_WORKFLOW_IDÚBITRISE_BUILD_NUMBERÚBITRISE_BUILD_URLÚBITRISE_SOURCE_DIRÚGIT_REPOSITORY_URLÚBITRISE_GIT_TAGÚGIT_CLONE_COMMIT_AUTHOR_NAMEÚGIT_CLONE_COMMIT_AUTHOR_EMAILÚGIT_CLONE_COMMIT_COMMITER_NAME)r7r6rrr rrrrrr8r9r:r;r<r=r>rŠr‹)r!rArBÚmessagerrrÚextract_bitrise}s2  ÿ           ñrÚBUDDYcCsœtjdtjd | d¡| d¡¡tj| d¡tj| d¡tj| d¡tj | d¡tj | d¡tj | d ¡tj | d ¡tj | d ¡tj| d ¡tj| d ¡i S)z#Extract CI tags from Buddy environ.Úbuddyz{0}/{1}ÚBUDDY_PIPELINE_IDÚBUDDY_EXECUTION_IDÚBUDDY_PIPELINE_NAMEÚBUDDY_EXECUTION_URLÚ BUDDY_SCM_URLÚBUDDY_EXECUTION_REVISIONÚBUDDY_EXECUTION_BRANCHÚBUDDY_EXECUTION_TAGÚ BUDDY_EXECUTION_REVISION_MESSAGEÚ'BUDDY_EXECUTION_REVISION_COMMITTER_NAMEÚ(BUDDY_EXECUTION_REVISION_COMMITTER_EMAIL)rrr r6r7rrrrr8r9r:r;r<rŠr‹rrrrÚ extract_buddyžs          ôrÚCODEBUILD_INITIATORc Csni}d|vr5| d¡}|r5| d¡r5| tjdtj| d¡tjtj| d¡| d¡| d¡dœdd i¡|S) z,Extract CI tags from codebuild environments.rÚ codepipelineÚawscodepipelineÚDD_PIPELINE_EXECUTION_IDÚCODEBUILD_BUILD_ARNÚDD_ACTION_EXECUTION_ID)rrrr\r_) r7r‡Úupdaterrr rrarb)r!ÚtagsÚcodebuild_initiatorrrrÚextract_codebuild±s"  ýúýÿr))raÚloggingrÕÚtypingÚtÚddtrace.ext.cirÚddtrace.testing.internalrÚddtrace.testing.internal.gitrÚ getLoggerrÚlogrÚCallableÚMutableMappingÚstrÚdictÚOptionalrrÚlistÚtupleÚ__annotations__r r#rFrfrsrržr¦r¨rÄrÜràrîrrrrrrrÚsR     ()* ,%,+,,(,,,,),!, ,, ,0