main.go

package main

import (
  "fmt"
  "log"
  "net/http"
  "os"
  "path"

  "github.com/gorilla/mux"
)

var bushwoodServer = ""
var bushwoodToken = ""
var templatePath = ""

func main() {
  httpPort := os.Getenv("HTTP_PORT")
  bakeryRoot := os.Getenv("BAKERY_ROOT")
  nfsServer := os.Getenv("BAKERY_ADDRESS")
  inventoryDbPath := os.Getenv("DB_PATH")
  templatePath = os.Getenv("TEMPLATE_PATH")
  bushwoodServer = os.Getenv("BUSHWOOD_SERVER")
  bushwoodToken = os.Getenv("BUSHWOOD_TOKEN")
  kpartxPath := os.Getenv("KPARTX_PATH")

  if httpPort == "" {
    log.Fatalln("HTTP_PORT env var not set")
  }

  if bakeryRoot == "" {
    log.Fatalln("BAKERY_ROOT env var not set")
  }

  if nfsServer == "" {
    log.Fatalln("BAKERY_ADDRESS env var not set")
  }

  if inventoryDbPath == "" {
    log.Fatalln("DB_PATH env var not set")
  }

  if templatePath == "" {
    log.Fatalln("TEMPLATE_PATH env var not set")
  }

  if bushwoodServer == "" {
    log.Fatalln("BUSHWOOD_SERVER env var not set")
  }

  if bushwoodToken == "" {
    log.Fatalln("BUSHWOOD_TOKEN env var not set")
  }

  if kpartxPath == "" {
    log.Fatalln("KPARTX_PATH env var not set")
  }

  nfsRoot := path.Join(bakeryRoot, "/nfs/")
  imageFolder := path.Join(bakeryRoot, "/bakeforms/")
  bootFolder := path.Join(bakeryRoot, "/boot/")
  mountRoot := path.Join(bakeryRoot, "/mnt")

  initFolders(nfsRoot, imageFolder, bootFolder, mountRoot)

  fb, err := newFileBackend(nfsServer, nfsRoot, bootFolder)
  if err != nil {
    log.Fatalln(err.Error())
  }

  diskmgr, err := NewDiskManager(fb)

  bakeforms, err := newBakeformInventory(imageFolder, mountRoot, fb, kpartxPath)
  if err != nil {
    log.Fatalln(err.Error())
  }
  defer bakeforms.UnmountAll()

  pile, err := NewPiManager(bakeforms, diskmgr, inventoryDbPath)
  if err != nil {
    log.Fatalln(err.Error())
  }

  fs, err := newFileServer(fb, pile, diskmgr)
  if err != nil {
    log.Fatalln(err.Error())
  }

  log.Println("Restoring power state")
  pis, err := pile.ListOven()
  for _, pi := range pis {
    err = pi.PowerOn()
    if err != nil {
      log.Printf("Could not restore power state of rPi with ID: %v. %v\n", pi.Id, err)
    }
  }

  r := mux.NewRouter()
  r.Path("/api/v1/files/{piId}/{filename}").Methods(http.MethodGet).HandlerFunc(fs.fileHandler) //Generates files for net booting

  r.Path("/api/v1/fridge").Methods(http.MethodGet).HandlerFunc(pile.FridgeHandler)
  r.Path("/api/v1/fridge").Methods(http.MethodPost).HandlerFunc(pile.BakeHandler)
  r.Path("/api/v1/fridge/{piId}").Methods(http.MethodDelete).HandlerFunc(pile.UnFridgeHandler)

  r.Path("/api/v1/oven/{piId}/powercycle").Methods(http.MethodPost).HandlerFunc(pile.RebootHandler)
  r.Path("/api/v1/oven/{piId}/disks").Methods(http.MethodPost).HandlerFunc(pile.AttachDiskHandler)
  r.Path("/api/v1/oven/{piId}/disks/{diskId}").Methods(http.MethodDelete).HandlerFunc(pile.DetachDiskHandler)
  r.Path("/api/v1/oven/{piId}/upload/{filename}").Methods(http.MethodPost).HandlerFunc(pile.UploadHandler)
  r.Path("/api/v1/oven/{piId}/download/{filename}").Methods(http.MethodGet).HandlerFunc(pile.DownloadHandler)
  r.Path("/api/v1/oven/{piId}").Methods(http.MethodGet).HandlerFunc(pile.GetPiHandler)
  r.Path("/api/v1/oven/{piId}").Methods(http.MethodDelete).HandlerFunc(pile.UnbakeHandler)
  r.Path("/api/v1/oven").Methods(http.MethodGet).HandlerFunc(pile.OvenHandler)

  r.Path("/api/v1/bakeforms").Methods(http.MethodGet).HandlerFunc(bakeforms.ListHandler)
  r.Path("/api/v1/bakeforms/{name}").Methods(http.MethodPost).HandlerFunc(bakeforms.UploadHandler)
  r.Path("/api/v1/bakeforms/{name}").Methods(http.MethodDelete).HandlerFunc(bakeforms.DeleteHandler)

  r.Path("/api/v1/disks/{diskId}").Methods(http.MethodDelete).HandlerFunc(diskmgr.destroyDiskHandler)
  r.Path("/api/v1/disks/{diskId}").Methods(http.MethodGet).HandlerFunc(diskmgr.getDiskHandler)
  r.Path("/api/v1/disks").Methods(http.MethodPost).HandlerFunc(diskmgr.createDiskHandler)
  r.Path("/api/v1/disks").Methods(http.MethodGet).HandlerFunc(diskmgr.listDisksHandler)

  log.Println("Ready to bake!")
  http.ListenAndServe(fmt.Sprintf(":%v", httpPort), r)

}