package agent

import "gitlink.org.cn/cloudream/common/pkgs/mq"

type ObjectService interface {
	StartPinningObject(msg *StartPinningObject) (*StartPinningObjectResp, *mq.CodeMessage)
	WaitPinningObject(msg *WaitPinningObject) (*WaitPinningObjectResp, *mq.CodeMessage)
}

// 启动Pin对象的任务
var _ = Register(Service.StartPinningObject)

type StartPinningObject struct {
	mq.MessageBodyBase
	FileHash string `json:"fileHash"`
}
type StartPinningObjectResp struct {
	mq.MessageBodyBase
	TaskID string `json:"taskID"`
}

func NewStartPinningObject(fileHash string) *StartPinningObject {
	return &StartPinningObject{
		FileHash: fileHash,
	}
}
func NewStartPinningObjectResp(taskID string) *StartPinningObjectResp {
	return &StartPinningObjectResp{
		TaskID: taskID,
	}
}
func (client *Client) StartPinningObject(msg *StartPinningObject, opts ...mq.RequestOption) (*StartPinningObjectResp, error) {
	return mq.Request(Service.StartPinningObject, client.rabbitCli, msg, opts...)
}

// 等待Pin对象的任务
var _ = Register(Service.WaitPinningObject)

type WaitPinningObject struct {
	mq.MessageBodyBase
	TaskID        string `json:"taskID"`
	WaitTimeoutMs int64  `json:"waitTimeout"`
}
type WaitPinningObjectResp struct {
	mq.MessageBodyBase
	IsComplete bool   `json:"isComplete"`
	Error      string `json:"error"`
}

func NewWaitPinningObject(taskID string, waitTimeoutMs int64) *WaitPinningObject {
	return &WaitPinningObject{
		TaskID:        taskID,
		WaitTimeoutMs: waitTimeoutMs,
	}
}
func NewWaitPinningObjectResp(isComplete bool, err string) *WaitPinningObjectResp {
	return &WaitPinningObjectResp{
		IsComplete: isComplete,
		Error:      err,
	}
}
func (client *Client) WaitPinningObject(msg *WaitPinningObject, opts ...mq.RequestOption) (*WaitPinningObjectResp, error) {
	return mq.Request(Service.WaitPinningObject, client.rabbitCli, msg, opts...)
}