changes for the CLI functions mgmt

Author: dstpierre

function/management.go

@@ -113,6 +113,22 @@ func GetByID(db *mongo.Database, id string) (ExecData, error) {
 	return result, nil
 }
 
+func GetByName(db *mongo.Database, name string) (ExecData, error) {
+	var result ExecData
+
+	filter := bson.M{"name": name}
+
+	ctx := context.Background()
+	sr := db.Collection("sb_functions").FindOne(ctx, filter)
+	if err := sr.Decode(&result); err != nil {
+		return result, err
+	} else if err := sr.Err(); err != nil {
+		return result, err
+	}
+
+	return result, nil
+}
+
 func List(db *mongo.Database) ([]ExecData, error) {
 	opt := &options.FindOptions{}
 	opt.SetProjection(bson.M{"h": 0})

function/runtime.go

@@ -126,7 +126,7 @@ func (env *ExecutionEnvironment) addDatabaseFunctions(vm *goja.Runtime) {
 	})
 	vm.Set("getById", func(call goja.FunctionCall) goja.Value {
 		if len(call.Arguments) != 2 {
-			return vm.ToValue(Result{Content: "argument missmatch: you need 2 arguments for get(repo, id)"})
+			return vm.ToValue(Result{Content: "argument missmatch: you need 2 arguments for get(col, id)"})
 		}
 		var col, id string
 		if err := vm.ExportTo(call.Argument(0), &col); err != nil {

functions.go

@@ -76,6 +76,11 @@ func (f *functions) del(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
+	if err := function.Delete(curDB, name); err != nil {
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+
 	w.WriteHeader(http.StatusOK)
 }
 
@@ -113,3 +118,41 @@ func (f *functions) exec(w http.ResponseWriter, r *http.Request) {
 
 	w.WriteHeader(http.StatusOK)
 }
+
+func (f *functions) list(w http.ResponseWriter, r *http.Request) {
+	conf, _, err := middleware.Extract(r, false)
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusUnauthorized)
+		return
+	}
+
+	curDB := client.Database(conf.Name)
+
+	results, err := function.List(curDB)
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+
+	respond(w, http.StatusOK, results)
+}
+
+func (f *functions) info(w http.ResponseWriter, r *http.Request) {
+	conf, _, err := middleware.Extract(r, false)
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusUnauthorized)
+		return
+	}
+
+	curDB := client.Database(conf.Name)
+
+	name := getURLPart(r.URL.Path, 3)
+
+	fn, err := function.GetByName(curDB, name)
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+
+	respond(w, http.StatusOK, fn)
+}

server.go

@@ -162,7 +162,9 @@ func Start(dbHost, port string) {
 	http.Handle("/fn/add", middleware.Chain(http.HandlerFunc(f.add), stdRoot...))
 	http.Handle("/fn/update", middleware.Chain(http.HandlerFunc(f.update), stdRoot...))
 	http.Handle("/fn/delete/", middleware.Chain(http.HandlerFunc(f.del), stdRoot...))
+	http.Handle("/fn/info/", middleware.Chain(http.HandlerFunc(f.info), stdRoot...))
 	http.Handle("/fn/exec", middleware.Chain(http.HandlerFunc(f.exec), stdAuth...))
+	http.Handle("/fn", middleware.Chain(http.HandlerFunc(f.list), stdRoot...))
 
 	// ui routes
 	webUI := ui{base: &db.Base{PublishDocument: volatile.PublishDocument}}