func main() {
flag.Parse()
shutdown.Logger = log.New(os.Stdout, "", log.LstdFlags)
shutdown.OnSignal(0, os.Interrupt, syscall.SIGTERM)
shutdown.SetTimeout(time.Second)
s, err := server.NewServer(*configfile)
if err != nil {
log.Fatal("Error loading server configuration:", err)
}
s.Run()
}
func main() {
//
flag.Usage = func() {
fmt.Fprintf(os.Stderr, "Usage: %s [-options] [command]\n", os.Args[0])
fmt.Println("Options:")
flag.PrintDefaults()
fmt.Println("Commands: (if none is given the doproxy server is started)")
fmt.Println(` add <id>`)
fmt.Println(` Add a running droplet to your inventory.`)
fmt.Println(` create [new-droplet-name]`)
fmt.Println(` Create a new backend and add it as a backend to the configuration.`)
fmt.Println(` If no name is given a name is generated.`)
fmt.Println(` delete <id>`)
fmt.Println(` Delete a backend with the given id.`)
fmt.Println(` destroy <id>`)
fmt.Println(` Destroy a running droplet with the given id.`)
fmt.Println(` list`)
fmt.Println(` List all currently running droplets.`)
fmt.Println(` reboot <id>`)
fmt.Println(` Reboot the backend with the given id.`)
fmt.Println(` sanitize [apply]`)
fmt.Println(` Sanitize the inventory. All droplets that cannot be located on`)
fmt.Println(` DigitalOcean will be listed, or removed if 'apply' is specified.`)
}
flag.Parse()
shutdown.Logger = log.New(os.Stdout, "", log.LstdFlags)
shutdown.OnSignal(0, os.Interrupt, syscall.SIGTERM)
shutdown.SetTimeout(time.Second)
args := flag.Args()
if len(args) == 0 {
s, err := server.NewServer(*configfile)
if err != nil {
log.Fatal("Error loading server configuration:", err)
}
s.Run()
return
}
cmd := args[0]
conf, err := server.ReadConfigFile(*configfile)
if err != nil {
log.Fatal("Error loading server configuration:", err)
}
// We do not want health checks to be running.
conf.Backend.DisableHealth = true
switch cmd {
case "create":
name := ""
if len(args) >= 2 {
name = args[1]
}
drop, err := server.CreateDroplet(*conf, name)
if err != nil {
log.Fatal("Error creating droplet:", err)
}
log.Println("Adding droplet to inventory")
inv, err := server.ReadInventory(conf.InventoryFile, conf.Backend)
if err != nil {
log.Fatal("Error loading inventory:", err)
}
be := server.NewDropletBackend(*drop, conf.Backend)
err = inv.AddBackend(be)
if err != nil {
log.Fatal("Error adding droplet to inventory:", err)
}
err = inv.SaveDroplets(conf.InventoryFile)
if err != nil {
log.Fatal("Error saving new inventory:", err)
}
log.Println("New inventory saved.")
case "list":
drops, err := server.ListDroplets(*conf)
if err != nil {
log.Fatal("Error listing droplets:", err)
}
fmt.Printf("%d Currently Running:\n", len(drops.Droplets))
for _, drop := range drops.Droplets {
fmt.Println("\n[[droplet]]\n" + drop.String())
}
case "add":
if len(args) < 2 {
log.Fatal("add: No ID supplied")
}
sid := args[1]
id, err := strconv.Atoi(sid)
if err != nil {
log.Fatalf("%q is not a valid ID. It must be a number ", sid)
}
inv, err := server.ReadInventory(conf.InventoryFile, conf.Backend)
if err != nil {
log.Fatal("Error loading inventory:", err)
}
_, ok := inv.BackendID(sid)
if ok {
log.Fatalf("Droplet with id %q already exists in inventory", sid)
}
drops, err := server.ListDroplets(*conf)
if err != nil {
log.Fatal("Error listing droplets:", err)
}
drop, ok := drops.DropletID(id)
if !ok {
log.Fatal("Unable to locate a running droplet with ID ", sid)
}
be, err := drop.ToBackend(conf.Backend)
if err != nil {
log.Fatal("Error listing droplets:", err)
}
err = inv.AddBackend(be)
if err != nil {
log.Fatal("Error adding backend:", err)
}
err = inv.SaveDroplets(conf.InventoryFile)
if err != nil {
log.Fatal("Error saving inventory:", err)
}
log.Printf("Backend %q added to inventory", sid)
case "sanitize":
apply := false
if len(args) >= 2 {
apply = args[1] == "apply"
}
inv, err := server.ReadInventory(conf.InventoryFile, conf.Backend)
if err != nil {
log.Fatal("Error loading inventory:", err)
}
drops, err := server.ListDroplets(*conf)
if err != nil {
log.Fatal("Error listing droplets:", err)
}
ids := inv.IDs()
var remove []string
for _, id := range ids {
n, err := strconv.Atoi(id)
if err != nil {
log.Println("warning: unable to parse id", id)
continue
}
_, ok := drops.DropletID(n)
if ok {
continue
}
be, ok := inv.BackendID(id)
if !ok {
continue
}
switch be.(type) {
case *server.DropletBackend:
remove = append(remove, id)
default:
log.Printf("Unknown backend type %T\n", be)
}
}
if apply {
for _, be := range remove {
fmt.Println("Removing", be)
err := inv.Remove(be)
if err != nil {
log.Fatal("Error removing item from inventory:", err)
}
}
err = inv.SaveDroplets(conf.InventoryFile)
if err != nil {
log.Fatal("Error saving inventory:", err)
}
} else {
switch len(remove) {
case 0:
fmt.Println("All items in inventory running")
case 1:
fmt.Println("1 iventory backend could not be located:")
defer func() {
fmt.Printf("Run '%s sanitize apply' to remove it.\n", os.Args[0])
}()
default:
fmt.Println(len(remove), "inventory backends could not be located:")
defer func() {
fmt.Printf("Run '%s sanitize apply' to remove them.\n", os.Args[0])
}()
}
for _, be := range remove {
fmt.Println("ID", be)
}
}
case "reboot":
if len(args) < 2 {
log.Fatal("No id supplied")
}
name := args[1]
n, err := strconv.Atoi(name)
if err != nil {
log.Fatalf("warning: unable to parse id %q", name)
}
inv, err := server.ReadInventory(conf.InventoryFile, conf.Backend)
if err != nil {
log.Fatal("Error loading inventory:", err)
}
drops, err := server.ListDroplets(*conf)
if err != nil {
log.Fatalln("Error listing droplets:", err)
}
drop, ok := drops.DropletID(n)
if !ok {
log.Fatalln("Cannot find any running droplet droplet with id", n)
}
// If we have it in the inventory
be, hasBe := inv.BackendID(name)
if hasBe {
log.Println("Removing backend", be.Name(), "from inventory")
if err := inv.Remove(name); err != nil {
log.Fatal("Error removing from inventory:", err)
}
if err := inv.SaveDroplets(conf.InventoryFile); err != nil {
log.Fatal("Error saving inventory:", err)
}
log.Println("Backend removed. Wait 5 seconds before reboot")
time.Sleep(time.Second * 5)
}
err = drop.Reboot(*conf)
if err != nil {
log.Println("Error rebooting:", n)
} else {
log.Println("Initiated reboot of", drop.ID, drop.Name)
}
// Re-add Backend
if hasBe {
time.Sleep(time.Second * 5)
if err := inv.AddBackend(be); err != nil {
log.Fatal("Error re-adding backend to inventory:", err)
}
if err := inv.SaveDroplets(conf.InventoryFile); err != nil {
log.Fatal("Error saving inventory:", err)
}
log.Println("Re-added backend to inventory")
}
case "delete":
if len(args) < 2 {
log.Fatal("No id supplied")
}
name := args[1]
inv, err := server.ReadInventory(conf.InventoryFile, conf.Backend)
if err != nil {
log.Fatal("Error loading inventory:", err)
}
_, ok := inv.BackendID(name)
if !ok {
log.Fatalln("Unable to locate a backend with ID", name)
}
err = inv.Remove(name)
if err != nil {
log.Fatalln("Unable to delete backend:", err)
}
err = inv.SaveDroplets(conf.InventoryFile)
if err != nil {
log.Fatalln("Error saving updated inventory:", err)
}
log.Printf("Backend %q deleted from inventory", name)
case "destroy":
if len(args) < 2 {
log.Fatal("No id supplied")
}
name := args[1]
n, err := strconv.Atoi(name)
if err != nil {
log.Fatalf("warning: unable to parse id %q", name)
}
drops, err := server.ListDroplets(*conf)
if err != nil {
log.Fatalln("Error listing droplets:", err)
}
drop, ok := drops.DropletID(n)
if !ok {
log.Fatalln("Cannot find any running droplet droplet with id", n)
}
inv, err := server.ReadInventory(conf.InventoryFile, conf.Backend)
if err != nil {
log.Fatal("Error loading inventory:", err)
}
_, ok = inv.BackendID(name)
if !ok {
log.Println("Unable to locate a backend with ID", name)
} else {
err = inv.Remove(name)
if err != nil {
log.Fatalln("Unable to delete backend:", err)
}
err = inv.SaveDroplets(conf.InventoryFile)
if err != nil {
log.Fatalln("Error saving updated inventory:", err)
}
log.Printf("Backend %s deleted from inventory", name)
time.Sleep(time.Second * 5)
}
err = drop.Delete(*conf)
if err != nil {
log.Fatalln("Error destroying droplet:", err)
}
log.Printf("Droplet %d %q destroyed", drop.ID, drop.Name)
case "help":
flag.Usage()
default:
flag.Usage()
log.Println("Unknown command:", cmd)
os.Exit(1)
}
}