hunter/context.go

package hunter

import "github.com/474420502/requests"

type IHunt interface {
	Hunt() (requests.IResponse, error)
}

// TaskContext 上下文
type TaskContext struct {
	hunter   *Hunter
	workflow *requests.Workflow

	parent  ITaskNode
	current ITaskNode
	autoid  int
}

// NewContext 任务上下文
func NewContext() *TaskContext {
	return &TaskContext{}
}

// AddTask 添加到当前子任务队列
func (cxt *TaskContext) AddTask(itask ITask) {
	if children := cxt.current.Children(); children == nil {
		cxt.current.SetChildren(cxt.hunter.createQueue())
	}
	bt := &BaseTask{task: itask}
	cxt.current.Children().Push(bt)
}

// AddParentTask 添加到当前任务队列
func (cxt *TaskContext) AddParentTask(itask ITask) {

	bt := &BaseTask{task: itask}
	cxt.current.Parent().Children().Push(bt)
}

// GetShare 获取share的数据, 存储用的
func (cxt *TaskContext) GetShare(key string) interface{} {
	if v, ok := cxt.hunter.share[key]; ok {
		return v
	}
	return nil
}

// SetShare 设置share的数据, 存储用的
func (cxt *TaskContext) SetShare(key string, value interface{}) {
	cxt.hunter.share[key] = value
}

// Session Get return session *requests.Session
func (cxt *TaskContext) Session() *requests.Session {
	if cxt.hunter.Session() == nil {
		cxt.hunter.SetSession(requests.NewSession())
	}
	return cxt.hunter.Session()
}

// Workflow Get return Workflow *requests.Workflow. not exists, return nil
func (cxt *TaskContext) Workflow() *requests.Workflow {
	return cxt.workflow
}

// SetWorkflow Set Workflow *requests.Workflow
func (cxt *TaskContext) SetWorkflow(workflow *requests.Workflow) {
	cxt.workflow = workflow
}

// TaskID Get Task ID
func (cxt *TaskContext) TaskID() string {
	return cxt.current.TaskID()
}

// Path curren  Task tree path.
func (cxt *TaskContext) Path() string {
	return cxt.current.Path()
}

// GetHunter 获取share的数据, 存储用的
func (cxt *TaskContext) GetHunter() *Hunter {
	return cxt.hunter
}

// Hunt Hunt() = cxt.Workflow().Execute()
func (cxt *TaskContext) Hunt() (requests.IResponse, error) {
	if ihunt, ok := cxt.current.Task().(IHunt); ok {
		return ihunt.Hunt()
	}
	return cxt.workflow.Execute()
}