import { Body, Controller, Post, Route, Security, SuccessResponse, Tags } from "tsoa";
import HttpStatusCode from "../interfaces/http-status";
import esClient from "../elasticsearch";
import { Search } from "../interfaces/search";
import { EhrFile } from "../interfaces/ehr-fs";

const DEFAULT_INDEX = process.env.ELASTICSEARCH_INDEX;

if (!DEFAULT_INDEX) throw Error("Default ElasticSearch index must be specified.");

@Route("/search")
export class SearchController extends Controller {
  @Post("/")
  @Tags("ค้นหา")
  @Security("bearerAuth")
  @SuccessResponse(HttpStatusCode.OK, "สำเร็จ")
  public async searchFile(@Body() search: Search): Promise<EhrFile[]> {
    const result = await esClient.search<EhrFile & { attachment: Record<string, string> }>({
      index: DEFAULT_INDEX,
      query: {
        bool: {
          must: search.AND?.map((v) => ({ match: { [v.field]: v.value } })),
          should: search.OR?.map((v) => ({ match: { [v.field]: v.value } })),
        },
      },
      size: 10000,
    });

    return result.hits.hits.length > 0
      ? result.hits.hits
          .map((v) => {
            if (!v._source) return;

            const { attachment, ...rest } = v._source;

            return rest;
          })
          .flatMap((v) => (!!v ? [v] : []))
      : [];
  }
}