Code cleanup, use HTTP POST header for api key

Author: jtJava

go.mod

@@ -5,6 +5,7 @@ go 1.19
 require (
 	github.com/go-chi/chi/v5 v5.0.7
 	github.com/google/uuid v1.3.0
+	go.mongodb.org/mongo-driver v1.10.2
 )
 
 require (
@@ -16,7 +17,6 @@ require (
 	github.com/xdg-go/scram v1.1.1 // indirect
 	github.com/xdg-go/stringprep v1.0.3 // indirect
 	github.com/youmark/pkcs8 v0.0.0-20181117223130-1be2e3e5546d // indirect
-	go.mongodb.org/mongo-driver v1.10.2 // indirect
 	golang.org/x/crypto v0.0.0-20220622213112-05595931fe9d // indirect
 	golang.org/x/sync v0.0.0-20210220032951-036812b2e83c // indirect
 	golang.org/x/text v0.3.7 // indirect

go.sum

@@ -1,9 +1,11 @@
 github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
 github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/go-chi/chi/v5 v5.0.7 h1:rDTPXLDHGATaeHvVlLcR4Qe0zftYethFucbjVQ1PxU8=
 github.com/go-chi/chi/v5 v5.0.7/go.mod h1:DslCQbL2OYiznFReuXYUmQ2hGd1aDpCnlMNITLSKoi8=
 github.com/golang/snappy v0.0.1 h1:Qgr9rKW7uDUkrbSmQeiDsGa8SjGyCOGtuasMWwvp2P4=
 github.com/golang/snappy v0.0.1/go.mod h1:/XxbfmMg8lxefKM7IXC3fBNl/7bRcc72aCRzEWrmP2Q=
+github.com/google/go-cmp v0.5.2 h1:X2ev0eStA3AbceY54o37/0PQ/UWqKEiiO2dKL5OPaFM=
 github.com/google/go-cmp v0.5.2/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
 github.com/google/uuid v1.3.0 h1:t6JiXgmwXMjEs8VusXIJk2BXHsn+wx8BZdTaoZ5fu7I=
 github.com/google/uuid v1.3.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
@@ -16,9 +18,12 @@ github.com/montanaflynn/stats v0.0.0-20171201202039-1bf9dbcd8cbe h1:iruDEfMl2E6f
 github.com/montanaflynn/stats v0.0.0-20171201202039-1bf9dbcd8cbe/go.mod h1:wL8QJuTMNUDYhXwkmfOly8iTdp5TEcJFWZD2D7SIkUc=
 github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
 github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
+github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
 github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
 github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
+github.com/stretchr/testify v1.6.1 h1:hDPOHmpOpP40lSULcqw7IrRb/u7w6RpDC9399XyoNd0=
 github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
+github.com/tidwall/pretty v1.0.0 h1:HsD+QiTn7sK6flMKIvNmpqz1qrpP3Ps6jOKIKMooyg4=
 github.com/tidwall/pretty v1.0.0/go.mod h1:XNkn88O1ChpSDQmQeStsy+sBenx6DDtFZJxhVysOjyk=
 github.com/xdg-go/pbkdf2 v1.0.0 h1:Su7DPu48wXMwC3bs7MCNG+z4FhcyEuz5dlvchbq0B0c=
 github.com/xdg-go/pbkdf2 v1.0.0/go.mod h1:jrpuAogTd400dnrH08LKmI/xc1MbPOebTwRqcT5RDeI=
@@ -43,8 +48,10 @@ golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=
 golang.org/x/text v0.3.7 h1:olpwvP2KacW1ZWvsR7uQhoyTYvKAupfQrRGBFM352Gk=
 golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ=
 golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ=
+golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543 h1:E7g+9GITq07hpfrRu66IVDexMakfv52eLZ2CXBWiKr4=
 golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
 gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
 gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
 gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
+gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
 gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=

handler.go

@@ -0,0 +1,83 @@
+package main
+
+import (
+	"Learning/license"
+	"context"
+	"encoding/json"
+	"github.com/go-chi/chi/v5"
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/bson/primitive"
+	"go.mongodb.org/mongo-driver/mongo"
+	"net/http"
+)
+
+type HandlerContext struct {
+	apiKey     string
+	collection *mongo.Collection
+}
+
+func (ctx *HandlerContext) allUnusedKeys(writer http.ResponseWriter, request *http.Request) {
+	ctx.getKeysWithFilter(writer, request, bson.M{"owner": nil})
+}
+
+func (ctx *HandlerContext) allKeys(writer http.ResponseWriter, request *http.Request) {
+	ctx.getKeysWithFilter(writer, request, nil)
+}
+
+func (ctx *HandlerContext) getKeysWithFilter(writer http.ResponseWriter, request *http.Request, filter primitive.M) {
+	if !ctx.checkApiKey(request.Header.Get("api-key")) {
+		writer.WriteHeader(http.StatusUnauthorized)
+		return
+	}
+
+	var cursor *mongo.Cursor
+	var err error
+	if filter == nil {
+		cursor, err = ctx.collection.Find(context.TODO(), bson.D{})
+	} else {
+		cursor, err = ctx.collection.Find(context.TODO(), filter)
+	}
+
+	if err != nil {
+		panic(err)
+	}
+
+	var results []bson.M
+	if err = cursor.All(context.TODO(), &results); err != nil {
+		panic(err)
+	}
+
+	err = json.NewEncoder(writer).Encode(results)
+}
+
+func (ctx *HandlerContext) registerKey(writer http.ResponseWriter, request *http.Request) {
+	if !ctx.checkApiKey(request.Header.Get("api-key")) {
+		writer.WriteHeader(http.StatusUnauthorized)
+		return
+	}
+
+	licenseKey := license.GenerateKey()
+	_, err := ctx.collection.InsertOne(context.TODO(), licenseKey)
+	if err != nil {
+		return
+	}
+
+	err = json.NewEncoder(writer).Encode(licenseKey)
+	if err != nil {
+		return
+	}
+}
+
+func (ctx *HandlerContext) validateKey(writer http.ResponseWriter, request *http.Request) {
+	key := chi.URLParam(request, "key")
+	var licenseKey bson.D
+	if err := ctx.collection.FindOne(context.TODO(), bson.M{"_id": key}).Decode(&licenseKey); err != nil {
+		_, _ = writer.Write([]byte("false"))
+		return
+	}
+	_, _ = writer.Write([]byte("true"))
+}
+
+func (ctx *HandlerContext) checkApiKey(apiKey string) (valid bool) {
+	return ctx.apiKey == apiKey
+}

license/license.go

@@ -11,6 +11,7 @@ func GenerateKey() License {
 
 type License struct {
 	Key          string    `json:"id" bson:"_id"`
+	Owner        string    `json:"owner" bson:"owner"`
 	CreationDate time.Time `json:"creation_date" bson:"creation_date"`
 	Disabled     bool      `json:"disabled" bson:"disabled"`
 }

main.go

@@ -1,20 +1,21 @@
 package main
 
 import (
-	"Learning/license"
 	"context"
-	"encoding/json"
-	"fmt"
 	"github.com/go-chi/chi/v5"
-	"go.mongodb.org/mongo-driver/bson"
 	"go.mongodb.org/mongo-driver/mongo"
 	"go.mongodb.org/mongo-driver/mongo/options"
 	"go.mongodb.org/mongo-driver/mongo/readpref"
 	"log"
 	"net/http"
+	"os"
 )
 
 func main() {
+	if len(os.Args) < 2 {
+		log.Fatal("You didn't set an api key.")
+	}
+
 	client, err := mongo.Connect(context.TODO(), options.Client().ApplyURI("mongodb://localhost:27017"))
 	if err != nil {
 		panic(err)
@@ -24,41 +25,23 @@ func main() {
 		panic(err)
 	}
 
-	collection = client.Database("default").Collection("licenses")
+	defer client.Disconnect(context.TODO())
 
-	router := chi.NewRouter()
-	router.Get("/create/{key}", registerKey)
-	router.Get("/validate/{key}", validateKey)
+	collection := client.Database("default").Collection("licenses")
 
-	err = http.ListenAndServe(":3000", router)
-	if err != nil {
-		return
-	}
-}
+	handlerContext := HandlerContext{apiKey: os.Args[1], collection: collection}
 
-var collection *mongo.Collection
+	router := chi.NewRouter()
 
-var licenses = make(map[string]license.License)
+	router.Get("/validate/{key}", handlerContext.validateKey)
 
-func registerKey(writer http.ResponseWriter, request *http.Request) {
-	licenseKey := license.GenerateKey()
-	_, err := collection.InsertOne(context.TODO(), licenseKey)
-	if err != nil {
-		return
-	}
+	// Requires API key
+	router.Post("/create", handlerContext.registerKey)
+	router.Post("/all", handlerContext.allKeys)
+	router.Post("/unused", handlerContext.allUnusedKeys)
 
-	licenses[licenseKey.Key] = licenseKey
-	err = json.NewEncoder(writer).Encode(licenseKey)
+	err = http.ListenAndServe(":3000", router)
 	if err != nil {
 		return
 	}
 }
-
-func validateKey(writer http.ResponseWriter, request *http.Request) {
-	key := chi.URLParam(request, "key")
-	var licenseKey bson.D
-	if err := collection.FindOne(context.TODO(), bson.M{"_id": key}).Decode(&licenseKey); err != nil {
-		log.Fatal(err)
-	}
-	fmt.Println(licenseKey)
-}