Update the EBS creation script
This commit is contained in:
parent
001ddd1799
commit
f3ec305c25
@ -1,11 +1,12 @@
|
|||||||
#! /usr/bin/env python
|
#! /usr/bin/env python
|
||||||
|
|
||||||
|
import os
|
||||||
import sys
|
import sys
|
||||||
from charon import deployment
|
|
||||||
from boto.ec2.blockdevicemapping import BlockDeviceMapping, BlockDeviceType
|
|
||||||
import charon.util
|
|
||||||
import time
|
import time
|
||||||
import argparse
|
import argparse
|
||||||
|
import charon.util
|
||||||
|
from charon import deployment
|
||||||
|
from boto.ec2.blockdevicemapping import BlockDeviceMapping, BlockDeviceType
|
||||||
|
|
||||||
parser = argparse.ArgumentParser(description='Create an EBS-backed NixOS AMI')
|
parser = argparse.ArgumentParser(description='Create an EBS-backed NixOS AMI')
|
||||||
parser.add_argument('--region', dest='region', required=True, help='EC2 region')
|
parser.add_argument('--region', dest='region', required=True, help='EC2 region')
|
||||||
@ -18,13 +19,14 @@ instance_type = "cc1.4xlarge" if args.hvm else "m1.small"
|
|||||||
key_name = args.key_name
|
key_name = args.key_name
|
||||||
ebs_size = 8 if args.hvm else 20
|
ebs_size = 8 if args.hvm else 20
|
||||||
|
|
||||||
|
|
||||||
# Start a NixOS machine in the given region.
|
# Start a NixOS machine in the given region.
|
||||||
f = open("ebs-creator-config.nix", "w")
|
f = open("ebs-creator-config.nix", "w")
|
||||||
f.write('''{{
|
f.write('''{{
|
||||||
machine =
|
machine =
|
||||||
{{ pkgs, ... }}:
|
{{ pkgs, ... }}:
|
||||||
{{
|
{{
|
||||||
deployment.ec2.region = "{0}";
|
deployment.ec2.region = "{0}";
|
||||||
deployment.ec2.keyPair = pkgs.lib.mkOverride 10 "{1}";
|
deployment.ec2.keyPair = pkgs.lib.mkOverride 10 "{1}";
|
||||||
deployment.ec2.blockDeviceMapping."/dev/xvdg".size = pkgs.lib.mkOverride 10 {2};
|
deployment.ec2.blockDeviceMapping."/dev/xvdg".size = pkgs.lib.mkOverride 10 {2};
|
||||||
}};
|
}};
|
||||||
@ -32,19 +34,20 @@ f.write('''{{
|
|||||||
'''.format(args.region, key_name, ebs_size))
|
'''.format(args.region, key_name, ebs_size))
|
||||||
f.close()
|
f.close()
|
||||||
|
|
||||||
db = deployment.open_database("./ebs-creator.charon")
|
db = deployment.open_database(deployment.get_default_state_file())
|
||||||
try:
|
try:
|
||||||
depl = deployment.open_deployment(db, "ebs-creator")
|
depl = deployment.open_deployment(db, "ebs-creator")
|
||||||
except Exception:
|
except Exception:
|
||||||
depl = deployment.create_deployment(db)
|
depl = deployment.create_deployment(db)
|
||||||
depl.name = "ebs-creator"
|
depl.name = "ebs-creator"
|
||||||
depl.auto_response = "y"
|
depl.auto_response = "y"
|
||||||
depl.nix_exprs = ["./ebs-creator.nix", "./ebs-creator-config.nix"]
|
depl.nix_exprs = [os.path.abspath("./ebs-creator.nix"), os.path.abspath("./ebs-creator-config.nix")]
|
||||||
if not args.keep: depl.destroy_resources()
|
if not args.keep: depl.destroy_resources()
|
||||||
depl.deploy(allow_reboot=True)
|
depl.deploy(allow_reboot=True)
|
||||||
|
|
||||||
m = depl.machines['machine']
|
m = depl.machines['machine']
|
||||||
|
|
||||||
|
|
||||||
# Do the installation.
|
# Do the installation.
|
||||||
device="/dev/xvdg"
|
device="/dev/xvdg"
|
||||||
if args.hvm:
|
if args.hvm:
|
||||||
@ -58,16 +61,8 @@ m.run_command("mkdir -p /mnt")
|
|||||||
m.run_command("mount {0} /mnt".format(device))
|
m.run_command("mount {0} /mnt".format(device))
|
||||||
m.run_command("touch /mnt/.ebs")
|
m.run_command("touch /mnt/.ebs")
|
||||||
m.run_command("mkdir -p /mnt/etc/nixos")
|
m.run_command("mkdir -p /mnt/etc/nixos")
|
||||||
# Kind of hacky until the nixos channel is updated to systemd
|
m.run_command("nix-channel --add http://nixos.org/channels/nixos-unstable")
|
||||||
#m.run_command("nix-channel --add http://nixos.org/channels/nixos-unstable")
|
m.run_command("nix-channel --update")
|
||||||
#m.run_command("nix-channel --update")
|
|
||||||
m.run_command("mkdir unpack")
|
|
||||||
m.run_command("cd unpack; (curl -L http://hydra.nixos.org/job/nixos/systemd/channel/latest/download | bzcat | tar xv)")
|
|
||||||
m.run_command("mkdir nixos")
|
|
||||||
m.run_command("mv unpack/*/* nixos")
|
|
||||||
m.run_command("mv nixos unpack/*")
|
|
||||||
m.run_command("nix-env -p /nix/var/nix/profiles/per-user/root/channels -i $(nix-store --add unpack/*)")
|
|
||||||
m.run_command("rm -fR unpack")
|
|
||||||
m.run_command("nixos-rebuild switch")
|
m.run_command("nixos-rebuild switch")
|
||||||
version = m.run_command("nixos-version", capture_stdout=True).replace('"', '').rstrip()
|
version = m.run_command("nixos-version", capture_stdout=True).replace('"', '').rstrip()
|
||||||
print >> sys.stderr, "NixOS version is {0}".format(version)
|
print >> sys.stderr, "NixOS version is {0}".format(version)
|
||||||
@ -89,7 +84,7 @@ if args.hvm:
|
|||||||
else:
|
else:
|
||||||
ami_name = "nixos-{0}-x86_64-ebs".format(version)
|
ami_name = "nixos-{0}-x86_64-ebs".format(version)
|
||||||
description = "NixOS {0} (x86_64; EBS root)".format(version)
|
description = "NixOS {0} (x86_64; EBS root)".format(version)
|
||||||
|
|
||||||
|
|
||||||
# Wait for the snapshot to finish.
|
# Wait for the snapshot to finish.
|
||||||
def check():
|
def check():
|
||||||
@ -161,6 +156,9 @@ image.set_launch_permissions(user_ids=[], group_names=["all"])
|
|||||||
|
|
||||||
m._conn.create_tags([ami_id], {'Name': ami_name})
|
m._conn.create_tags([ami_id], {'Name': ami_name})
|
||||||
|
|
||||||
|
time.sleep(5)
|
||||||
|
|
||||||
|
|
||||||
# Do a test deployment to make sure that the AMI works.
|
# Do a test deployment to make sure that the AMI works.
|
||||||
f = open("ebs-test.nix", "w")
|
f = open("ebs-test.nix", "w")
|
||||||
f.write(
|
f.write(
|
||||||
@ -180,7 +178,7 @@ f.close()
|
|||||||
test_depl = deployment.create_deployment(db)
|
test_depl = deployment.create_deployment(db)
|
||||||
test_depl.auto_response = "y"
|
test_depl.auto_response = "y"
|
||||||
test_depl.name = "ebs-creator-test"
|
test_depl.name = "ebs-creator-test"
|
||||||
test_depl.nix_exprs = [ "./ebs-test.nix" ]
|
test_depl.nix_exprs = [os.path.abspath("./ebs-test.nix")]
|
||||||
test_depl.deploy(create_only=True)
|
test_depl.deploy(create_only=True)
|
||||||
test_depl.machines['machine'].run_command("nixos-version")
|
test_depl.machines['machine'].run_command("nixos-version")
|
||||||
if not args.keep:
|
if not args.keep:
|
||||||
|
@ -1,12 +1,11 @@
|
|||||||
{
|
{
|
||||||
network.description = "NixOS EBS creator";
|
network.description = "NixOS EBS creator";
|
||||||
|
|
||||||
machine =
|
machine =
|
||||||
{ config, pkgs, ... }:
|
{ config, pkgs, ... }:
|
||||||
{ deployment.targetEnv = "ec2";
|
{ deployment.targetEnv = "ec2";
|
||||||
deployment.ec2.instanceType = "m1.small";
|
deployment.ec2.instanceType = "m1.small";
|
||||||
deployment.ec2.keyPair = "eelco";
|
deployment.ec2.securityGroups = [ "admin" ];
|
||||||
deployment.ec2.securityGroups = [ "eelco-test" ];
|
|
||||||
environment.systemPackages = [ pkgs.parted ];
|
environment.systemPackages = [ pkgs.parted ];
|
||||||
fileSystems = [];
|
fileSystems = [];
|
||||||
};
|
};
|
||||||
|
Loading…
x
Reference in New Issue
Block a user