ec2-github-runner/src/index.js at 00d75de459a05d394e5d19bfd3c04ab2374193e4 · machulav/ec2-github-runner · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
const aws = require('./aws');
const gh = require('./gh');
const config = require('./config');
const core = require('@actions/core');

function setOutput(label, ec2InstanceId) {
  core.setOutput('label', label);
  core.setOutput('ec2-instance-id', ec2InstanceId);
}

async function start() {
  const label = config.generateUniqueLabel();
  const githubRegistrationToken = await gh.getRegistrationToken();
  const ec2InstanceId = await aws.startEc2Instance(label, githubRegistrationToken);
  setOutput(label, ec2InstanceId);
  await aws.waitForInstanceRunning(ec2InstanceId);
  await gh.waitForRunnerRegistered(label);
}

async function stop() {
  await aws.terminateEc2Instance();
  await gh.removeRunner();
}

async function resume() {
  const label = config.input.label;
  const githubRegistrationToken = await gh.getRegistrationToken();
  const ec2InstanceId = config.input.ec2InstanceId;
  await aws.waitForInstanceStopped(ec2InstanceId);
  await aws.resumeEc2Instance(ec2InstanceId);
  setOutput(label, ec2InstanceId);
  await aws.waitForInstanceRunning(ec2InstanceId);
  await gh.waitForRunnerRegistered(label);
}

async function pause() {
  await aws.stopEc2Instance();
}

(async function () {
  try {
    if (config.input.mode === 'start') {
      await start();
    } else if (config.input.mode === 'stop') {
      await stop();
    } else if (config.input.mode === 'resume') {
      await resume();
    } else if (config.input.mode === 'pause') {
      await pause();
    }
  } catch (error) {
    core.error(error);
    core.setFailed(error.message);
  }
})();