Merge pull request #18 from getnelson/blueprints/create

Add support for creating and inspecting blueprints

Author: timperrett

README.md

@@ -94,6 +94,22 @@ $ nelson namespace create --datacenter dc123 --namespace foobar
 $ nelson ns create --dc dc123 --ns foobar
 ```
 
+### Blueprint Operations
+
+```
+# proof a template before commiting it to nelson
+$ nelson blueprint proof -f /path/to/gpu-kubernetes.mustache
+
+# create a blueprint from a template on your client host
+$ nelson blueprint create -n somename -f /path/to/gpu-kubernetes.mustache
+
+# inspect a template
+nelson blueprint inspect -n cpu-cron-job
+nelson blueprint inspect -n cpu-cron-job -r HEAD
+nelson blueprint inspect cpu-cron-job@HEAD
+nelson blueprint inspect cpu-cron-job@6
+```
+
 ### Unit Operations
 
 ```

src/github.com/getnelson/nelson/blueprint.go

@@ -24,6 +24,8 @@ import (
 	"strconv"
 )
 
+/////////////////// PROOFING BLUEPRINTS ///////////////////
+
 /*
  * {
  *   "content": "CAgICAgIHBsYW5zOg0KICAgICAgICAgIC0gZGVmYXVsdA=="
@@ -59,3 +61,79 @@ func ProofBlueprint(req ProofBlueprintWire, http *gorequest.SuperAgent, cfg *Con
 		return "", errs
 	}
 }
+
+/////////////////// CREATING BLUEPRINTS ///////////////////
+
+/*
+ * {
+ *    "name": "use-nvidia-1080ti",
+ *    "description": "only scheudle on nodes with nvida 1080ti hardware"
+ *    "sha256": "1e34a423ebe1fafeda8277386ede3263b01357e490b124b69bc0bfb493e64140"
+ *    "template": "<base64 encoded template>"
+ * }
+ */
+type CreateBlueprintResponse struct {
+	Name        string `json:"name"`
+	Description string `json:"description"`
+	Revision    string `json:"revision"`
+	Sha256      string `json:"sha256"`
+	Template    string `json:"template"`
+	CreatedAt   int64  `json:"created_at"`
+}
+
+type CreateBlueprintRequest struct {
+	Name        string `json:"name"`
+	Description string `json:"description"`
+	Sha256      string `json:"sha256"`
+	Template    string `json:"template"`
+}
+
+func CreateBlueprint(req CreateBlueprintRequest, http *gorequest.SuperAgent, cfg *Config) (out CreateBlueprintResponse, err []error) {
+
+	r, body, errs := AugmentRequest(
+		http.Post(cfg.Endpoint+"/v1/blueprints"), cfg).Send(req).EndBytes()
+
+	var result CreateBlueprintResponse
+
+	if errs != nil {
+		return result, errs
+	}
+
+	if r.StatusCode/100 != 2 {
+		errs = append(errs, errors.New("Unexpectedly recieved a "+strconv.Itoa(r.StatusCode)+" reponse from the server."))
+		return result, errs
+	} else {
+		if err := json.Unmarshal(body, &result); err != nil {
+			errs = append(errs, errors.New("Unexpected response from Nelson server"))
+		}
+		return result, errs
+	}
+}
+
+/////////////////// LISTING BLUEPRINTS ///////////////////
+
+// func ListBlueprints(http *gorequest.SuperAgent, cfg *Config) (out CreateBlueprintResponse, err []error) {
+// }
+
+/////////////////// INSPECTING BLUEPRINTS ///////////////////
+
+func InspectBlueprint(namedRevision string, http *gorequest.SuperAgent, cfg *Config) (out CreateBlueprintResponse, err []error) {
+	r, body, errs := AugmentRequest(
+		http.Get(cfg.Endpoint+"/v1/blueprints/"+namedRevision), cfg).EndBytes()
+
+	var result CreateBlueprintResponse
+
+	if errs != nil {
+		return result, errs
+	}
+
+	if r.StatusCode/100 != 2 {
+		errs = append(errs, errors.New("Unexpectedly recieved a "+strconv.Itoa(r.StatusCode)+" reponse from the server."))
+		return result, errs
+	} else {
+		if err := json.Unmarshal(body, &result); err != nil {
+			errs = append(errs, errors.New("Unexpected response from Nelson server"))
+		}
+		return result, errs
+	}
+}

src/github.com/getnelson/nelson/main.go

@@ -17,6 +17,7 @@
 package main
 
 import (
+	"crypto/sha256"
 	"encoding/base64"
 	"fmt"
 	"github.com/parnurzeal/gorequest"
@@ -66,6 +67,7 @@ func main() {
 	var selectedNoGrace bool
 	var repository string
 	var owner string
+	var selectedName string
 
 	app.Flags = []cli.Flag{
 		cli.IntFlag{
@@ -138,7 +140,7 @@ func main() {
 						cli.StringFlag{
 							Name:        "source, file, f",
 							Value:       "",
-							Usage:       "The blueprint template file to proof",
+							Usage:       "Path to the blueprint template file to proof",
 							Destination: &selectedManifest,
 						},
 					},
@@ -165,6 +167,98 @@ func main() {
 						return nil
 					},
 				},
+				{
+					Name:  "create",
+					Usage: "Commit the template to Nelson",
+					Flags: []cli.Flag{
+						cli.StringFlag{
+							Name:        "name, n",
+							Value:       "",
+							Usage:       "Name of Blueprint",
+							Destination: &selectedName,
+						},
+						cli.StringFlag{
+							Name:        "description, d",
+							Value:       "",
+							Usage:       "Description of Blueprint templte",
+							Destination: &description,
+						},
+						cli.StringFlag{
+							Name:        "source, file, f",
+							Value:       "",
+							Usage:       "Path to the Blueprint template file",
+							Destination: &selectedManifest,
+						},
+					},
+					Action: func(c *cli.Context) error {
+						if len(selectedManifest) <= 0 {
+							return cli.NewExitError("No blueprint template file specified.", 1)
+						}
+						manifest, err := ioutil.ReadFile(selectedManifest)
+						if err != nil {
+							return cli.NewExitError("Could not read "+selectedManifest, 1)
+						}
+						sum := sha256.Sum256(manifest)
+						sha := fmt.Sprintf("%x", sum)
+						manifestBase64 := base64.StdEncoding.EncodeToString(manifest)
+						wire := CreateBlueprintRequest{
+							Name:        selectedName,
+							Description: description,
+							Sha256:      sha,
+							Template:    manifestBase64,
+						}
+
+						pi.Start()
+						cfg := LoadDefaultConfigOrExit(http)
+						r, e := CreateBlueprint(wire, http, cfg)
+						pi.Stop()
+						if e != nil {
+							return cli.NewExitError("Unable to create blueprint.", 1)
+						} else {
+							fmt.Println("Successfully created blueprint " + r.Name + "@" + r.Revision + ".")
+							fmt.Println("@HEAD will point to revision " + r.Revision + " until future revisions are committed.")
+						}
+						return nil
+					},
+				},
+				{
+					Name:  "inspect",
+					Usage: "Yield the template for a given revision",
+					Flags: []cli.Flag{
+						cli.StringFlag{
+							Name:        "name, n",
+							Value:       "",
+							Usage:       "Name of Blueprint",
+							Destination: &selectedName,
+						},
+						cli.StringFlag{
+							Name:        "revision, r",
+							Value:       "",
+							Usage:       "Specific revision of Blueprint template; e.g HEAD, 3, 5",
+							Destination: &selectedVersion,
+						},
+					},
+					Action: func(c *cli.Context) error {
+						var bpName string
+						namedRevision := c.Args().First()
+						if len(namedRevision) > 0 && len(selectedName) < 1 {
+							bpName = namedRevision
+						} else {
+							bpName = selectedName + "@" + selectedVersion
+						}
+
+						pi.Start()
+						cfg := LoadDefaultConfigOrExit(http)
+						r, e := InspectBlueprint(bpName, http, cfg)
+						pi.Stop()
+						if e != nil {
+							return cli.NewExitError("Unable to create blueprint.", 1)
+						} else {
+							fmt.Println(r.Template)
+						}
+						return nil
+					},
+				},
 			},
 		},
 		////////////////////////////// DATACENTER //////////////////////////////////