aicra/util.go

package aicra

import (
	"log"
	"net/http"

	"git.xdrm.io/go/aicra/api"
	"git.xdrm.io/go/aicra/internal/config"
	"git.xdrm.io/go/aicra/internal/reqdata"
)

// extractParameters extracts parameters for the request and checks
// every single one according to configuration options
func (s *httpServer) extractParameters(store *reqdata.Store, methodParam map[string]*config.Parameter) (map[string]interface{}, api.Error) {

	// init vars
	parameters := make(map[string]interface{})

	// for each param of the config
	for name, param := range methodParam {

		// 1. extract value
		p, isset := store.Set[name]

		// 2. fail if required & missing
		if !isset && !param.Optional {
			apiErr := api.ErrorMissingParam()
			apiErr.SetArguments(name)
			return nil, apiErr
		}

		// 3. optional & missing: set default value
		if !isset {
			p = &reqdata.Parameter{
				Parsed: true,
				File:   param.Type == "FILE",
				Value:  nil,
			}
			if param.Default != nil {
				p.Value = *param.Default
			}

			// we are done
			parameters[param.Rename] = p.Value
			continue
		}

		// 4. parse parameter if not file
		if !p.File {
			p.Parse()
		}

		// 5. fail on unexpected multipart file
		waitFile, gotFile := param.Type == "FILE", p.File
		if gotFile && !waitFile || !gotFile && waitFile {
			apiErr := api.ErrorInvalidParam()
			apiErr.SetArguments(param.Rename, "FILE")
			return nil, apiErr
		}

		// 6. ignore type check if file
		if gotFile {
			parameters[param.Rename] = p.Value
			continue
		}

		// 7. check type
		if s.Checkers.Run(param.Type, p.Value) != nil {
			apiErr := api.ErrorInvalidParam()
			apiErr.SetArguments(param.Rename, param.Type, p.Value)
			return nil, apiErr
		}

		parameters[param.Rename] = p.Value

	}

	return parameters, api.ErrorSuccess()
}

var handledMethods = []string{http.MethodGet, http.MethodPost, http.MethodPut, http.MethodDelete}

// Prints an error as HTTP response
func logError(res *api.Response) {
	log.Printf("[http.fail] %v\n", res)
}

// logService logs a service details
func logService(s config.Service, path string) {
	for _, method := range handledMethods {
		if m := s.Method(method); m != nil {
			if path == "" {
				log.Printf("    ->\t%s\t'/'\n", method)
			} else {
				log.Printf("    ->\t%s\t'%s'\n", method, path)
			}
		}
	}

	if s.Children != nil {
		for subPath, child := range s.Children {
			logService(*child, path+"/"+subPath)
		}
	}
}