From 7b7847571aeab5bdf1b762cee3fd9303c3a21478 Mon Sep 17 00:00:00 2001
From: Chandler Swift <chandler@chandlerswift.com>
Date: Sat, 3 Aug 2024 23:13:27 -0500
Subject: [PATCH] Add basic create-server.py script for spinning up new nixos
 servers from vultr

---
 .gitignore       |  1 +
 create-server.py | 75 ++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 76 insertions(+)
 create mode 100644 .gitignore
 create mode 100755 create-server.py

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..7ea4304
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1 @@
+keys.toml
diff --git a/create-server.py b/create-server.py
new file mode 100755
index 0000000..9236837
--- /dev/null
+++ b/create-server.py
@@ -0,0 +1,75 @@
+#!/usr/bin/env nix-shell
+#! nix-shell -i python3 -p python3 python3Packages.requests
+# https://wiki.nixos.org/wiki/Nix-shell_shebang#Python
+
+import requests
+import tomllib
+import json
+import argparse
+import os
+import base64
+
+nixos_infect_script="""
+#!/bin/sh
+
+curl https://raw.githubusercontent.com/elitak/nixos-infect/master/nixos-infect | NIX_CHANNEL=nixos-24.05 bash
+"""
+
+parser = argparse.ArgumentParser(description='Create a NixOS Vultr Server.')
+parser.add_argument("label", help="Name for the server")
+parser.add_argument("--region", default="ord")
+parser.add_argument("--plan-type", default="vhp")
+parser.add_argument("--vcpu-count", type=int, default=1)
+parser.add_argument("--ram", type=int, default=2048)
+parser.add_argument("--os", default="Debian 12", help="Prefix search")
+parser.add_argument("--hostname")
+
+args = parser.parse_args()
+
+with open(os.path.join(os.path.dirname(os.path.realpath(__file__)), "keys.toml"), "rb") as f:
+    config = tomllib.load(f)
+
+headers = {
+    "Authorization": f"Bearer {config["vultr-token"]}",
+}
+
+plans = requests.get("https://api.vultr.com/v2/plans?per_page=500").json()['plans']
+plans = [p for p in plans if p['type'] == args.plan_type]
+plans = [p for p in plans if p['vcpu_count'] == args.vcpu_count]
+plans = [p for p in plans if p['ram'] == args.ram]
+plans = [p for p in plans if args.region in p['locations']]
+plans = [p for p in plans if "amd" in p['id']] # HACK: don't key on ID
+if len(plans) != 1:
+    print(plans)
+    raise Exception(f"Expected exactly 1 plan; got {len(plans)}")
+plan = plans[0]
+print(f"Using plan {plan}")
+
+
+oses = requests.get("https://api.vultr.com/v2/os?per_page=500").json()['os']
+oses = [o for o in oses if o['name'].startswith(args.os)]
+if len(oses) != 1:
+    print(oses)
+    raise Exception(f"Expected exactly 1 OS; got {len(oses)}")
+chosen_os = oses[0]
+
+sshkey_id = requests.get("https://api.vultr.com/v2/ssh-keys", headers=headers).json()['ssh_keys'][0]['id']
+
+data = {
+    "plan": plan['id'],
+    "region": args.region,
+    "label": args.label,
+    "os_id": chosen_os['id'],
+    "backups": "disabled",
+    "hostname": args.hostname or args.label.split('.')[0],
+    "sshkey_id": [sshkey_id],
+    "user_data": base64.b64encode(nixos_infect_script.encode()).decode(),
+}
+
+print(data)
+
+if input("Confirm? ")[0] != 'y':
+    raise Exception('aborted')
+
+res = requests.post("https://api.vultr.com/v2/instances", headers=headers, data=json.dumps(data)).json()
+print(res)