package main

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

	log "github.com/sirupsen/logrus"
)

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: \n%v\n", request)
	var response *ipp.Response
	switch request.Operation() {
	case ipp.GetPrinterAttributes:
		response = handleGetPrinterAttributes(request)
	case ipp.PrintJob:
		response = handlePrintJob(request, r.Body, requestID)
	case ipp.GetJobs:
		response = handleGetJobs(request, requestID)
	case ipp.ValidateJob:
		response = handleValidateJob(request)
	case ipp.GetJobAttributes:
		response = handleGetJobAttributes(request, requestID)
	default:
		response = ipp.NewResponse(ipp.ClientErrorBadRequest, request.RequestID())
	}

	log.Infof("Upstream Response:\n%v\n", response)
	data := response.Marshal()
	w.Write(data)
}