utils.ts

import { BaseQuickwitDataSource } from "./base";
import { useState, useEffect, useCallback } from "react";
import{ MetricFindValue } from '@grafana/data';

/**
 * Provide suggestions based on datasource fields
 */

export type Suggestion = {
  from: number,
  options: Array<{
    label: string,
    detail?: string,
    type?: string,
  }>
}

export function useDatasourceFields(datasource: BaseQuickwitDataSource) {
  const [fields, setFields] = useState<MetricFindValue[]>([]);

  useEffect(() => {
    if (datasource.getTagKeys) {
      datasource.getTagKeys({ searchable: true }).then(setFields);
    }
  }, [datasource, setFields]);

  const getSuggestions = useCallback(async (word: string): Promise<Suggestion> => {
    let suggestions: Suggestion = { from: 0, options: [] };

    const wordIsField = word.match(/([^:\s]+):'?([^'\s]*)'?/);
    if (wordIsField?.length) {
      const [_match, fieldName, _fieldValue] = wordIsField;
      const candidateValues = await datasource.getTagValues({ key: fieldName });
      suggestions.from = fieldName.length + 1; // Replace only the value part
      suggestions.options = candidateValues.map(v => ({
        type: 'text',
        label: typeof v.text === 'number' ? `${v.text}` : `'${v.text}'`
      }));
    } else {
      const candidateFields = fields;
      suggestions.from = 0;
      suggestions.options = candidateFields.map(f => ({
        type: 'variable',
        label: f.text,
        detail: `${f.value}`
      }));
    }
    return suggestions;

  }, [datasource, fields]);

  return {fields, getSuggestions}
}