energylink/public/app/plugins/datasource/input/InputDatasource.ts

// Types
import {
  DataQueryRequest,
  SeriesData,
  DataQueryResponse,
  DataSourceApi,
  DataSourceInstanceSettings,
} from '@grafana/ui/src/types';
import { InputQuery, InputOptions } from './types';

export class InputDatasource implements DataSourceApi<InputQuery, InputOptions> {
  data: SeriesData[];

  // Filled in by grafana plugin system
  name?: string;

  // Filled in by grafana plugin system
  id?: number;

  constructor(instanceSettings: DataSourceInstanceSettings<InputOptions>) {
    if (instanceSettings.jsonData) {
      this.data = instanceSettings.jsonData.data;
    }

    if (!this.data) {
      this.data = [];
    }
  }

  getDescription(data: SeriesData[]): string {
    if (!data) {
      return '';
    }
    if (data.length > 1) {
      const count = data.reduce((acc, series) => {
        return acc + series.rows.length;
      }, 0);
      return `${data.length} Series, ${count} Rows`;
    }
    const series = data[0];
    return `${series.fields.length} Fields, ${series.rows.length} Rows`;
  }

  /**
   * Convert a query to a simple text string
   */
  getQueryDisplayText(query: InputQuery): string {
    if (query.data) {
      return 'Panel Data: ' + this.getDescription(query.data);
    }
    return `Shared Data From: ${this.name} (${this.getDescription(this.data)})`;
  }

  metricFindQuery(query: string, options?: any) {
    return new Promise((resolve, reject) => {
      const names = [];
      for (const series of this.data) {
        for (const field of series.fields) {
          // TODO, match query/options?
          names.push({
            text: field.name,
          });
        }
      }
      resolve(names);
    });
  }

  query(options: DataQueryRequest<InputQuery>): Promise<DataQueryResponse> {
    const results: SeriesData[] = [];
    for (const query of options.targets) {
      if (query.hide) {
        continue;
      }
      const data = query.data ? query.data : this.data;
      for (const series of data) {
        results.push({
          refId: query.refId,
          ...series,
        });
      }
    }
    return Promise.resolve({ data: results });
  }

  testDatasource() {
    return new Promise((resolve, reject) => {
      let rowCount = 0;
      let info = `${this.data.length} Series:`;
      for (const series of this.data) {
        info += ` [${series.fields.length} Fields, ${series.rows.length} Rows]`;
        rowCount += series.rows.length;
      }

      if (rowCount > 0) {
        resolve({
          status: 'success',
          message: info,
        });
      }
      reject({
        status: 'error',
        message: 'No Data Entered',
      });
    });
  }
}

export default InputDatasource;