// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package service

import (
	"encoding/json"
	"fmt"

	"github.com/EtixLabs/cameradar"
	"github.com/EtixLabs/cameradar/server/adaptor/jsonrpc2"
	v "gopkg.in/go-playground/validator.v9"
)

func (c *Cameradar) handleRequest(message string) {
	var ret []cmrdr.Stream
	var JSONRPCErr jsonrpc2.Error

	var request jsonrpc2.Request
	err := json.Unmarshal([]byte(message), &request)
	if err != nil {
		JSONRPCErr = jsonrpc2.Error{
			Code:    jsonrpc2.ParseError,
			Message: jsonrpc2.ParseErrorMessage,
			Data:    err.Error(),
		}
	}

	validate := v.New()
	err = validate.Struct(request)
	if err != nil {
		JSONRPCErr = jsonrpc2.Error{
			Code:    jsonrpc2.InvalidRequest,
			Message: jsonrpc2.InvalidRequestMessage,
			Data:    err.Error(),
		}
	}

	var options Options
	err = json.Unmarshal([]byte(request.Params), &options)
	if err != nil {
		JSONRPCErr = jsonrpc2.Error{
			Code:    jsonrpc2.InvalidParams,
			Message: jsonrpc2.InvalidParamsMessage,
			Data:    err.Error(),
		}
	}

	c.SetOptions(options)

	switch request.Method {
	case "discover":
		ret, err = c.Discover()
	case "attack_credentials":
		ret, err = c.Discover()
	case "attack_routes":
		ret, err = c.Discover()
	case "discover_and_attack":
		ret, err = c.DiscoverAndAttack()
	default:
		JSONRPCErr = jsonrpc2.Error{
			Code:    jsonrpc2.MethodNotFound,
			Message: jsonrpc2.MethodNotFoundMessage,
			Data:    err.Error(),
		}
	}
	if err != nil {
		JSONRPCErr = jsonrpc2.Error{
			Code:    jsonrpc2.InternalError,
			Message: jsonrpc2.InternalErrorMessage,
			Data:    err.Error(),
		}
	}

	result, err := json.Marshal(ret)
	if err != nil {
		JSONRPCErr = jsonrpc2.Error{
			Code:    jsonrpc2.InternalError,
			Message: jsonrpc2.InternalErrorMessage,
			Data:    err.Error(),
		}
	}
	c.respondToClient(string(result), request.ID, JSONRPCErr)
}

func (c *Cameradar) respondToClient(result, ID string, JSONRPCErr jsonrpc2.Error) {
	println(result)
	r := jsonrpc2.Response{
		JSONRPC: "2.0",
		Result:  result,
		Error:   JSONRPCErr,
		ID:      ID,
	}

	response, err := json.Marshal(r)
	if err != nil {
		c.toClient <- "{\"jsonrpc\": \"2.0\",\"result\":null,\"error\":{\"code\":" + fmt.Sprint(jsonrpc2.InternalError) + ",\"" + jsonrpc2.InternalErrorMessage + "\",\"data\":\"could not marshal response\"},\"id\":\"" + ID + "\"}"
	}

	c.toClient <- string(response)
}