config.go

package main

import (
	"encoding/json"
	"io"
	"log"
	"os"
	"strings"

	"gioman/service"
	"gioman/state"
)

type (
	config struct {
		Requests []requestConfig `json:"requests,omitempty"`
	}
	requestConfig struct {
		Name    string         `json:"name,omitempty"`
		URL     string         `json:"url,omitempty"`
		Method  string         `json:"method,omitempty"`
		Headers []state.Header `json:"headers"`
	}
)

func configFromFilepath(path string) (cfg config, err error) {
	// TODO change the filename path, look for the config folder of the system?
	file, err := os.Open(path)
	if err != nil {
		return
	}
	defer file.Close()

	decoder := json.NewDecoder(file)
	err = decoder.Decode(&cfg)
	return
}

func (cfg *config) save(w io.Writer) error {
	encoder := json.NewEncoder(w)
	encoder.SetIndent("" /*prefix*/, "  " /*indent*/)
	return encoder.Encode(cfg)
}

func (cfg *config) setRequests(requests []state.Request) {
	cfg.Requests = cfg.Requests[:0]
	for _, r := range requests {
		cfg.Requests = append(cfg.Requests, requestConfig{
			Name:    r.Name,
			URL:     r.URL,
			Method:  r.Method.String(),
			Headers: r.Headers,
		})
	}
}

func (cfg *config) requests() (requests []state.Request) {
	for _, r := range cfg.Requests {
		method, ok := service.Methods[strings.ToUpper(r.Method)]
		if !ok {
			method = service.GET
			log.Printf("requests: stored request %q: unknown method %q, assuming %q", r.Name, r.Method, method)
		}
		requests = append(requests, state.Request{
			Name:    r.Name,
			URL:     r.URL,
			Method:  method,
			Headers: r.Headers,
		})
	}
	return
}