package abstr

import (
	"context"
	"reflect"
	"time"

	"go.mongodb.org/mongo-driver/mongo"
)

// 数据库操作默认超时时间(5 秒)
const DefaultTimeout = 5 * time.Second

type DBHelper interface {
	Aggregate(ctx context.Context, collname string, pipeline []map[string]interface{}, value interface{}) error
	CountDocuments(ctx context.Context, collname string, filter map[string]interface{}) (int64, error)
	CreateOne(ctx context.Context, collname string, rec interface{}) error
	CreateMany(ctx context.Context, collname string, recs ...interface{}) error
	FindOne(ctx context.Context, collname string, value interface{}, options ...FindOption) error
	FindOneAndDelete(ctx context.Context, collname string, filter map[string]interface{}, value interface{}, options ...FindOption) error
	List(ctx context.Context, collname string, value interface{}, options ...FindOption) (*Page, error)
	// $pull
	Pull(ctx context.Context, collname string, filter map[string]interface{}, value interface{}) error
	RemoveMany(ctx context.Context, collname string, filter map[string]interface{}) error
	// $unset
	Unmark(ctx context.Context, collname string, filter map[string]interface{}, value interface{}) error
	UpdateMany(ctx context.Context, collname string, filter interface{}, update interface{}) (*mongo.UpdateResult, error)
	UpdateOne(ctx context.Context, collname string, filter map[string]interface{}, value interface{}, options ...FindOption) error
	// 对 UpdateOne 的扩展，允许使用 $set 之外的操作符
	UpdateOneMore(ctx context.Context, collname string, filter map[string]interface{}, value interface{}, options ...FindOption) error
}

//Init init db session
//   t  要注册的类型
type Init func(dbc DBConfig, t ...reflect.Type) error

//Decrypt decrypt db config
type Decrypt func(src string) string

type DBConfig struct {
	URI        string `yaml:"uri"`
	PoolSize   int    `yaml:"poolSize"`
	SSLEnabled bool   `yaml:"sslEnabled"`
	RootCA     string `yaml:"rootCAFile"`
	Timeout    string `yaml:"timeout"`
	VerifyPeer bool   `yaml:"verifyPeer"`
}