db

type AssignUnassignedWorkParams struct {
	Worker  uuid.NullUUID `json:"worker"`
	Column2 interface{}   `json:"column_2"`
}

type ChangeOccurenceStatusParams struct {
	Status Status `json:"status"`
	ID     int32  `json:"id"`
}

type CreateHistoryParams struct {
	OccurenceID int32     `json:"occurence_id"`
	Schedule    uuid.UUID `json:"schedule"`
	Status      Status    `json:"status"`
	Details     string    `json:"details"`
	Manual      bool      `json:"manual"`
	ScheduledAt time.Time `json:"scheduled_at"`
	StartedAt   time.Time `json:"started_at"`
	CompletedAt time.Time `json:"completed_at"`
}

type CreateOccurenceParams struct {
	Schedule  uuid.UUID     `json:"schedule"`
	Worker    uuid.NullUUID `json:"worker"`
	Manual    bool          `json:"manual"`
	Status    Status        `json:"status"`
	Occurence sql.NullTime  `json:"occurence"`
}

type CreateScheduleParams struct {
	ID     uuid.UUID `json:"id"`
	Cron   string    `json:"cron"`
	Hook   string    `json:"hook"`
	Owner  string    `json:"owner"`
	Active bool      `json:"active"`
	Till   time.Time `json:"till"`
	Data   string    `json:"data"`
}

type DBTX interface {
	ExecContext(context.Context, string, ...interface{}) (sql.Result, error)
	PrepareContext(context.Context, string) (*sql.Stmt, error)
	QueryContext(context.Context, string, ...interface{}) (*sql.Rows, error)
	QueryRowContext(context.Context, string, ...interface{}) *sql.Row
}

type History struct {
	OccurenceID int32     `json:"occurence_id"`
	Schedule    uuid.UUID `json:"schedule"`
	Status      Status    `json:"status"`
	Details     string    `json:"details"`
	Manual      bool      `json:"manual"`
	ScheduledAt time.Time `json:"scheduled_at"`
	StartedAt   time.Time `json:"started_at"`
	CompletedAt time.Time `json:"completed_at"`
}

type ListHistoryParams struct {
	Schedule uuid.UUID `json:"schedule"`
	Limit    int32     `json:"limit"`
	Offset   int32     `json:"offset"`
}

type ListHistoryRow struct {
	OccurenceID  int32     `json:"occurence_id"`
	Schedule     uuid.UUID `json:"schedule"`
	Status       Status    `json:"status"`
	Details      string    `json:"details"`
	Manual       bool      `json:"manual"`
	ScheduledAt  time.Time `json:"scheduled_at"`
	StartedAt    time.Time `json:"started_at"`
	CompletedAt  time.Time `json:"completed_at"`
	TotalRecords int64     `json:"-"`
}

type ListSchedulesParams struct {
	Owner  string `json:"owner"`
	Limit  int32  `json:"limit"`
	Offset int32  `json:"offset"`
}

type ListSchedulesRow struct {
	ID           uuid.UUID `json:"id"`
	Cron         string    `json:"cron"`
	Hook         string    `json:"hook"`
	Owner        string    `json:"owner"`
	Data         string    `json:"data"`
	Active       bool      `json:"active"`
	Till         time.Time `json:"till"`
	CreatedAt    time.Time `json:"created_at"`
	LastModified time.Time `json:"last_modified"`
	TotalRecords int64     `json:"total_records"`
}

type ListWorkersParams struct {
	Limit  int32 `json:"limit"`
	Offset int32 `json:"offset"`
}

type NextOccurence struct {
	ID          int32         `json:"id"`
	Schedule    uuid.UUID     `json:"schedule"`
	Worker      uuid.NullUUID `json:"worker"`
	Manual      bool          `json:"manual"`
	Status      Status        `json:"status"`
	Occurence   sql.NullTime  `json:"occurence"`
	LastUpdated time.Time     `json:"last_updated"`
}

type NullStatus struct {
	Status Status
	Valid  bool // Valid is true if Status is not NULL
}

type PunchCard struct {
	ID        uuid.UUID    `json:"id"`
	LastPunch time.Time    `json:"last_punch"`
	CreatedAt sql.NullTime `json:"created_at"`
}

type Querier interface {
	AssignUnassignedWork(ctx context.Context, arg AssignUnassignedWorkParams) error
	ChangeOccurenceStatus(ctx context.Context, arg ChangeOccurenceStatusParams) error
	CreateHistory(ctx context.Context, arg CreateHistoryParams) (History, error)
	CreateOccurence(ctx context.Context, arg CreateOccurenceParams) (NextOccurence, error)
	CreateSchedule(ctx context.Context, arg CreateScheduleParams) (Schedule, error)
	CreateWorker(ctx context.Context, id uuid.UUID) (PunchCard, error)
	DeleteOccurence(ctx context.Context, id int32) error
	DeleteSchedule(ctx context.Context, id uuid.UUID) error
	DeleteWorker(ctx context.Context, id uuid.UUID) error
	GetNextImmediateWork(ctx context.Context, worker uuid.NullUUID) (time.Time, error)
	GetOccurence(ctx context.Context, id int32) (NextOccurence, error)
	GetSchedule(ctx context.Context, id uuid.UUID) (Schedule, error)
	GetWorker(ctx context.Context, id uuid.UUID) (PunchCard, error)
	ListHistory(ctx context.Context, arg ListHistoryParams) ([]ListHistoryRow, error)
	ListSchedules(ctx context.Context, arg ListSchedulesParams) ([]ListSchedulesRow, error)
	ListWorkers(ctx context.Context, arg ListWorkersParams) ([]PunchCard, error)
	MyExpiredWork(ctx context.Context, worker uuid.NullUUID) ([]NextOccurence, error)
	ProveLiveliness(ctx context.Context, id uuid.UUID) error
	RemoveDeadWorkers(ctx context.Context) error
	UpdateSchedule(ctx context.Context, arg UpdateScheduleParams) (Schedule, error)
	UpdateStatusAndDetails(ctx context.Context, arg UpdateStatusAndDetailsParams) (History, error)
	ValidSchedulesWithoutOccurence(ctx context.Context) ([]Schedule, error)
}

type Queries struct {
	// contains filtered or unexported fields
}

type SQLStore struct {
	*Queries
	// contains filtered or unexported fields
}

type Schedule struct {
	ID     uuid.UUID `json:"id"`
	Cron   string    `json:"cron"`
	Hook   string    `json:"hook"`
	Owner  string    `json:"owner"`
	Data   string    `json:"data"`
	Active bool      `json:"active"`
	// till what timestamp this schedule will run
	Till         time.Time `json:"till"`
	CreatedAt    time.Time `json:"created_at"`
	LastModified time.Time `json:"last_modified"`
}

type Status string

type Store interface {
	Querier
	UpdateHistoryAndOccurence(ctx context.Context, schedule Schedule, occurence NextOccurence) error
	UpdateHistoryAndDeleteOccurence(ctx context.Context, params UpdateHistoryAndDeleteOccurenceParams) error
	CreateScheduleAddNextOccurence(ctx context.Context, schedule CreateScheduleParams, occurence CreateOccurenceParams) (Schedule, error)
	Close() error
}

type UpdateHistoryAndDeleteOccurenceParams struct {
	Schedule  Schedule
	Occurence NextOccurence
	Details   string
	Status    Status
}

type UpdateScheduleParams struct {
	ID     uuid.UUID `json:"id"`
	Cron   string    `json:"cron"`
	Hook   string    `json:"hook"`
	Active bool      `json:"active"`
	Till   time.Time `json:"till"`
	Data   string    `json:"data"`
}

type UpdateStatusAndDetailsParams struct {
	OccurenceID int32  `json:"occurence_id"`
	Details     string `json:"details"`
	Status      Status `json:"status"`
}