package main

import (
	"fmt"
	"strings"

	onlyoffice "github.com/eslider/go-onlyoffice"
	"github.com/spf13/cobra"
)

func init() {
	rootCmd.AddCommand(searchCmd())
}

// searchCmd queries the OnlyOffice Elasticsearch index directly. The REST
// /api/2.0/files/@search endpoint only searches file names in the database;
// content search needs ES (see docs/elasticsearch.md).
func searchCmd() *cobra.Command {
	var (
		content bool
		folder  string
		limit   int
		backend string
		asJSON  bool
	)
	cmd := &cobra.Command{
		Use:   "search QUERY",
		Short: "Full-text search over documents by name, optionally by content (Elasticsearch)",
		Long: "Search the OnlyOffice Documents index.\n\n" +
			"By default only file names are matched. With --content the query also\n" +
			"matches extracted document text (document.attachment.content); this covers\n" +
			"Office formats (docx/xlsx/pptx) and is slower.\n\n" +
			"--backend own queries the separate index populated by `oo index`\n" +
			"(ONLYOFFICE_ES_TEXT_INDEX, default oo_docs_text) instead, which also holds\n" +
			"PDFs and scans (see docs/elasticsearch.md).\n\n" +
			"Requires ONLYOFFICE_ES_URL (and optionally ONLYOFFICE_ES_INDEX,\n" +
			"ONLYOFFICE_TENANT). See docs/elasticsearch.md for the tunnel setup.",
		Args: cobra.ExactArgs(1),
		RunE: func(cmd *cobra.Command, args []string) error {
			if asJSON {
				outputFormat = "json"
			}
			var (
				searcher onlyoffice.Searcher
				err      error
			)
			switch strings.ToLower(strings.TrimSpace(backend)) {
			case "", "oo", "elasticsearch":
				searcher, err = onlyoffice.NewESSearcher(onlyoffice.ESConfigFromEnv())
			case "own", "es-text":
				searcher, err = onlyoffice.NewESTextIndex(onlyoffice.ESTextConfigFromEnv())
			default:
				return fmt.Errorf("unknown search backend %q (want oo|own)", backend)
			}
			if err != nil {
				return err
			}
			hits, err := searcher.Search(cmd.Context(), onlyoffice.SearchQuery{
				Text:      args[0],
				InContent: content,
				FolderID:  folder,
				Limit:     limit,
			})
			if err != nil {
				return err
			}
			rows := make([]map[string]any, 0, len(hits))
			for _, h := range hits {
				rows = append(rows, map[string]any{
					"id":        h.ID,
					"title":     h.Title,
					"folder":    h.ParentID,
					"score":     h.Score,
					"highlight": h.Highlight,
				})
			}
			if outputFormat == "json" {
				printJSON(rows)
				return nil
			}
			printTable([]string{"id", "title", "folder", "score", "highlight"}, rows)
			return nil
		},
	}
	cmd.Flags().BoolVar(&content, "content", false, "also match extracted document content")
	cmd.Flags().StringVar(&folder, "folder", "", "limit to a Documents folder id")
	cmd.Flags().IntVar(&limit, "limit", 20, "maximum number of results")
	cmd.Flags().StringVar(&backend, "backend", "oo", "index to query: oo (OnlyOffice) | own (oo index)")
	cmd.Flags().BoolVar(&asJSON, "json", false, "shorthand for --output json")
	return cmd
}