// Copyright (C) 2014-2018 Goodrain Co., Ltd.
// RAINBOND, Application Management Platform

// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version. For any non-GPL usage of Rainbond,
// one or multiple Commercial Licenses authorized by Goodrain Co., Ltd.
// must be obtained first.

// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.

package client

import (
	"encoding/json"
	"fmt"
	"time"

	"github.com/goodrain/rainbond/mq/api/grpc/pb"
	"github.com/sirupsen/logrus"
	context "golang.org/x/net/context"
	grpc "google.golang.org/grpc"
	"google.golang.org/grpc/backoff"
	"google.golang.org/grpc/keepalive"
)

// BuilderTopic builder for linux
var BuilderTopic = "builder"
var BuilderHealth = "builder-health"

// WindowsBuilderTopic builder for windows
var WindowsBuilderTopic = "windows_builder"

// WorkerTopic worker topic
var WorkerTopic = "worker"
var WorkerHealth = "worker-health"

// SourceScanTopic source code scan topic
var SourceScanTopic = "source-scan"

// MQClient mq  client
type MQClient interface {
	pb.TaskQueueClient
	Close()
	SendBuilderTopic(t TaskStruct) error
}

type mqClient struct {
	pb.TaskQueueClient
	ctx    context.Context
	cancel context.CancelFunc
}

// NewMqClient new a mq client
func NewMqClient(mqAddr string) (MQClient, error) {
	ctx, cancel := context.WithCancel(context.Background())

	// 配置 keepalive 参数，防止长时间空闲后连接被中间网络设备断开
	kaParams := keepalive.ClientParameters{
		Time:                10 * time.Second, // 每10秒发送一次 keepalive ping
		Timeout:             3 * time.Second,  // 等待 keepalive ping 响应的超时时间
		PermitWithoutStream: true,             // 即使没有活跃的 stream 也发送 keepalive
	}

	// 配置 backoff 参数，加快服务重启后的重连速度
	backoffConfig := backoff.Config{
		BaseDelay:  1 * time.Second,  // 初始重试延迟
		Multiplier: 1.5,              // 指数退避倍数
		Jitter:     0.2,              // 20% 随机抖动
		MaxDelay:   10 * time.Second, // 最大重试延迟（默认 120s 太长）
	}

	conn, err := grpc.DialContext(ctx, mqAddr,
		grpc.WithInsecure(),
		grpc.WithKeepaliveParams(kaParams),
		grpc.WithConnectParams(grpc.ConnectParams{
			Backoff:           backoffConfig,
			MinConnectTimeout: 5 * time.Second,
		}),
	)
	if err != nil {
		cancel()
		return nil, err
	}
	cli := pb.NewTaskQueueClient(conn)
	client := &mqClient{
		ctx:    ctx,
		cancel: cancel,
	}
	client.TaskQueueClient = cli
	return client, nil
}

// Close mq grpc client must be closed after uesd
func (m *mqClient) Close() {
	m.cancel()
}

// TaskStruct task struct
type TaskStruct struct {
	Topic    string
	Arch     string
	TaskType string
	TaskBody interface{}
}

// buildTask build task
func buildTask(t TaskStruct) (*pb.EnqueueRequest, error) {
	var er pb.EnqueueRequest
	taskJSON, err := json.Marshal(t.TaskBody)
	if err != nil {
		logrus.Errorf("tran task json error")
		return &er, err
	}
	er.Topic = t.Topic
	er.Message = &pb.TaskMessage{
		TaskType:   t.TaskType,
		CreateTime: time.Now().Format(time.RFC3339),
		TaskBody:   taskJSON,
		User:       "rainbond",
		Arch:       t.Arch,
	}
	return &er, nil
}

// SendBuilderTopic -
func (m *mqClient) SendBuilderTopic(t TaskStruct) error {
	request, err := buildTask(t)
	if err != nil {
		return fmt.Errorf("create task body error %s", err.Error())
	}
	ctx, cancel := context.WithTimeout(m.ctx, time.Second*5)
	defer cancel()
	_, err = m.TaskQueueClient.Enqueue(ctx, request)
	if err != nil {
		return fmt.Errorf("send enqueue request error %s", err.Error())
	}
	return nil
}