import.go

package stream_chat

import (
	"context"
	"net/http"
	"net/url"
	"strconv"
	"time"
)

type ImportTaskHistory struct {
	CreatedAt time.Time `json:"created_at"`
	NextState string    `json:"next_state"`
	PrevState string    `json:"prev_state"`
}

type ImportMode string

const (
	InsertMode ImportMode = "insert"
	UpsertMode ImportMode = "upsert"
)

type ImportTask struct {
	CreatedAt time.Time            `json:"created_at"`
	Path      string               `json:"path"`
	Mode      ImportMode           `json:"mode"`
	History   []*ImportTaskHistory `json:"history"`
	ID        string               `json:"id"`
	State     string               `json:"state"`
	UpdatedAt time.Time            `json:"updated_at"`
	Result    interface{}          `json:"result"`
	Size      *int                 `json:"size"`
}

type ListImportsOptions struct {
	Limit  int
	Offset int
}

type CreateImportResponse struct {
	ImportTask *ImportTask `json:"import_task"`
	Response
}

type CreateImportURLResponse struct {
	Path      string `json:"path"`
	UploadURL string `json:"upload_url"`
	Response
}

type GetImportResponse struct {
	ImportTask *ImportTask `json:"import_task"`
	Response
}

type ListImportsResponse struct {
	ImportTasks []*ImportTask `json:"import_tasks"`
	Response
}

// CreateImportURL creates a new import URL.
// Note: Do not use this.
// It is present for internal usage only.
// This function can, and will, break and/or be removed at any point in time.
func (c *Client) CreateImportURL(ctx context.Context, filename string) (*CreateImportURLResponse, error) {
	var resp CreateImportURLResponse
	err := c.makeRequest(ctx, http.MethodPost, "import_urls", nil, map[string]string{"filename": filename}, &resp)

	return &resp, err
}

type createImportRequest struct {
	Path        string `json:"path"`
	Mode        string `json:"mode"`
	MergeCustom *bool  `json:"merge_custom,omitempty"`
}

type CreateImportOption func(*createImportRequest)

// WithMergeCustom sets whether custom data should be merged during import.
func WithMergeCustom(mergeCustom bool) CreateImportOption {
	return func(r *createImportRequest) {
		r.MergeCustom = &mergeCustom
	}
}

// CreateImport creates a new import task.
// Note: Do not use this.
// It is present for internal usage only.
// This function can, and will, break and/or be removed at any point in time.
func (c *Client) CreateImport(ctx context.Context, filePath string, mode ImportMode, opts ...CreateImportOption) (*CreateImportResponse, error) {
	req := createImportRequest{
		Path: filePath,
		Mode: string(mode),
	}
	for _, opt := range opts {
		opt(&req)
	}

	var resp CreateImportResponse
	err := c.makeRequest(ctx, http.MethodPost, "imports", nil, req, &resp)

	return &resp, err
}

// GetImport returns an import task.
// Note: Do not use this.
// It is present for internal usage only.
// This function can, and will, break and/or be removed at any point in time.
func (c *Client) GetImport(ctx context.Context, id string) (*GetImportResponse, error) {
	var resp GetImportResponse
	err := c.makeRequest(ctx, http.MethodGet, "imports/"+id, nil, nil, &resp)

	return &resp, err
}

// ListImports returns all import tasks.
// Note: Do not use this.
// It is present for internal usage only.
// This function can, and will, break and/or be removed at any point in time.
func (c *Client) ListImports(ctx context.Context, opts *ListImportsOptions) (*ListImportsResponse, error) {
	params := url.Values{}
	if opts != nil {
		params.Set("limit", strconv.Itoa(opts.Limit))
		params.Set("offset", strconv.Itoa(opts.Offset))
	}

	var resp ListImportsResponse
	err := c.makeRequest(ctx, http.MethodGet, "imports", params, nil, &resp)

	return &resp, err
}