package main

import (
	"context"
	"flag"
	"ippserver/packages/ipp"
	"ippserver/packages/mdnsserver"
	"net/http"
	"sync"

	log "github.com/sirupsen/logrus"
)

var (
	loglevel string
	location string
)

func init() {
	flag.StringVar(&loglevel, "loglevel", "info", "The wanted loglevel error/info/debug")
}

func main() {
	customFormatter := new(log.TextFormatter)
	customFormatter.TimestampFormat = "2006-01-02 15:04:05"
	log.SetFormatter(customFormatter)
	customFormatter.FullTimestamp = true
	log.SetLevel(log.InfoLevel)

	ctx, cancel := context.WithCancel(context.Background())
	defer cancel()
	go mdnsserver.Run(ctx)

	http.HandleFunc("/ipp/print", handle)

	log.Info("http server started on :1234")
	err := http.ListenAndServe(":1234", nil)
	if err != nil {
		log.Fatal("ListenAndServe: " + err.Error())
	}
}

var mut sync.Mutex
var requestID uint32

func handle(w http.ResponseWriter, r *http.Request) {
	mut.Lock()
	requestID++
	mut.Unlock()
	if r.Method != http.MethodPost {
		http.Error(w, "Unsupported method", http.StatusMethodNotAllowed)
	}

	request := ipp.NewRequest(0, 0)
	request.UnMarshal(r.Body)
	log.Infof("Upstream Request: id: %v op: %v", request.RequestID(), request.Operation())
	var response *ipp.Response
	var err error
	switch request.Operation() {
	case ipp.GetPrinterAttributes:
		response = handleGetPrinterAttributes(request)
	case ipp.PrintJob:
		response, err = handlePrintJob(request, r.Body, requestID)
	case ipp.GetJobs:
		response, err = handleGetJobs(request, requestID)
	case ipp.ValidateJob:
		response = handleValidateJob(request)
	case ipp.GetJobAttributes:
		response, err = handleGetJobAttributes(request, requestID)
	default:
		response = ipp.NewResponse(ipp.ClientErrorBadRequest, request.RequestID())
	}
	if err != nil {
		log.Errorf("Failed to handle request: %v", err.Error())
		response = ipp.NewResponse(ipp.ServerErrorServiceUnavailable, request.RequestID())
	}
	log.Infof("Upstream Response: %v", response.Header())
	data := response.Marshal()
	w.Write(data)
}