package main

import (
	"bytes"
	"encoding/json"
	"etrismartfarmpoccontroller/constants"
	"etrismartfarmpoccontroller/devicemanager"
	"etrismartfarmpoccontroller/router"
	"fmt"
	"io/ioutil"
	"net/http"
)

// func runBootstrap() {
// 	l, err := net.NewListenUDP("udp4", "", net.WithHeartBeat(time.Second*5))
// 	if err != nil {
// 		log.Println(err)
// 		return
// 	}
// 	defer l.Close()

// 	init := false
// 	minTimeout := time.Second * 5
// 	timeout := minTimeout

// 	d := func() {
// 		s := udp.NewServer(udp.WithTransmission(time.Second, timeout/2, 2))
// 		var wg sync.WaitGroup
// 		defer wg.Wait()
// 		defer s.Stop()
// 		wg.Add(1)
// 		go func() {
// 			defer wg.Done()
// 			s.Serve(l)
// 		}()

// 		ctx, cancel := context.WithTimeout(context.Background(), timeout)
// 		defer cancel()

// 		req, err := client.NewGetRequest(ctx, "/bs") /* msg.Option{
// 			ID:    msg.URIQuery,
// 			Value: []byte("rt=oic.wk.d"),
// 		}*/
// 		if err != nil {
// 			panic(fmt.Errorf("cannot create discover request: %w", err))
// 		}

// 		req.SetMessageID(udpmessage.GetMID())
// 		req.SetType(udpmessage.NonConfirmable)
// 		defer pool.ReleaseMessage(req)

// 		err = s.DiscoveryRequest(req, bootstrapAddr, func(cc *client.ClientConn, resp *pool.Message) {
// 			b, err := ioutil.ReadAll(resp.Body())
// 			if err != nil {
// 				panic(err)
// 			}
// 			fmt.Println(string(b))
// 			init = true
// 		})
// 		if err != nil {
// 			panic(err)
// 		}
// 	}
// 	for {
// 		d()
// 		if init {
// 			break
// 		}
// 	}
// }

func register() error {
	// Controller 이름을 읽어옴
	payload := map[string]string{}
	payload["cname"] = constants.Config["cname"]
	fmt.Println(constants.Config["cname"])
	b, err := json.Marshal(payload)
	if err != nil {
		return err
	}

	// Controller 등록 메시지 송신
	resp, err := http.Post(
		fmt.Sprintf("http://%s/%s", constants.Config["serverAddr"], "controllers"),
		"application/json",
		bytes.NewReader(b),
	)

	if err != nil {
		return err
	}

	// 응답 메시지 수신
	b, err = ioutil.ReadAll(resp.Body)
	if err != nil {
		return err
	}

	json.Unmarshal(b, &payload)

	// 등록 후 생성된 Controller ID 저장
	constants.Set("cid", payload["cid"])

	return nil
}

func main() {

	if constants.Config["cid"] == "" {
		err := register()
		if err != nil {
			panic(err)
		}
	}

	run, _ := devicemanager.NewManager()
	run()

	http.ListenAndServe(":4000", router.NewRouter())
}