package main

import (
	"context"
	"encoding/json"
	"fmt"
	"io"
	"net/http"
	"time"

	"github.com/oarkflow/mq"
	"github.com/oarkflow/mq/dag"
	"github.com/oarkflow/mq/examples/tasks"
)

var d *dag.DAG

func main() {
	d = dag.New(mq.WithSyncMode(true), mq.WithTLS(true, "./certs/server.crt", "./certs/server.key"), mq.WithCAPath("./certs/ca.crt"))
	d.AddNode("queue1", tasks.Node1, true)
	d.AddNode("queue2", tasks.Node2)
	d.AddNode("queue3", tasks.Node3)
	d.AddNode("queue4", tasks.Node4)

	d.AddNode("queue5", tasks.CheckCondition)
	d.AddNode("queue6", tasks.Pass)
	d.AddNode("queue7", tasks.Fail)

	d.AddCondition("queue5", map[string]string{"pass": "queue6", "fail": "queue7"})
	d.AddEdge("queue1", "queue2")
	d.AddEdge("queue2", "queue4")
	d.AddEdge("queue3", "queue5")

	d.AddLoop("queue2", "queue3")
	d.Prepare()
	go func() {
		d.Start(context.Background(), ":8081")
	}()
	go func() {
		time.Sleep(3 * time.Second)
		result := d.Send(context.Background(), []byte(`[{"user_id": 1}, {"user_id": 2}]`))
		if result.Error != nil {
			panic(result.Error)
		}
		fmt.Println("Response", string(result.Payload))
	}()

	time.Sleep(10 * time.Second)
	/*d.Prepare()
	http.HandleFunc("POST /publish", requestHandler("publish"))
	http.HandleFunc("POST /request", requestHandler("request"))
	err := d.Start(context.TODO(), ":8083")
	if err != nil {
		panic(err)
	}*/
}

func requestHandler(requestType string) func(w http.ResponseWriter, r *http.Request) {
	return func(w http.ResponseWriter, r *http.Request) {
		if r.Method != http.MethodPost {
			http.Error(w, "Invalid request method", http.StatusMethodNotAllowed)
			return
		}
		var payload []byte
		if r.Body != nil {
			defer r.Body.Close()
			var err error
			payload, err = io.ReadAll(r.Body)
			if err != nil {
				http.Error(w, "Failed to read request body", http.StatusBadRequest)
				return
			}
		} else {
			http.Error(w, "Empty request body", http.StatusBadRequest)
			return
		}
		var rs mq.Result
		if requestType == "request" {
			rs = d.Request(context.Background(), payload)
		} else {
			rs = d.Send(context.Background(), payload)
		}
		w.Header().Set("Content-Type", "application/json")
		result := map[string]any{
			"message_id": rs.TaskID,
			"payload":    string(rs.Payload),
			"error":      rs.Error,
		}
		json.NewEncoder(w).Encode(result)
	}
}