.gitignore

@@ -1,4 +1,4 @@
 dospin.json
-test_config.yaml
+test_config.json
 dospin
 main

dospin.go

@@ -27,7 +27,7 @@ type cmdOptions struct {
 func parseCmdOptions() cmdOptions {
 	var o cmdOptions
 	flag.StringVar(&o.cmd, "cmd", CMD_SERVE, "Mode to run command in ("+CMD_SERVE+","+CMD_SPINDOWNALL+")")
-	flag.StringVar(&o.config, "config", "dospin.yaml", "Path to the dospin config file")
+	flag.StringVar(&o.config, "config", "dospin.json", "Path to the dospin config file")
 	flag.StringVar(&o.logFile, "logFile", "stdout", "Path to the dospin log file")
 	flag.Parse()
 	return o
@@ -47,7 +47,7 @@ func spindownAll(opts cmdOptions) {
 	}
 }
 
-func runServer(opts cmdOptions) int {
+func runServer(opts cmdOptions) {
 	if opts.logFile != "stdout" {
 		logFile, err := os.OpenFile(opts.logFile, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0664)
 		if err == nil {
@@ -78,8 +78,8 @@ func runServer(opts cmdOptions) int {
 		}
 	}
 
-	done := make(chan int)
+	done := make(chan interface{})
-	return <-done
+	<-done
 }
 
 func main() {

dospin.json

@@ -0,0 +1,20 @@
+{
+	"ApiToken": "<your token here>",
+	"Servers": {
+		"minecraft": {
+			"Ports": [25565],
+			"UsePublicIP": false,
+			"InitialSize": "4gb",
+			"Size": "4gb",
+			"Region": "nyc1",
+			"SshKeys": [
+				"Key1",
+				"gtalent2@gmail.com"
+			],
+			"UsePersistentImage": false,
+			"ImageSlug": "ubuntu-16-04-x64",
+			"Volumes": ["volume-nyc1-01"],
+			"UserData": "#!/bin/bash\napt-get update\napt-get install -y docker.io\nmkdir -p /mnt/volume-nyc1-01\nmount -o discard,defaults /dev/disk/by-id/scsi-0DO_Volume_volume-nyc1-01 /mnt/volume-nyc1-01\necho /dev/disk/by-id/scsi-0DO_Volume_volume-nyc1-01 /mnt/volume-nyc1-01 ext4 defaults,nofail,discard 0 0 | tee -a /etc/fstab\ndocker run -d --restart=always -p 25565:25565 -v /mnt/volume-nyc1-01/minecraft-server:/minecraft-server -w /minecraft-server -t java:8-alpine sh start.sh"
+		}
+	}
+}

dospin.yaml

@@ -1,26 +0,0 @@
----
-api_token: <your token here>
-servers:
-  minecraft:
-    ports:
-    - 25565
-    activity_timeout_min: 20m
-    use_public_ip: false
-    initial_size: 4gb
-    size: 4gb
-    region: nyc1
-    ssh_keys:
-    - Key1
-    - gtalent2@gmail.com
-    use_persistent_image: false
-    image_slug: ubuntu-16-04-x64
-    volumes:
-    - volume-nyc1-01
-    user_data: |-
-      #!/bin/bash
-      apt-get update
-      apt-get install -y docker.io
-      mkdir -p /mnt/volume-nyc1-01
-      mount -o discard,defaults /dev/disk/by-id/scsi-0DO_Volume_volume-nyc1-01 /mnt/volume-nyc1-01
-      echo /dev/disk/by-id/scsi-0DO_Volume_volume-nyc1-01 /mnt/volume-nyc1-01 ext4 defaults,nofail,discard 0 0 | tee -a /etc/fstab
-      docker run -d --restart=always -p 25565:25565 -v /mnt/volume-nyc1-01/minecraft-server:/minecraft-server -w /minecraft-server -t java:8-alpine sh start.sh

droplethandler.go

@@ -8,7 +8,6 @@
 package main
 
 import (
-	"context"
 	"errors"
 	"github.com/digitalocean/godo"
 	"golang.org/x/oauth2"
@@ -31,14 +30,12 @@ func (t *tokenSource) Token() (*oauth2.Token, error) {
 
 type DropletHandler struct {
 	client   *godo.Client
-	ctx      context.Context
 	settings Settings
 }
 
 func NewDropletHandler(settings Settings) *DropletHandler {
 	retval := new(DropletHandler)
 	retval.settings = settings
-	retval.ctx = context.TODO()
 
 	// setup DO client
 	tokenSource := &tokenSource{settings.ApiToken}
@@ -97,7 +94,7 @@ func (me *DropletHandler) Spinup(name string) (string, error) {
 		}
 
 		log.Println("Spinup: Creating " + name)
-		droplet, _, err := me.client.Droplets.Create(me.ctx, createRequest)
+		droplet, _, err := me.client.Droplets.Create(createRequest)
 		if err != nil {
 			log.Println(err)
 			if droplet == nil {
@@ -106,7 +103,7 @@ func (me *DropletHandler) Spinup(name string) (string, error) {
 		}
 		// wait until machine is ready
 		for {
-			d, _, err := me.client.Droplets.Get(me.ctx, droplet.ID)
+			d, _, err := me.client.Droplets.Get(droplet.ID)
 			if err != nil {
 				log.Println(err)
 				return "", err
@@ -124,7 +121,7 @@ func (me *DropletHandler) Spinup(name string) (string, error) {
 
 			// resize
 			log.Println("Spinup: Resizing " + name)
-			action, _, err := me.client.DropletActions.Resize(me.ctx, droplet.ID, vd.Size, false)
+			action, _, err := me.client.DropletActions.Resize(droplet.ID, vd.Size, false)
 			if err != nil || !me.actionWait(action.ID) {
 				return "", err
 			}
@@ -132,7 +129,7 @@ func (me *DropletHandler) Spinup(name string) (string, error) {
 
 			// power back on
 			log.Println("Spinup: Powering on " + name)
-			action, _, err = me.client.DropletActions.PowerOn(me.ctx, droplet.ID)
+			action, _, err = me.client.DropletActions.PowerOn(droplet.ID)
 			if err != nil || !me.actionWait(action.ID) {
 				return "", err
 			}
@@ -142,7 +139,7 @@ func (me *DropletHandler) Spinup(name string) (string, error) {
 		// delete the image
 		if image.ID > 0 {
 			log.Println("Spinup: Deleting image " + name)
-			_, err = me.client.Images.Delete(me.ctx, image.ID)
+			_, err = me.client.Images.Delete(image.ID)
 			if err != nil {
 				log.Println("Spinup: Could not delete image: ", err)
 			} else {
@@ -153,7 +150,7 @@ func (me *DropletHandler) Spinup(name string) (string, error) {
 		// get the private IP and return it
 
 		// get new copy of droplet that has IP
-		droplet, _, err = me.client.Droplets.Get(me.ctx, droplet.ID)
+		droplet, _, err = me.client.Droplets.Get(droplet.ID)
 		if err == nil {
 			if vd.UsePublicIP {
 				return droplet.PublicIPv4()
@@ -182,7 +179,7 @@ func (me *DropletHandler) Spindown(name string) error {
 	// snapshot existing droplet
 	if me.settings.Servers[name].UsePersistentImage {
 		log.Println("Spindown: Creating image " + name)
-		action, _, err := me.client.DropletActions.Snapshot(me.ctx, droplet.ID, DROPLET_NS+name)
+		action, _, err := me.client.DropletActions.Snapshot(droplet.ID, DROPLET_NS+name)
 		if err != nil || !me.actionWait(action.ID) {
 			return err
 		}
@@ -191,7 +188,7 @@ func (me *DropletHandler) Spindown(name string) error {
 
 	// delete droplet
 	log.Println("Spindown: Deleting droplet " + name)
-	_, err = me.client.Droplets.Delete(me.ctx, droplet.ID)
+	_, err = me.client.Droplets.Delete(droplet.ID)
 	if err != nil {
 		return err
 	}
@@ -219,7 +216,7 @@ func (me *DropletHandler) poweroff(name string) error {
 				break
 			}
 			time.Sleep(100 * time.Millisecond)
-			_, _, err = me.client.DropletActions.Shutdown(me.ctx, droplet.ID)
+			_, _, err = me.client.DropletActions.Shutdown(droplet.ID)
 			if err != nil {
 				log.Println("Power down of", name, "failed:", err)
 			}
@@ -241,7 +238,7 @@ func (me *DropletHandler) getDroplet(name string) (godo.Droplet, error) {
 			Page:    page,
 			PerPage: perPage,
 		}
-		images, _, err := me.client.Droplets.List(me.ctx, opt)
+		images, _, err := me.client.Droplets.List(opt)
 		if err != nil {
 			break
 		}
@@ -273,7 +270,7 @@ func (me *DropletHandler) getSnapshot(name string) (godo.Image, error) {
 			Page:    page,
 			PerPage: perPage,
 		}
-		images, _, err := me.client.Images.ListUser(me.ctx, opt)
+		images, _, err := me.client.Images.ListUser(opt)
 		if err != nil {
 			break
 		}
@@ -296,7 +293,7 @@ func (me *DropletHandler) getSnapshot(name string) (godo.Image, error) {
 
 func (me *DropletHandler) actionWait(actionId int) bool {
 	for {
-		a, _, err := me.client.Actions.Get(me.ctx, actionId)
+		a, _, err := me.client.Actions.Get(actionId)
 		if err != nil {
 			log.Println("Action retrieval failed: ", err)
 		} else if a.Status == "completed" {
@@ -320,7 +317,7 @@ func (me *DropletHandler) sshKeys(keyNames []string) []godo.DropletCreateSSHKey
 			Page:    page,
 			PerPage: perPage,
 		}
-		keys, _, err := me.client.Keys.List(me.ctx, opt)
+		keys, _, err := me.client.Keys.List(opt)
 		if err != nil {
 			break
 		}

scripts/list_droplets.sh

@@ -0,0 +1,2 @@
+#! /usr/bin/env sh
+curl -X GET -H "Content-Type: application/json" -H "Authorization: Bearer `cat dospin.json | jq -r .ApiToken`" "https://api.digitalocean.com/v2/droplets?page=1&per_page=100&private=true" | jq .

scripts/list_images.sh

@@ -0,0 +1,2 @@
+#! /usr/bin/env sh
+curl -X GET -H "Content-Type: application/json" -H "Authorization: Bearer `cat dospin.json | jq -r .ApiToken`" "https://api.digitalocean.com/v2/images?page=1&per_page=100&private=true" | jq .

scripts/list_keys.sh

@@ -0,0 +1,4 @@
+#! /usr/bin/env sh
+
+echo Keys
+curl -X GET -H "Content-Type: application/json" -H "Authorization: Bearer `cat dospin.json | jq -r .ApiToken`" "https://api.digitalocean.com/v2/account/keys" | jq .

scripts/list_volumes.sh

@@ -0,0 +1,4 @@
+#! /usr/bin/env sh
+
+echo Volumes
+curl -X GET -H "Content-Type: application/json" -H "Authorization: Bearer `cat dospin.json | jq -r .ApiToken`" "https://api.digitalocean.com/v2/volumes" | jq .

servermanager.go

@@ -39,7 +39,6 @@ type ServerManager struct {
 	connStatus        chan ConnStatus
 	lastKeepAliveTime time.Time
 	server            ServerHandler
-	activityTimeout   time.Duration
 }
 
 func NewServerManager(name string, server ServerHandler, settings Settings) *ServerManager {
@@ -53,13 +52,6 @@ func NewServerManager(name string, server ServerHandler, settings Settings) *Ser
 	sm.server = server
 	sm.lastKeepAliveTime = time.Now()
 
-	activityTimeout, err := time.ParseDuration(settings.Servers[sm.name].ActivityTimeout)
-	if err != nil { // invalid timeout, default to 5 minutes
-		activityTimeout = time.Duration(5 * time.Minute)
-	}
-	sm.activityTimeout = activityTimeout
-	log.Println("ServerManager: ", name, " has activity timeout of ", sm.activityTimeout.String())
-
 	return sm
 }
 
@@ -69,7 +61,8 @@ func NewServerManager(name string, server ServerHandler, settings Settings) *Ser
 func (me *ServerManager) Serve() {
 	// TODO: see if server is currently up, and setup port forwarding if so
 
-	ticker := time.NewTicker(1 * time.Minute)
+	activityTimeout := time.Duration(5 * time.Minute)
+	ticker := time.NewTicker(activityTimeout)
 
 	// event loop
 	for running := true; running; {
@@ -81,7 +74,7 @@ func (me *ServerManager) Serve() {
 		case action := <-me.in:
 			running = me.serveAction(action)
 		case <-ticker.C:
-			if time.Since(me.lastKeepAliveTime) > me.activityTimeout {
+			if time.Since(me.lastKeepAliveTime) > activityTimeout {
 				running = me.serveAction(serverManagerEvent{eventType: SERVERMANAGER_SPINDOWN})
 			}
 		}

settings.go

@@ -8,27 +8,26 @@
 package main
 
 import (
-	"gopkg.in/yaml.v2"
+	"encoding/json"
 	"io/ioutil"
 )
 
 type Settings struct {
-	ApiToken string            `yaml:"api_token"`
+	ApiToken string
-	Servers  map[string]Server `yaml:"servers"`
+	Servers  map[string]Server
 }
 
 type Server struct {
-	Ports              []int    `yaml:"ports"`
+	Ports              []int
-	ActivityTimeout    string   `yaml:"activity_timeout"`
+	UsePublicIP        bool
-	UsePublicIP        bool     `yaml:"use_public_ip"`
+	InitialSize        string
-	InitialSize        string   `yaml:"initial_size"`
+	Size               string
-	Size               string   `yaml:"size"`
+	Region             string
-	Region             string   `yaml:"region"`
+	UsePersistentImage bool
-	UsePersistentImage bool     `yaml:"use_persistent_image"`
+	ImageSlug          string
-	ImageSlug          string   `yaml:"image_slug"`
+	UserData           string
-	UserData           string   `yaml:"user_data"`
+	SshKeys            []string
-	SshKeys            []string `yaml:"ssh_keys"`
+	Volumes            []string
-	Volumes            []string `yaml:"volumes"`
 }
 
 func loadSettings(path string) (Settings, error) {
@@ -38,7 +37,7 @@ func loadSettings(path string) (Settings, error) {
 		return s, err
 	}
 
-	err = yaml.Unmarshal(data, &s)
+	err = json.Unmarshal(data, &s)
 	if err != nil {
 		return s, err
 	}