types

type AmassRangerEntry interface {
	Network() net.IPNet
	AutonomousSystem() int
	Source() *Source
}

type Asset struct {
	Session uuid.UUID `json:"sessionToken,omitempty"`
	Name    string    `json:"assetName,omitempty"`
	Data    AssetData `json:"data,omitempty"`
}

type AssetData struct {
	OAMAsset oam.Asset     `json:"asset"`
	OAMType  oam.AssetType `json:"type"`
}

type AssetPipeline struct {
	Pipeline *pipeline.Pipeline
	Queue    *PipelineQueue
}

type Dispatcher interface {
	DispatchEvent(e *Event) error
	Shutdown()
}

type EmailMeta struct {
	VerifyAttempted bool
	Verified        bool
}

type Event struct {
	Name       string
	Entity     *dbt.Entity
	Meta       interface{}
	Dispatcher Dispatcher
	Session    Session
}

type EventDataElement struct {
	Event *Event
	Error error
	Queue chan *EventDataElement
}

type Handler struct {
	Plugin       Plugin
	Name         string
	Priority     int
	MaxInstances int
	EventType    oam.AssetType
	Transforms   []string
	Callback     func(*Event) error
}

type PipelineQueue struct {
	queue.Queue
}

type Plugin interface {
	Name() string
	Start(r Registry) error
	Stop()
}

type Registry interface {
	Log() *slog.Logger
	RegisterHandler(h *Handler) error
	BuildPipelines() error
	GetPipeline(eventType oam.AssetType) (*AssetPipeline, error)
}

type Session interface {
	ID() uuid.UUID
	Log() *slog.Logger
	PubSub() *pubsub.Logger
	Config() *config.Config
	Scope() *scope.Scope
	DB() repository.Repository
	Cache() *cache.Cache
	Queue() SessionQueue
	CIDRanger() cidranger.Ranger
	TmpDir() string
	Stats() *SessionStats
	Done() bool
	Kill()
}

type SessionManager interface {
	NewSession(cfg *config.Config) (Session, error)
	AddSession(s Session) error
	CancelSession(id uuid.UUID)
	GetSession(id uuid.UUID) Session
	GetSessions() []Session
	Shutdown()
}

type SessionQueue interface {
	Has(e *dbt.Entity) bool
	Append(e *dbt.Entity) error
	Next(atype oam.AssetType, num int) ([]*dbt.Entity, error)
	Processed(e *dbt.Entity) error
	Delete(e *dbt.Entity) error
	Close() error
}

type SessionStats struct {
	sync.Mutex
	WorkItemsCompleted int `json:"workItemsCompleted"`
	WorkItemsTotal     int `json:"workItemsTotal"`
}

type Source struct {
	Name       string
	Confidence int
}