Add option for custom root volume storage size

Author: xJonathanLEI

README.md

@@ -206,6 +206,7 @@ Now you're ready to go!
 | `aws-resource-tags`                                                                                                                                                          | Optional. Used only with the `start` mode. | Specifies tags to add to the EC2 instance and any attached storage. <br><br> This field is a stringified JSON array of tag objects, each containing a `Key` and `Value` field (see example below). <br><br> Setting this requires additional AWS permissions for the role launching the instance (see above).                         |
 | `runner-home-dir`                                                                                                                                                              | Optional. Used only with the `start` mode. | Specifies a directory where pre-installed actions-runner software and scripts are located.<br><br> |
 | `pre-runner-script`                                                                                                                                                              | Optional. Used only with the `start` mode. | Specifies bash commands to run before the runner starts.  It's useful for installing dependencies with apt-get, yum, dnf, etc. For example:<pre>          - name: Start EC2 runner<br>            with:<br>              mode: start<br>              ...<br>              pre-runner-script: \|<br>                 sudo yum update -y && \ <br>                 sudo yum install docker git libicu -y<br>                 sudo systemctl enable docker</pre>
+| `storage-size`                                                                                                                                                               | Optional. Used only with the `start` mode. | Specifies the root volume size in GB.                                                                                                                                                                                                                                                                                                 |
 <br><br> |
 
 ### Environment variables

action.yml

@@ -69,6 +69,10 @@ inputs:
     description: >-
       Specifies bash commands to run before the runner starts. It's useful for installing dependencies with apt-get, yum, dnf, etc.
     required: false
+  storage-size:
+    description: >-
+      Specifies the root volume size in GB.
+    required: false
 
 outputs:
   label:

dist/index.js

@@ -62837,6 +62837,18 @@ async function startEc2Instance(label, githubRegistrationToken) {
 
   const userData = buildUserDataScript(githubRegistrationToken, label);
 
+  const blockDeviceMappings = config.input.storageSize
+    ? [
+        {
+          DeviceName: '/dev/sda1',
+          Ebs: {
+            DeleteOnTermination: true,
+            VolumeSize: parseInt(config.input.storageSize),
+          },
+        },
+      ]
+    : undefined;
+
   const params = {
     ImageId: config.input.ec2ImageId,
     InstanceType: config.input.ec2InstanceType,
@@ -62847,6 +62859,7 @@ async function startEc2Instance(label, githubRegistrationToken) {
     SecurityGroupIds: [config.input.securityGroupId],
     IamInstanceProfile: { Name: config.input.iamRoleName },
     TagSpecifications: config.tagSpecifications,
+    BlockDeviceMappings: blockDeviceMappings,
   };
 
   try {
@@ -62923,6 +62936,7 @@ class Config {
       iamRoleName: core.getInput('iam-role-name'),
       runnerHomeDir: core.getInput('runner-home-dir'),
       preRunnerScript: core.getInput('pre-runner-script'),
+      storageSize: core.getInput('storage-size'),
     };
 
     const tags = JSON.parse(core.getInput('aws-resource-tags'));

src/aws.js

@@ -37,6 +37,18 @@ async function startEc2Instance(label, githubRegistrationToken) {
 
   const userData = buildUserDataScript(githubRegistrationToken, label);
 
+  const blockDeviceMappings = config.input.storageSize
+    ? [
+        {
+          DeviceName: '/dev/sda1',
+          Ebs: {
+            DeleteOnTermination: true,
+            VolumeSize: parseInt(config.input.storageSize),
+          },
+        },
+      ]
+    : undefined;
+
   const params = {
     ImageId: config.input.ec2ImageId,
     InstanceType: config.input.ec2InstanceType,
@@ -47,6 +59,7 @@ async function startEc2Instance(label, githubRegistrationToken) {
     SecurityGroupIds: [config.input.securityGroupId],
     IamInstanceProfile: { Name: config.input.iamRoleName },
     TagSpecifications: config.tagSpecifications,
+    BlockDeviceMappings: blockDeviceMappings,
   };
 
   try {

src/config.js

@@ -15,6 +15,7 @@ class Config {
       iamRoleName: core.getInput('iam-role-name'),
       runnerHomeDir: core.getInput('runner-home-dir'),
       preRunnerScript: core.getInput('pre-runner-script'),
+      storageSize: core.getInput('storage-size'),
     };
 
     const tags = JSON.parse(core.getInput('aws-resource-tags'));