package controllers

import (
	"encoding/json"
	"errors"
	"fmt"
	"strings"

	oclib "cloud.o-forge.io/core/oc-lib"
	"cloud.o-forge.io/core/oc-lib/models/workflow"
	"cloud.o-forge.io/core/oc-lib/tools"
	beego "github.com/beego/beego/v2/server/web"
)

// stringReadCloser wraps a strings.Reader to satisfy the multipart.File interface.
type stringReadCloser struct {
	*strings.Reader
}

func (s *stringReadCloser) Close() error { return nil }

// Operations about workflow
type WorkflowController struct {
	beego.Controller
}

var logger = oclib.GetLogger()

var paths = map[tools.DataType]map[tools.METHOD]string{ // paths to call other OC services
	tools.BOOKING: {
		tools.POST:   "/booking/",
		tools.DELETE: "/booking/:id",
	},
	tools.PEER: {
		tools.POST: "/status/",
	},
	tools.WORKFLOW: {
		tools.PUT:    "/:id?is_remote=true",
		tools.DELETE: "/:id?is_remote=true",
	},
}

// @Title Search
// @Description search workspace
// @Param	search		path 	string	true		"the word search you want to get"
// @Param	is_draft	query 	string	false		"draft wished"
// @Success 200 {workspace} models.workspace
// @router /search/:search [get]
func (o *WorkflowController) Search() {
	// user, peerID, groups := oclib.ExtractTokenInfo(*o.Ctx.Request)
	// store and return Id or post with UUID
	search := o.Ctx.Input.Param(":search")
	isDraft := o.Ctx.Input.Query("is_draft")
	// o.Data["json"] = oclib.NewRequest(oclib.LibDataEnum(oclib.WORKFLOW), user, peerID, groups, nil).Search(nil, search, isDraft == "true")
	o.Data["json"] = oclib.NewRequestAdmin(oclib.LibDataEnum(oclib.WORKFLOW), nil).Search(nil, search, isDraft == "true")

	o.ServeJSON()
}

// @Title Update
// @Description create workflows
// @Param	id		path 	string	true		"the workflowid you want to get"
// @Param	body		body 	models.workflow	true		"The workflow content"
// @Success 200 {object} models.workflow
// @router /:id [put]
func (o *WorkflowController) Put() {
	user, _, _ := oclib.ExtractTokenInfo(*o.Ctx.Request)
	// store and return Id or post with UUID
	var res map[string]interface{}
	id := o.Ctx.Input.Param(":id")
	json.Unmarshal(o.Ctx.Input.CopyBody(10000000), &res)
	fmt.Println("res", res["links"])
	caller := tools.NewHTTPCaller(paths) // create a new HTTP caller
	caller.Disabled = oclib.IsQueryParamsEquals(o.Ctx.Input, "is_remote", true)
	o.Ctx.Input.Param("is_remote")
	// data := oclib.NewRequest(oclib.LibDataEnum(oclib.WORKFLOW), user, peerID, groups, caller).UpdateOne(res, id)
	data := oclib.NewRequestAdmin(oclib.LibDataEnum(oclib.WORKFLOW), caller).UpdateOne(res, id)
	if wf := data.ToWorkflow(); wf != nil {
		EmitNATS(user, tools.CREATE_RESOURCE, wf)
	}
	o.Data["json"] = data
	o.ServeJSON()
}

// @Title Create
// @Description create workflows
// @Param	data		body 	json	true		"body for data content (Json format)"
// @Success 200 {object} models.workflow
// @router / [post]
func (o *WorkflowController) Post() {
	//  user, peerID, groups := oclib.ExtractTokenInfo(*o.Ctx.Request)
	var res map[string]interface{}
	json.Unmarshal(o.Ctx.Input.CopyBody(10000000), &res)
	caller := tools.NewHTTPCaller(paths) // create a new HTTP caller
	caller.Disabled = oclib.IsQueryParamsEquals(o.Ctx.Input, "is_remote", true)
	// data := oclib.NewRequest(oclib.LibDataEnum(oclib.WORKFLOW), user, peerID, groups, caller).StoreOne(res)
	data := oclib.NewRequestAdmin(oclib.LibDataEnum(oclib.WORKFLOW), caller).StoreOne(res)
	o.Data["json"] = data
	o.ServeJSON()
}

// @Title Publish
// @Description create workflows
// @Param	id		path 	string	true		"the workflowid you want to get"
// @Success 200 {object} models.workflow
// @router /publish/:id [post]
func (o *WorkflowController) Publish() {
	user, peerID, groups := oclib.ExtractTokenInfo(*o.Ctx.Request)
	id := o.Ctx.Input.Param(":id")
	data := oclib.NewRequest(oclib.LibDataEnum(oclib.WORKFLOW), user, peerID, groups, nil).LoadOne(id)
	if data.Data != nil { // copy existing workflow as a resource
		o.Data["json"] = oclib.NewRequest(oclib.LibDataEnum(oclib.WORKFLOW), user, peerID, groups, nil).CopyOne(data.Data.Serialize(data.Data))
	} else {
		o.Data["json"] = data
	}
	o.ServeJSON()
}

// @Title GetAll
// @Description find workflow by workflowid
// @Param	is_draft	query 	string	false		"draft wished"
// @Success 200 {workflow} models.workflow
// @router / [get]
func (o *WorkflowController) GetAll() {
	// user, peerID, groups := oclib.ExtractTokenInfo(*o.Ctx.Request)
	isDraft := o.Ctx.Input.Query("is_draft")
	// o.Data["json"] = oclib.NewRequest(oclib.LibDataEnum(oclib.WORKFLOW), user, peerID, groups, nil).LoadAll(isDraft == "true")

	o.Data["json"] = oclib.NewRequestAdmin(oclib.LibDataEnum(oclib.WORKFLOW), nil).LoadAll(isDraft == "true")
	o.ServeJSON()
}

// @Title Get
// @Description find workflow by workflowid
// @Param	id		path 	string	true		"the workflowid you want to get"
// @Success 200 {workflow} models.workflow
// @router /:id [get]
func (o *WorkflowController) Get() {
	// user, peerID, groups := oclib.ExtractTokenInfo(*o.Ctx.Request)
	id := o.Ctx.Input.Param(":id")
	caller := tools.NewHTTPCaller(paths) // create a new HTTP caller
	caller.Disabled = oclib.IsQueryParamsEquals(o.Ctx.Input, "is_remote", true)
	// .Data["json"] = oclib.NewRequest(oclib.LibDataEnum(oclib.WORKFLOW), user, peerID, groups, caller).LoadOne(id)
	o.Data["json"] = oclib.NewRequestAdmin(oclib.LibDataEnum(oclib.WORKFLOW), caller).LoadOne(id)

	o.ServeJSON()
}

// @Title Delete
// @Description delete the workflow
// @Param	id		path 	string	true		"The workflowId you want to delete"
// @Success 200 {string} delete success!
// @router /:id [delete]
func (o *WorkflowController) Delete() {
	user, peerID, groups := oclib.ExtractTokenInfo(*o.Ctx.Request)
	id := o.Ctx.Input.Param(":id")
	caller := tools.NewHTTPCaller(paths) // create a new HTTP caller
	caller.Disabled = oclib.IsQueryParamsEquals(o.Ctx.Input, "is_remote", true)
	data := oclib.NewRequest(oclib.LibDataEnum(oclib.WORKFLOW), user, peerID, groups, caller).DeleteOne(id)
	if wf := data.ToWorkflow(); wf != nil {
		EmitNATS(user, tools.REMOVE_RESOURCE, wf)
	}
	o.Data["json"] = data
	o.ServeJSON()
}

// @Title Check
// @Description check booking
// @Param	id				path 	string		"the booking workflow id"
// @Param	start_date		path 	string		"the booking start date format "2006-01-02T15:04:05"
// @Param	end_date		path 	string		"the booking end date" format 2006-01-02T15:04:05"
// @Success 200 {object} models.object
// @router /check/:id/:start_date/:end_date [get]
func (o *WorkflowController) Check() {
	// store and return Id or post with UUID
	/*
	 * Check if the booking is available
	 */
	id := o.Ctx.Input.Param(":id")
	if id == "" {
		o.Data["json"] = map[string]interface{}{
			"data": map[string]interface{}{
				"is_available": false,
			},
			"code":  400,
			"error": errors.New("invalid date format"),
		}
	} else {
		// user, peerID, groups := oclib.ExtractTokenInfo(*o.Ctx.Request)
		//res := oclib.NewRequest(oclib.LibDataEnum(oclib.WORKFLOW), user, peerID, groups, nil).LoadOne(id)
		res := oclib.NewRequestAdmin(oclib.LibDataEnum(oclib.WORKFLOW), nil).LoadOne(id)

		if res.Data != nil {
			workflow := res.ToWorkflow()
			caller := tools.NewHTTPCaller(map[tools.DataType]map[tools.METHOD]string{ // paths to call other OC services
				tools.PEER: {
					tools.POST: "/status/",
				},
				tools.BOOKING: {
					tools.GET: "/booking/check/:id/" + o.Ctx.Input.Param(":start_date") + "/" + o.Ctx.Input.Param(":end_date"),
				},
			})
			isAvailable, err := workflow.CheckBooking(caller) // check booking
			fmt.Println(err)
			logger.Info().Msg(fmt.Sprintf("Result of availability check between %v and %v : %v", o.Ctx.Input.Param(":start_date"), o.Ctx.Input.Param(":end_date"), isAvailable))
			code := 200
			if !isAvailable { // if not available then its a conflict
				logger.Info().Msg("Returning code 409 after checking availability, error value :" + fmt.Sprint(err))
				code = 409
			}
			o.Data["json"] = map[string]interface{}{
				"data": map[string]interface{}{
					"is_available": isAvailable,
				},
				"code":  code,
				"error": err,
			}
		} else {
			o.Data["json"] = res
		}
	}
	o.ServeJSON()
}

// @Title PostPlantUML
// @Description parse plantuml text and return the formed workflow object
// @Param body body string true "PlantUML text content"
// @Success 200 {object} models.workflow
// @Failure 406 {string} string "Bad request"
// @router /plantuml [post]
func (o *WorkflowController) PostPlantUML() {
	user, peerID, groups := oclib.ExtractTokenInfo(*o.Ctx.Request)
	body := o.Ctx.Input.CopyBody(1000000)
	req := &tools.APIRequest{Username: user, PeerID: peerID, Groups: groups}
	newWorkflow := &workflow.Workflow{}
	reader := &stringReadCloser{strings.NewReader(string(body))}
	wf, err := newWorkflow.ExtractFromPlantUML(reader, req)
	if err != nil {
		o.Data["json"] = map[string]interface{}{"data": nil, "code": 406, "error": err.Error()}
		o.ServeJSON()
		return
	}
	o.Data["json"] = map[string]interface{}{"data": wf, "code": 200, "error": nil}
	o.ServeJSON()
}

// @Title GetPlantUML
// @Description export a workflow as plantuml text
// @Param id path string true "the workflow id to export"
// @Success 200 {string} string "PlantUML text"
// @Failure 404 {string} string "Not found"
// @router /plantuml/:id [get]
func (o *WorkflowController) GetPlantUML() {
	id := o.Ctx.Input.Param(":id")
	res := oclib.NewRequestAdmin(oclib.LibDataEnum(oclib.WORKFLOW), nil).LoadOne(id)
	wf := res.ToWorkflow()
	if wf == nil {
		o.Data["json"] = map[string]interface{}{"data": nil, "code": 404, "error": "workflow not found"}
		o.ServeJSON()
		return
	}
	o.Ctx.Output.Header("Content-Type", "text/plain; charset=utf-8")
	o.Ctx.Output.Body([]byte(wf.ToPlantUML()))
}

func EmitNATS(user string, method tools.NATSMethod, wf *workflow.Workflow) {
	if b, err := json.Marshal(wf); err == nil {
		tools.NewNATSCaller().SetNATSPub(method, tools.NATSResponse{
			FromApp:  "oc-catalog",
			Datatype: tools.WORKFLOW,
			Method:   int(method),
			Payload:  b,
		})
	}
}