package databag

import (
	"encoding/base64"
	"encoding/json"
	"errors"
	"log"
	"net/http"
	"net/http/httptest"
	"os"
	"runtime"
	"strings"
)

func WriteResponse(w http.ResponseWriter, v interface{}) {
	body, err := json.Marshal(v)
	if err != nil {
		_, file, line, _ := runtime.Caller(1)
		p, _ := os.Getwd()
		log.Printf("%s:%d %s", strings.TrimPrefix(file, p), line, err.Error())
		w.WriteHeader(http.StatusInternalServerError)
	} else {
		w.Header().Set("Content-Type", "application/json")
		w.Write(body)
	}
}

func ReadResponse(w *httptest.ResponseRecorder, v interface{}) error {
	resp := w.Result()
	if resp.StatusCode != 200 {
		return errors.New("response failed")
	}
	if v == nil {
		return nil
	}
	dec := json.NewDecoder(resp.Body)
	dec.Decode(v)
	return nil
}

func SetBasicAuth(r *http.Request, login string) {
	auth := base64.StdEncoding.EncodeToString([]byte(login))
	r.Header.Add("Authorization", "Basic "+auth)
}

func SetBearerAuth(r *http.Request, token string) {
	r.Header.Add("Authorization", "Bearer "+token)
}

func SetCredentials(r *http.Request, login string) {
	auth := base64.StdEncoding.EncodeToString([]byte(login))
	r.Header.Add("Credentials", "Basic "+auth)
}

func ParseRequest(r *http.Request, w http.ResponseWriter, obj interface{}) error {
	r.Body = http.MaxBytesReader(w, r.Body, APPBodyLimit)
	dec := json.NewDecoder(r.Body)
	dec.DisallowUnknownFields()
	return dec.Decode(&obj)
}

func EnableCors(w *http.ResponseWriter) {
	(*w).Header().Set("Access-Control-Allow-Origin", "*")
}