Make hat size random

cmd/client/main.go

@@ -3,12 +3,15 @@ package main
 import (
 	"context"
 	"fmt"
+	"math/rand"
 	"net/http"
 	"os"
 
 	"haberdasher-twirp/haberdasher"
 )
 
+const MaxSize = 12
+
 func main() {
 	var host string = "http://localhost:8080"
 	if len(os.Args) > 1 {
@@ -17,7 +20,7 @@ func main() {
 
 	client := haberdasher.NewHaberdasherProtobufClient(host, &http.Client{})
 
-	hat, err := client.MakeHat(context.Background(), &haberdasher.Size{Inches: 12})
+	hat, err := client.MakeHat(context.Background(), &haberdasher.Size{Inches: int32(rand.Intn(MaxSize))})
 	if err != nil {
 		fmt.Printf("oh no: %v\n", err)
 		os.Exit(1)

cmd/main.go

@@ -1,19 +1,39 @@
 package main
 
 import (
+	"fmt"
 	"net/http"
 	"os"
+	"path"
 
 	"haberdasher-twirp/haberdasher"
 	"haberdasher-twirp/internal/haberdasherserver"
 )
 
 func main() {
+	var cwd, dbPath, dbDir string
+	var err error
+
 	var bind string = ":8080"
+
+	cwd, err = os.Getwd()
+	dbPath = path.Join(cwd, "var/hat.db")
+	dbDir = path.Dir(dbPath)
+
+	if _, err = os.Stat(dbDir); os.IsNotExist(err) {
+		err = os.Mkdir(dbDir, 0700)
+		if err != nil {
+			fmt.Printf("boltdb: %s\n", err)
+		}
+	}
+
 	if len(os.Args) > 1 {
 		bind = os.Args[1]
 	}
-	server := &haberdasherserver.Server{}
+
+	fmt.Printf("boltdb: %s\n", dbPath)
+
+	server := &haberdasherserver.Server{DBPath: dbPath}
 	twirpHandler := haberdasher.NewHaberdasherServer(server)
 
 	http.ListenAndServe(bind, twirpHandler)

internal/haberdasherserver/server.go

@@ -16,10 +16,12 @@ import (
 
 const dbBucket = "hats"
 
-type Server struct{}
+type Server struct {
+	DBPath string
+}
 
 type Hat struct {
-	Inches int32  `json:"inchues"`
+	Inches int32  `json:"inches"`
 	Color  string `json:"color"`
 	Name   string `json:"name"`
 }
@@ -32,7 +34,7 @@ type Store struct {
 	db *bolt.DB
 }
 
-func HatToHatModel(h *pb.Hat) Hat {
+func PbToHatModel(h *pb.Hat) Hat {
 	return Hat{
 		Inches: h.Inches,
 		Color:  h.Color,
@@ -40,7 +42,7 @@ func HatToHatModel(h *pb.Hat) Hat {
 	}
 }
 
-func HatModelToHat(h Hat) *pb.Hat {
+func HatModelToPb(h Hat) *pb.Hat {
 	return &pb.Hat{
 		Inches: h.Inches,
 		Color:  h.Color,
@@ -48,23 +50,26 @@ func HatModelToHat(h Hat) *pb.Hat {
 	}
 }
 
-func HatsModelToHats(hs []Hat) (hats *pb.Hats) {
+func HatsModelToPb(hs []Hat) (hats *pb.Hats) {
 	hats = &pb.Hats{}
 	for _, h := range hs {
-		hat := HatModelToHat(h)
+		hat := HatModelToPb(h)
 		hats.Hats = append(hats.Hats, hat)
 	}
 	return hats
 }
 
-func HatQueryToHatQueryModel(q *pb.HatQuery) HatQuery {
+func PbToHatQueryModel(q *pb.HatQuery) HatQuery {
 	return HatQuery{
 		Limit: q.Limit,
 	}
 }
 
 func (s *Server) MakeHat(ctx context.Context, size *pb.Size) (hat *pb.Hat, err error) {
-	st, _ := NewStore("hat.db", 0600, nil)
+	var colors []string = []string{"white", "black", "red", "blue", "white"}
+	var names []string = []string{"bowler", "baseball cap", "top hat", "derby", "tricorne"}
+
+	st, _ := NewStore(s.DBPath, 0600, nil)
 	defer st.Close()
 
 	if size.Inches <= 0 {
@@ -73,39 +78,24 @@ func (s *Server) MakeHat(ctx context.Context, size *pb.Size) (hat *pb.Hat, err e
 
 	h := Hat{
 		Inches: size.Inches,
-		Color:  []string{"white", "black", "red", "blue"}[rand.Intn(4)],
+		Color:  colors[rand.Intn(len(colors))],
-		Name:   []string{"bowler", "baseball cap", "top hat", "derby"}[rand.Intn(3)],
+		Name:   names[rand.Intn(len(names))],
 	}
 
 	fmt.Printf("made hat: %+v\n", h)
 
 	st.SaveHat(h)
 
-	return HatModelToHat(h), nil
+	return HatModelToPb(h), nil
 }
 
 func (s *Server) ListHats(ctx context.Context, q *pb.HatQuery) (hats *pb.Hats, err error) {
-	/*
+	st, _ := NewStore(s.DBPath, 0600, nil)
-		var hat *pb.Hat
-
-		hats = &pb.Hats{}
-
-		for i := 0; i < 10; i++ {
-			hat = &pb.Hat{
-				Inches: int32(rand.Intn(12)),
-				Color:  []string{"white", "black", "red", "blue"}[rand.Intn(4)],
-				Name:   []string{"bowler", "baseball cap", "top hat", "derby"}[rand.Intn(3)],
-			}
-			hats.Hats = append(hats.Hats, hat)
-		}
-	*/
-
-	st, _ := NewStore("hat.db", 0600, nil)
 	defer st.Close()
 
-	hs, err := st.ListHats(HatQueryToHatQueryModel(q))
+	hs, err := st.ListHats(PbToHatQueryModel(q))
 
-	return HatsModelToHats(hs), nil
+	return HatsModelToPb(hs), nil
 }
 
 func itob(v uint64) []byte {