// Copyright 2022 PingCAP, Inc.
//
// 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,
// See the License for the specific language governing permissions and
// limitations under the License.

package main

import (
	"context"
	"net/http"

	"github.com/gorilla/mux"
	"github.com/pingcap/fn"
	"github.com/pingcap/log"
	"go.uber.org/zap"

	"github.com/pingcap-inc/compute-engine/pkg/service"
)

type TriggerRequest struct {
	Hook string                 `json:"hook"`
	Tags map[string]interface{} `json:"tags"`
	Args map[string]interface{} `json:"args"`
}

type TriggerResponse struct {
	ID   string                 `json:"id"`
	Data map[string]interface{} `json:"data"`
}

func TriggerTaskSync(ctx context.Context, req *http.Request, params *TriggerRequest) (*TriggerResponse, error) {
	task := mux.Vars(req)["task"]
	s := ctx.Value(InstanceKey).(service.Scheduler)
	id, data, err := s.TriggerSync(ctx, task, params.Tags, params.Args)
	if err != nil {
		log.Error("trigger task error", zap.Error(err))
		return nil, fn.ErrorWithStatusCode(err, http.StatusInternalServerError)
	}
	return &TriggerResponse{
		ID:   id,
		Data: data,
	}, nil
}

func RetryTaskSync(ctx context.Context, req *http.Request) (*TriggerResponse, error) {
	id := mux.Vars(req)["instance"]
	s := ctx.Value(InstanceKey).(service.Scheduler)
	id, data, err := s.RetrySync(ctx, id)
	if err != nil {
		log.Error("retry task error", zap.Error(err))
		return nil, fn.ErrorWithStatusCode(err, http.StatusInternalServerError)
	}
	return &TriggerResponse{
		ID:   id,
		Data: data,
	}, nil
}

func TriggerTaskAsync(ctx context.Context, req *http.Request, params *TriggerRequest) (*TriggerResponse, error) {
	task := mux.Vars(req)["task"]
	s := ctx.Value(InstanceKey).(service.Scheduler)
	id, err := s.TriggerAsync(ctx, task, params.Hook, params.Tags, params.Args)
	if err != nil {
		log.Error("trigger task error", zap.Error(err))
		return nil, fn.ErrorWithStatusCode(err, http.StatusInternalServerError)
	}
	return &TriggerResponse{
		ID: id,
	}, nil
}

func RetryTaskAsync(ctx context.Context, req *http.Request) (*TriggerResponse, error) {
	id := mux.Vars(req)["instance"]
	s := ctx.Value(InstanceKey).(service.Scheduler)
	id, err := s.RetryAsync(ctx, id)
	if err != nil {
		log.Error("retry task error", zap.Error(err))
		return nil, fn.ErrorWithStatusCode(err, http.StatusInternalServerError)
	}
	return &TriggerResponse{
		ID: id,
	}, nil
}