typesense

type APIClientInterface interface {
	api.ClientWithResponsesInterface
	api.ClientInterface
}

type AliasInterface interface {
	Retrieve(ctx context.Context) (*api.CollectionAlias, error)
	Delete(ctx context.Context) (*api.CollectionAlias, error)
}

type AliasesInterface interface {
	Upsert(ctx context.Context, aliasName string, aliasSchema *api.CollectionAliasSchema) (*api.CollectionAlias, error)
	Retrieve(ctx context.Context) ([]*api.CollectionAlias, error)
}

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

type ApiCallOption func(*ApiCall)

type Client struct {
	MultiSearch MultiSearchInterface
	// contains filtered or unexported fields
}

type ClientConfig struct {
	ServerURL                   string
	NearestNode                 string // optional
	Nodes                       []string
	NumRetries                  int
	RetryInterval               time.Duration
	HealthcheckInterval         time.Duration
	APIKey                      string
	ConnectionTimeout           time.Duration
	CircuitBreakerName          string
	CircuitBreakerMaxRequests   uint32
	CircuitBreakerInterval      time.Duration
	CircuitBreakerTimeout       time.Duration
	CircuitBreakerReadyToTrip   circuit.GoBreakerReadyToTripFunc
	CircuitBreakerOnStateChange circuit.GoBreakerOnStateChangeFunc
}

type ClientOption func(*Client)

type CollectionInterface[T any] interface {
	Retrieve(ctx context.Context) (*api.CollectionResponse, error)
	Delete(ctx context.Context) (*api.CollectionResponse, error)
	Documents() DocumentsInterface
	Document(documentID string) DocumentInterface[T]
	Overrides() OverridesInterface
	Override(overrideID string) OverrideInterface
	Synonyms() SynonymsInterface
	Synonym(synonymID string) SynonymInterface
	Update(context.Context, *api.CollectionUpdateSchema) (*api.CollectionUpdateSchema, error)
}

type CollectionsInterface interface {
	Create(ctx context.Context, schema *api.CollectionSchema) (*api.CollectionResponse, error)
	Retrieve(ctx context.Context) ([]*api.CollectionResponse, error)
}

type DocumentInterface[T any] interface {
	Retrieve(ctx context.Context) (T, error)
	Update(ctx context.Context, document any) (T, error)
	Delete(ctx context.Context) (T, error)
}

type DocumentsInterface interface {
	// Create returns indexed document
	Create(ctx context.Context, document interface{}) (map[string]interface{}, error)
	// Update updates documents matching the filter_by condition
	Update(ctx context.Context, updateFields interface{}, params *api.UpdateDocumentsParams) (int, error)
	// Upsert returns indexed/updated document
	Upsert(context.Context, interface{}) (map[string]interface{}, error)
	// Delete returns number of deleted documents
	Delete(ctx context.Context, filter *api.DeleteDocumentsParams) (int, error)
	// Search performs document search in collection
	Search(ctx context.Context, params *api.SearchCollectionParams) (*api.SearchResult, error)
	// Export returns all documents from index in jsonl format
	Export(ctx context.Context) (io.ReadCloser, error)
	// Import returns json array. Each item of the response indicates
	// the result of each document present in the request body (in the same order).
	Import(ctx context.Context, documents []interface{}, params *api.ImportDocumentsParams) ([]*api.ImportDocumentResponse, error)
	// ImportJsonl accepts documents and returns result in jsonl format. Each line of the
	// response indicates the result of each document present in the
	// request body (in the same order).
	ImportJsonl(ctx context.Context, body io.Reader, params *api.ImportDocumentsParams) (io.ReadCloser, error)
}

type HTTPError struct {
	Status int
	Body   []byte
}

type KeyInterface interface {
	Retrieve(ctx context.Context) (*api.ApiKey, error)
	Delete(ctx context.Context) (*api.ApiKey, error)
}

type KeysInterface interface {
	Create(context.Context, *api.ApiKeySchema) (*api.ApiKey, error)
	Retrieve(context.Context) ([]*api.ApiKey, error)
	GenerateScopedSearchKey(searchKey string, params map[string]interface{}) (string, error)
}

type MultiSearchInterface interface {
	Perform(ctx context.Context, commonSearchParams *api.MultiSearchParams, searchParams api.MultiSearchSearchesParameter) (*api.MultiSearchResult, error)
	PerformWithContentType(ctx context.Context, commonSearchParams *api.MultiSearchParams, searchParams api.MultiSearchSearchesParameter, contentType string) (*api.MultiSearchResponse, error)
}

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

type OperationsInterface interface {
	Snapshot(ctx context.Context, snapshotPath string) (bool, error)
	Vote(ctx context.Context) (bool, error)
}

type OverrideInterface interface {
	Retrieve(ctx context.Context) (*api.SearchOverride, error)
	Delete(ctx context.Context) (*api.SearchOverride, error)
}

type OverridesInterface interface {
	Upsert(ctx context.Context, overrideID string, overrideSchema *api.SearchOverrideSchema) (*api.SearchOverride, error)
	Retrieve(ctx context.Context) ([]*api.SearchOverride, error)
}

type PresetInterface interface {
	Retrieve(ctx context.Context) (*api.PresetSchema, error)
	Delete(ctx context.Context) (*api.PresetDeleteSchema, error)
}

type PresetsInterface interface {
	Retrieve(ctx context.Context) ([]*api.PresetSchema, error)
	Upsert(ctx context.Context, presetName string, presetValue *api.PresetUpsertSchema) (*api.PresetSchema, error)
}

type SynonymInterface interface {
	// Retrieve a single search synonym
	Retrieve(ctx context.Context) (*api.SearchSynonym, error)
	// Delete a synonym associated with a collection
	Delete(ctx context.Context) (*api.SearchSynonym, error)
}

type SynonymsInterface interface {
	// Create or update a synonym
	Upsert(ctx context.Context, synonymID string, synonymSchema *api.SearchSynonymSchema) (*api.SearchSynonym, error)
	// List all collection synonyms
	Retrieve(ctx context.Context) ([]*api.SearchSynonym, error)
}