controllers

type AccessLogController struct {
	BaseController
}

type BaseController struct {
	web.Controller
	IsServerAPI bool //server工作模式是否是api方式
}

type ConfigController struct {
	BaseController
}

type DashboardController struct {
	BaseController
}

type DashboardStatisticData struct {
	IP            int `json:"ip_count"`
	Domain        int `json:"domain_count"`
	Vulnerability int `json:"vulnerability_count"`
	ActiveTask    int `json:"task_active"`
	Worker        int `json:"worker_count"`
}

type DataTableResponseData struct {
	Draw            int           `json:"draw"`
	RecordsTotal    int           `json:"recordsTotal"`
	RecordsFiltered int           `json:"recordsFiltered"`
	Data            []interface{} `json:"data"`
}

type DatableRequestParam struct {
	Draw   int `form:"draw"`
	Start  int `form:"start"`
	Length int `form:"length"`
}

type DefaultConfig struct {
	//portscan
	CmdBin string `json:"cmdbin" form:"cmdbin"`
	Port   string `json:"port" form:"port"`
	Rate   int    `json:"rate" form:"rate"`
	Tech   string `json:"tech" form:"tech"`
	IsPing bool   `json:"ping" form:"ping"`
	//task
	IpSliceNumber   int    `json:"ipslicenumber" form:"ipslicenumber"`
	PortSliceNumber int    `json:"portslicenumber" form:"portslicenumber"`
	Version         string `json:"version" form:"version"`
	TaskWorkspace   string `json:"taskworkspace" form:"taskworkspace"`
	//fingerprint
	IsHttpx          bool `json:"httpx" form:"httpx"`
	IsScreenshot     bool `json:"screenshot" form:"screenshot"`
	IsFingerprintHub bool `json:"fingerprinthub" form:"fingerprinthub"`
	IsIconHash       bool `json:"iconhash" form:"iconhash"`
	IsFingerprintx   bool `json:"fingerprintx" form:"fingerprintx"`
	// onlineapi
	IsFofa           bool   `json:"fofa" form:"fofa"`
	IsQuake          bool   `json:"quake" form:"quake"`
	IsHunter         bool   `json:"hunter" form:"hunter"`
	ServerChanToken  string `json:"serverchan" form:"serverchan"`
	DingTalkToken    string `json:"dingtalk" form:"dingtalk"`
	FeishuToken      string `json:"feishu" form:"feishu"`
	FofaToken        string `json:"fofatoken" form:"fofatoken"`
	HunterToken      string `json:"huntertoken" form:"huntertoken"`
	QuakeToken       string `json:"quaketoken" form:"quaketoken"`
	ChinazToken      string `json:"chinaztoken" form:"chinaztoken"`
	SearchPageSize   int    `json:"pagesize" form:"pagesize"`
	SearchLimitCount int    `json:"limitcount" form:"limitcount"`
	// domainscan
	Wordlist           string `json:"wordlist" form:"wordlist"`
	IsSubDomainFinder  bool   `json:"subfinder" form:"subfinder"`
	IsSubDomainBrute   bool   `json:"subdomainbrute" form:"subdomainbrute"`
	IsSubDomainCrawler bool   `json:"subdomaincrawler" form:"subdomaincrawler"`
	IsIgnoreCDN        bool   `json:"ignorecdn" form:"ignorecdn"`
	IsIgnoreOutofChina bool   `json:"ignoreoutofchina" form:"ignoreoutofchina"`
	IsPortscan         bool   `json:"portscan" form:"portscan"`
	IsWhois            bool   `json:"whois" form:"whois"`
	IsICP              bool   `json:"icp" form:"icp"`
	// proxy
	ProxyList string `json:"proxyList" form:"proxyList"`
	// wiki:feishu
	FeishuAppId        string `json:"feishuappid" form:"feishuappid"`
	FeishuAppSecret    string `json:"feishusecret" form:"feishuappsecret"`
	FeishuRefreshToken string `json:"feishurefreshtoken" form:"feishurefreshtoken"`
}

type DocumentInfo struct {
	Id             int    `json:"id"`
	Index          int    `json:"index"`
	Title          string `json:"title"`
	SpaceId        string `json:"space_id"`
	NodeToken      string `json:"node_token"`
	Comment        string `json:"comment"`
	PinIndex       int    `json:"pin_index"`
	ExportPathFile string `json:"export"`
	CreateTime     string `json:"create_time"`
	UpdateTime     string `json:"update_time"`
}

type DomainAttrFullInfo struct {
	IP            map[string]struct{}
	TitleSet      map[string]int
	BannerSet     map[string]int
	FingerSet     map[string]struct{}
	StatusCodeSet map[string]struct{}
	DomainAttr    []DomainAttrInfo
	IconImageSet  map[string]string
	TlsData       map[string]struct{}
	SourceSet     map[string]struct{}
	DomainCDN     string
	DomainCNAME   string
}

type DomainAttrInfo struct {
	Id         int    `json:"id"`
	DomainId   int    `json:"domainId"`
	Port       int    `json:"port"`
	Tag        string `json:"tag"`
	Content    string `json:"content"`
	CreateTime string `json:"create_datetime"`
	UpdateTime string `json:"update_datetime"`
}

type DomainController struct {
	BaseController
}

type DomainExportInfo struct {
	Domain      string
	IP          []string
	Port        []string
	StatusCode  []string
	Title       []string
	Finger      []string
	Banner      []string
	TlsData     []string
	Source      []string
	IsIPCDN     bool
	DomainCDN   string
	DomainCNAME string
}

type DomainInfo struct {
	Id            int
	Domain        string
	Organization  string
	IP            []string
	Port          []int
	PortAttr      []PortAttrInfo
	Finger        []string
	StatusCode    []string
	Title         map[string]int
	Banner        map[string]int
	TitleString   string
	BannerString  string
	ColorTag      string
	Memo          string
	Vulnerability []VulnerabilityInfo
	CreateTime    string
	UpdateTime    string
	Screenshot    []ScreenshotFileInfo
	DomainAttr    []DomainAttrInfo
	DisableFofa   bool
	IconHashes    []IconHashWithFofa
	TlsData       []string
	DomainCDN     string
	DomainCNAME   string
	Workspace     string
	WorkspaceGUID string
	PinIndex      string
	Source        []string
	WikiDocs      []DocumentInfo
}

type DomainListData struct {
	Id             int            `json:"id"`
	Index          int            `json:"index"`
	FldDomain      string         `json:"fld_domain"`
	Domain         string         `json:"domain"`
	IP             []string       `json:"ip"`
	Port           []int          `json:"port"`
	StatusCode     []string       `json:"statuscode"`
	Title          map[string]int `json:"title"`
	Banner         map[string]int `json:"banner"`
	ColorTag       string         `json:"color_tag"`
	MemoContent    string         `json:"memo_content"`
	Vulnerability  string         `json:"vulnerability"`
	HoneyPot       string         `json:"honeypot"`
	ScreenshotFile []string       `json:"screenshot"`
	DomainCDN      string         `json:"domaincdn"`
	DomainCNAME    string         `json:"domaincname"`
	IsIPCDN        bool           `json:"ipcdn"`
	IconImage      []string       `json:"iconimage"`
	WorkspaceId    int            `json:"workspace"`
	WorkspaceGUID  string         `json:"workspace_guid"`
	PinIndex       int            `json:"pinindex"`
	WikiDocs       string         `json:"wiki_docs"`
}

type DomainStatisticInfo struct {
	Domain    map[string][]string
	DomainIP  map[string]string
	Subdomain map[string]int
	IP        map[string]int
	IPSubnet  map[string]int
}

type GlobalSessionData struct {
	IpAddressIp     string `json:"ip_address_ip"`
	IpAddressDomain string `json:"ip_address_domain"`
	DomainAddress   string `json:"domain_address"`
	Port            string `json:"port"`
	OrgId           string `json:"session_org_id"`
}

type IPController struct {
	BaseController
}

type IPExportInfo struct {
	IP         string
	Port       int
	Location   string
	StatusCode string
	TitleSet   map[string]struct{}
	BannerSet  map[string]struct{}
	FingerSet  map[string]struct{}
	TlsDataSet map[string]struct{}
	HttpxSet   map[string]struct{}
	SourceSet  map[string]struct{}
}

type IPInfo struct {
	Id            int
	IP            string
	IPFormatted   string
	Organization  string
	Status        string
	Location      string
	Port          []int
	Title         map[string]int
	Banner        map[string]int
	TitleString   string
	BannerString  string
	PortAttr      []PortAttrInfo
	Domain        []string
	ColorTag      string
	Memo          string
	Vulnerability []VulnerabilityInfo
	CreateTime    string
	UpdateTime    string
	Screenshot    []ScreenshotFileInfo
	DisableFofa   bool
	IconHashes    []IconHashWithFofa
	TlsData       []string
	Workspace     string
	WorkspaceGUID string
	PinIndex      string
	WikiDocs      []DocumentInfo
}

type IPListData struct {
	Id             int            `json:"id"`
	Index          int            `json:"index"`
	IP             string         `json:"ip"`
	IPFormatted    string         `json:"ipf"`
	Location       string         `json:"location"`
	Port           []string       `json:"port"`
	Title          map[string]int `json:"title"`
	Banner         map[string]int `json:"banner"`
	ColorTag       string         `json:"color_tag"`
	MemoContent    string         `json:"memo_content"`
	Vulnerability  string         `json:"vulnerability"`
	HoneyPot       string         `json:"honeypot"`
	ScreenshotFile []string       `json:"screenshot"`
	CloudName      string         `json:"cloudname"`
	IsCDN          bool           `json:"cdn"`
	IconImage      []string       `json:"iconimage"`
	WorkspaceId    int            `json:"workspace"`
	WorkspaceGUID  string         `json:"workspace_guid"`
	PinIndex       int            `json:"pinindex"`
	WikiDocs       string         `json:"wiki_docs"`
}

type IPStatisticInfo struct {
	IP       map[string]int
	IPSubnet map[string]int
	Port     map[string]int
	Location map[string]int
}

type IconHashWithFofa struct {
	IconHash  string
	IconImage string
	FofaUrl   string
}

type KeySearchController struct {
	BaseController
}

type KeyWordInfo struct {
	Id             int    `json:"id"`
	OrgId          int    `json:"org_id"`
	KeyWord        string `json:"key_word"`
	IsFofa         bool   `json:"fofa"`
	IsHunter       bool   `json:"hunter"`
	IsQuake        bool   `json:"quake"`
	SearchTime     string `json:"search_time"`
	ExcludeWords   string `json:"exclude_words"`
	CheckMod       string `json:"check_mod"`
	IsDelete       bool   `json:"is_delete"`
	Count          int    `json:"count"`
	CreateDatetime string `json:"create_datetime"`
	UpdateDatetime string `json:"update_datetime"`
}

type KeyWordList struct {
	Id             int    `json:"id"`
	Index          int    `json:"index"`
	OrgId          string `json:"org_id"`
	KeyWord        string `json:"key_word"`
	Engine         string `json:"engine"`
	SearchTime     string `json:"search_time"`
	ExcludeWords   string `json:"exclude_words"`
	CheckMod       string `json:"check_mod"`
	IsDelete       bool   `json:"is_delete"`
	Count          int    `json:"count"`
	CreateDatetime string `json:"create_datetime"`
	UpdateDatetime string `json:"update_datetime"`
	WorkspaceId    int    `json:"workspace"`
}

type LoginController struct {
	BaseController
}

type MyCustomClaims struct {
	TokenData
	jwt.RegisteredClaims
}

type OnlineUserInfo struct {
	IP           string
	LoginTime    time.Time
	UpdateTime   time.Time
	UpdateNumber int64
}

type OnlineUserInfoData struct {
	Index        int    `json:"index"`
	IP           string `json:"ip"`
	LoginTime    string `json:"login_time"`
	UpdateTime   string `json:"update_time"`
	UpdateNumber int64  `json:"update_number"`
}

type OrganizationController struct {
	BaseController
}

type OrganizationData struct {
	Id             int    `json:"id" form:"id"`
	Index          int    `json:"index" form:"-"`
	OrgName        string `json:"org_name" form:"org_name"`
	Status         string `json:"status" form:"status"`
	SortOrder      int    `json:"sort_order" form:"sort_order"`
	CreateDatetime string `json:"create_time" form:"-"`
	UpdateDatetime string `json:"update_time" form:"-"`
	WorkspaceId    int    `json:"workspace"`
}

type OrganizationSelectData struct {
	Id      int    `json:"id"`
	OrgName string `json:"name"`
}

type PortAttrInfo struct {
	Id                 int
	PortId             int
	IP                 string
	IPFormatted        string
	Port               string
	Tag                string
	Content            string
	Source             string
	FofaLink           string
	CreateTime         string
	UpdateTime         string
	TableBackgroundSet bool
}

type PortInfo struct {
	PortNumbers      []int
	PortStatus       map[int]string
	TitleSet         map[string]int
	BannerSet        map[string]int
	PortAttr         []PortAttrInfo
	IconHashImageSet map[string]string
	TlsDataSet       map[string]struct{}
}

type RequestRole string

type RuntimeLogController struct {
	BaseController
}

type RuntimeLogData struct {
	Id         int    `json:"id"`
	Index      int    `json:"index"`
	Source     string `json:"source"`
	File       string `json:"file"`
	Func       string `json:"func"`
	Level      string `json:"level"`
	Message    string `json:"message"`
	CreateTime string `json:"create_datetime"`
	UpdateTime string `json:"update_datetime"`
}

type RuntimeLogInfo struct {
	Id         int
	Source     string
	File       string
	Func       string
	Level      string
	Message    string
	CreateTime string
	UpdateTime string
}

type ScreenshotFileInfo struct {
	ScreenShotFile          string
	ScreenShotThumbnailFile string
	Tooltip                 string
}

type StartedTaskInfo struct {
	TaskName     string `json:"task_name"`
	TaskArgs     string `json:"task_args"`
	TaskStarting string `json:"task_starting"`
}

type StatusResponseData struct {
	Status string `json:"status"`
	Msg    string `json:"msg"`
}

type TaskController struct {
	BaseController
}

type TaskCronInfo struct {
	Id          int
	TaskId      string
	TaskName    string
	Status      string
	KwArgs      string
	CronRule    string
	LastRunTime string
	CreateTime  string
	UpdateTime  string
	RunCount    int
	Comment     string
	Workspace   string
}

type TaskCronListData struct {
	Id          int    `json:"id"`
	Index       int    `json:"index"`
	TaskId      string `json:"task_id""`
	TaskName    string `json:"task_name"`
	Status      string `json:"status"`
	KwArgs      string `json:"kwargs"`
	CronRule    string `json:"cron_rule"`
	CreateTime  string `json:"create_time"`
	LastRunTime string `json:"lastrun_time"`
	NextRunTime string `json:"nextrun_time"`
	RunCount    int    `json:"run_count"`
	Comment     string `json:"comment"`
}

type TaskInfo struct {
	Id            int
	TaskId        string
	Worker        string
	TaskName      string
	State         string
	Result        string
	KwArgs        string
	ReceivedTime  string
	StartedTime   string
	SucceededTime string
	FailedTime    string
	RetriedTime   string
	RevokedTime   string
	Runtime       string
	CreateTime    string
	UpdateTime    string
	ResultFile    string
	RunTaskInfo   []TaskListData
	Workspace     string
}

type TaskInfoData struct {
	TaskInfo string `json:"task_info"`
}

type TaskListData struct {
	Id           int    `json:"id"`
	Index        string `json:"index"`
	TaskId       string `json:"task_id"`
	Worker       string `json:"worker"`
	TaskName     string `json:"task_name"`
	State        string `json:"state"`
	Result       string `json:"result"`
	KwArgs       string `json:"kwargs"`
	ReceivedTime string `json:"received"`
	StartedTime  string `json:"started"`
	CreateTime   string `json:"created"`
	UpdateTime   string `json:"updated"`
	Runtime      string `json:"runtime"`
	ResultFile   string `json:"resultfile"`
	TaskType     string `json:"tasktype"`
}

type TokenData struct {
	User      string `json:"user"`
	UserRole  string `json:"userRole"`
	Workspace int    `json:"workspace"`
}

type UserController struct {
	BaseController
}

type UserData struct {
	Id              int    `json:"id" form:"id"`
	Index           int    `json:"index" form:"-"`
	UserName        string `json:"user_name" form:"user_name"`
	UserPassword    string `json:"user_password" form:"user_password"`
	UserDescription string `json:"user_description" form:"user_description"`
	UserRole        string `json:"user_role" form:"user_role"`
	State           string `json:"state" form:"state"`
	SortOrder       int    `json:"sort_order" form:"sort_order"`
	CreateDatetime  string `json:"create_time" form:"-"`
	UpdateDatetime  string `json:"update_time" form:"-"`
}

type VulController struct {
	BaseController
}

type VulnerabilityData struct {
	Id          int    `json:"id"`
	Index       int    `json:"index"`
	Target      string `json:"target"`
	Url         string `json:"url"`
	PocFile     string `json:"poc_file"`
	Source      string `json:"source"`
	CreateTime  string `json:"create_datetime"`
	UpdateTime  string `json:"update_datetime"`
	WorkspaceId int    `json:"workspace"`
}

type VulnerabilityInfo struct {
	Id         int
	Target     string
	Url        string
	PocFile    string
	Source     string
	Extra      string
	CreateTime string
	UpdateTime string
	Workspace  string
}

type WikiController struct {
	BaseController
}

type WorkerController struct {
	BaseController
}

type WorkerStatusData struct {
	Index                    int    `json:"index"`
	WorkName                 string `json:"worker_name"`
	WorkerTopic              string `json:"worker_topic"`
	CreateTime               string `json:"create_time"`
	UpdateTime               string `json:"update_time"`
	TaskExecutedNumber       int    `json:"task_number"`
	TaskStartedNumber        int    `json:"started_number"`
	CPULoad                  string `json:"cpu_load"`
	MemUsage                 string `json:"mem_used"`
	EnableManualReloadFlag   bool   `json:"enable_manual_reload_flag"`
	EnableManualFileSyncFlag bool   `json:"enable_manual_file_sync_flag"`
	HeartColor               string `json:"heart_color"`
	IsDaemonProcess          bool   `json:"daemon_process"`
}

type WorkspaceController struct {
	BaseController
}

type WorkspaceData struct {
	Id                   int    `json:"id" form:"id"`
	Index                int    `json:"index" form:"-"`
	WorkspaceName        string `json:"workspace_name" form:"workspace_name"`
	WorkspaceGUID        string `json:"workspace_guid" form:"workspace_guid"`
	WorkspaceDescription string `json:"workspace_description" form:"workspace_description"`
	State                string `json:"state" form:"state"`
	SortOrder            int    `json:"sort_order" form:"sort_order"`
	WikiSpaceId          string `json:"wiki_space_id" form:"wiki_space_id"`
	CreateDatetime       string `json:"create_time" form:"-"`
	UpdateDatetime       string `json:"update_time" form:"-"`
}

type WorkspaceInfo struct {
	WorkspaceInfoList []WorkspaceInfoData
	CurrentWorkspace  string
}

type WorkspaceInfoData struct {
	WorkspaceId   string `json:"workspaceId"`
	WorkspaceName string `json:"workspaceName"`
	WikiSpaceId   string `json:"wikiSpaceId"`
	Enable        bool   `json:"enable"`
}