1+ import fs from 'fs' ;
2+
3+ // Set OWNER and REPOSITORY based on the environment
4+ const OWNER =
5+ process . env . CIRCLE_PROJECT_USERNAME || process . env . OWNER || 'MetaMask' ;
6+ const REPOSITORY =
7+ process . env . CIRCLE_PROJECT_REPONAME ||
8+ process . env . REPOSITORY ||
9+ 'metamask-extension' ;
10+ const CIRCLE_BASE_URL = 'https://circleci.com/api/v2' ;
11+
12+ /**
13+ * Retrieves the pipeline ID for a given branch and optional commit hash.
14+ *
15+ * @param {string } branch - The branch name to fetch the pipeline for.
16+ * @param {string } [headCommitHash] - Optional commit hash to match a specific pipeline.
17+ * @returns {Promise<string> } A promise that resolves to the pipeline ID.
18+ * @throws Will throw an error if no pipeline is found or if the HTTP request fails.
19+ */
20+ export async function getPipelineId ( branch : string , headCommitHash ?: string ) : Promise < string > {
21+ const url = `${ CIRCLE_BASE_URL } ${ OWNER } ${ REPOSITORY } ${ branch } ;
22+ const response = await fetch ( url ) ;
23+ if ( ! response . ok ) {
24+ throw new Error ( `Failed to fetch pipeline data: ${ response . statusText } ) ;
25+ }
26+
27+ const pipelineData = await response . json ( ) ;
28+ const pipelineItem = headCommitHash
29+ ? pipelineData . items . find ( ( item : any ) => item . vcs . revision === headCommitHash )
30+ : pipelineData . items [ 0 ] ;
31+ if ( ! pipelineItem ) {
32+ throw new Error ( 'Pipeline ID not found' ) ;
33+ }
34+ console . log ( 'pipelineId:' , pipelineItem . id ) ;
35+
36+ return pipelineItem . id ;
37+ }
38+
39+ /**
40+ * Retrieves the workflow ID for a given pipeline ID.
41+ *
42+ * @param {string } pipelineId - The ID of the pipeline to fetch the workflow for.
43+ * @returns {Promise<string> } A promise that resolves to the workflow ID.
44+ * @throws Will throw an error if no workflow is found or if the HTTP request fails.
45+ */
46+ export async function getWorkflowId ( pipelineId : string ) : Promise < string > {
47+ const url = `${ CIRCLE_BASE_URL } ${ pipelineId } ;
48+ const response = await fetch ( url ) ;
49+ if ( ! response . ok ) {
50+ throw new Error ( `Failed to fetch workflow data: ${ response . statusText } ) ;
51+ }
52+ const workflowData = await response . json ( ) ;
53+ const workflowId = workflowData . items [ 0 ] ?. id ;
54+ if ( ! workflowId ) {
55+ throw new Error ( 'Workflow ID not found' ) ;
56+ }
57+ console . log ( 'workflowId:' , workflowId ) ;
58+ return workflowId ;
59+ }
60+
61+ /**
62+ * Retrieves a list of jobs for a given workflow ID.
63+ *
64+ * @param {string } workflowId - The ID of the workflow to fetch jobs for.
65+ * @returns {Promise<any[]> } A promise that resolves to an array of jobs.
66+ * @throws Will throw an error if no jobs are found or if the HTTP request fails.
67+ */
68+ export async function getJobsByWorkflowId ( workflowId : string ) : Promise < any [ ] > {
69+ const url = `${ CIRCLE_BASE_URL } ${ workflowId } ;
70+ const response = await fetch ( url ) ;
71+ if ( ! response . ok ) {
72+ throw new Error ( `Failed to fetch jobs: ${ response . statusText } ) ;
73+ }
74+ const jobs = ( await response . json ( ) ) . items ;
75+ return jobs ;
76+ }
77+
78+ /**
79+ * Retrieves job details for a given workflow ID and optional job name.
80+ *
81+ * @param {string } workflowId - The ID of the workflow to fetch job details for.
82+ * @param {string } [jobName] - Optional job name to match a specific job.
83+ * @returns {Promise<any> } A promise that resolves to the job details.
84+ * @throws Will throw an error if no job details are found or if the HTTP request fails.
85+ */
86+ export async function getJobDetails ( workflowId : string , jobName ?: string ) : Promise < any > {
87+ const jobs = await getJobsByWorkflowId ( workflowId ) ;
88+ const jobDetails = jobName
89+ ? jobs . find ( ( item : any ) => item . name === jobName )
90+ : jobs [ 0 ] ;
91+ if ( ! jobDetails ) {
92+ throw new Error ( 'Job details not found' ) ;
93+ }
94+ return jobDetails ;
95+ }
96+
97+ /**
98+ * Retrieves the artifact URL for a given branch, commit hash, job name, and artifact name.
99+ * @param {string } branch - The branch name.
100+ * @param {string } headCommitHash - The commit hash of the branch.
101+ * @param {string } jobName - The name of the job that produced the artifact.
102+ * @param {string } artifactName - The name of the artifact to retrieve.
103+ * @returns {Promise<string> } A promise that resolves to the artifact URL.
104+ * @throws Will throw an error if the artifact is not found or if any HTTP request fails.
105+ */
106+ export async function getArtifactUrl ( branch : string , headCommitHash : string , jobName : string , artifactName : string ) : Promise < string > {
107+ const pipelineId = await getPipelineId ( branch , headCommitHash ) ;
108+ const workflowId = await getWorkflowId ( pipelineId ) ;
109+ const jobDetails = await getJobDetails ( workflowId , jobName ) ;
110+
111+ const jobNumber = jobDetails . job_number ;
112+ console . log ( 'Job number' , jobNumber ) ;
113+
114+ const artifactResponse = await fetch ( `${ CIRCLE_BASE_URL } ${ OWNER } ${ REPOSITORY } ${ jobNumber } ) ;
115+ const artifactData = await artifactResponse . json ( ) ;
116+ const artifact = artifactData . items . find ( ( item : any ) => item . path . includes ( artifactName ) ) ;
117+
118+ if ( ! artifact ) {
119+ throw new Error ( `Artifact ${ artifactName } ) ;
120+ }
121+
122+ const artifactUrl = artifact . url ;
123+ console . log ( 'Artifact URL:' , artifactUrl ) ; ;
124+
125+ return artifactUrl ;
126+ }
127+
128+ /**
129+ * Downloads an artifact from a given URL and saves it to the specified output path.
130+ * @param {string } artifactUrl - The URL of the artifact to download.
131+ * @param {string } outputFilePath - The path where the artifact should be saved.
132+ * @returns {Promise<void> } A promise that resolves when the download is complete.
133+ * @throws Will throw an error if the download fails or if the file cannot be written.
134+ */
135+ export async function downloadArtifact ( artifactUrl : string , outputFilePath : string ) : Promise < void > {
136+ try {
137+ // Ensure the directory exists
138+ const dir = require ( 'path' ) . dirname ( outputFilePath ) ;
139+ if ( ! fs . existsSync ( dir ) ) {
140+ fs . mkdirSync ( dir , { recursive : true } ) ;
141+ }
142+
143+ console . log ( `Downloading artifact from URL: ${ artifactUrl } ${ outputFilePath } ) ;
144+
145+ // Download the artifact
146+ const artifactDownloadResponse = await fetch ( artifactUrl ) ;
147+ if ( ! artifactDownloadResponse . ok ) {
148+ throw new Error ( `Failed to download artifact: ${ artifactDownloadResponse . statusText } ) ;
149+ }
150+ const artifactArrayBuffer = await artifactDownloadResponse . arrayBuffer ( ) ;
151+ const artifactBuffer = Buffer . from ( artifactArrayBuffer ) ;
152+ fs . writeFileSync ( outputFilePath , artifactBuffer ) ;
153+
154+ if ( ! fs . existsSync ( outputFilePath ) ) {
155+ throw new Error ( `Failed to download artifact to ${ outputFilePath } ) ;
156+ }
157+
158+ console . log ( `Artifact downloaded successfully to ${ outputFilePath } ) ;
159+ } catch ( error ) {
160+ console . error ( `Error during artifact download: ${ error . message } ) ;
161+ throw error ;
162+ }
163+ }
164+
165+ /**
166+ * Reads the content of a file.
167+ * @param filePath - The path to the file.
168+ * @returns The content of the file.
169+ */
170+ export function readFileContent ( filePath : string ) : string {
171+ if ( ! fs . existsSync ( filePath ) ) {
172+ throw new Error ( `File not found: ${ filePath } ) ;
173+ }
174+
175+ const content = fs . readFileSync ( filePath , 'utf-8' ) . trim ( ) ;
176+ return content ;
177+ }
178+
179+ /**
180+ * Sleep function to pause execution for a specified number of seconds.
181+ * @param seconds - The number of seconds to sleep.
182+ */
183+ export function sleep ( seconds : number ) {
184+ return new Promise ( resolve => setTimeout ( resolve , seconds * 1000 ) ) ;
185+ }
0 commit comments