76 lines
2.5 KiB
TypeScript
76 lines
2.5 KiB
TypeScript
/**
|
|
* Copyright (c) Microsoft Corporation.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the 'License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
import type { GridAgentLaunchOptions, GridFactory } from './gridServer';
|
|
import https from 'https';
|
|
import { debug } from '../utilsBundle';
|
|
|
|
const repoName = process.env.GITHUB_AGENT_REPO;
|
|
if (!repoName)
|
|
throw new Error('GITHUB_AGENT_REPO is not specified.');
|
|
|
|
const repoAccessToken = process.env.GITHUB_AGENT_REPO_ACCESS_TOKEN;
|
|
if (!repoAccessToken)
|
|
throw new Error('GITHUB_AGENT_REPO_ACCESS_TOKEN is not specified.');
|
|
|
|
const log = debug(`pw:grid:server`);
|
|
|
|
const githubFactory: GridFactory = {
|
|
name: 'Agents hosted on Github',
|
|
// Standard VM is 3-core on mac and 2-core on win and lin
|
|
capacity: 4,
|
|
launchTimeout: 10 * 60_000,
|
|
retireTimeout: 1 * 60 * 60_000,
|
|
statusUrl: (runId: string) => {
|
|
return `https://github.com/${repoName}/actions/runs/${runId}`;
|
|
},
|
|
launch: async (options: GridAgentLaunchOptions) => {
|
|
await createWorkflow(options);
|
|
},
|
|
};
|
|
|
|
async function createWorkflow(inputs: GridAgentLaunchOptions): Promise<boolean> {
|
|
if (!['windows', 'linux', 'macos'].includes(inputs.os)) {
|
|
log(`unsupported OS: ${inputs.os}`);
|
|
return false;
|
|
}
|
|
return new Promise(fulfill => {
|
|
log(`triggering workflow ${JSON.stringify(inputs)}`);
|
|
const req = https.request(`https://api.github.com/repos/${repoName}/actions/workflows/agent.yml/dispatches`, {
|
|
method: 'POST',
|
|
headers: {
|
|
'User-Agent': 'request',
|
|
'Accept': 'application/vnd.github.v3+json',
|
|
'Authorization': `token ${repoAccessToken}`,
|
|
}
|
|
}, response => {
|
|
log(`workflow ${inputs.agentId} response: ${response.statusCode} ${response.statusMessage}`);
|
|
const success = !!response.statusCode && 200 <= response.statusCode && response.statusCode < 300;
|
|
fulfill(success);
|
|
});
|
|
req.on('error', e => {
|
|
log(`failed to create workflow ${inputs.agentId}`);
|
|
fulfill(false);
|
|
});
|
|
req.end(JSON.stringify({
|
|
'ref': 'refs/heads/main',
|
|
inputs
|
|
}));
|
|
});
|
|
}
|
|
|
|
export default githubFactory;
|