grafana/public/app/features/dataframe-import/utils.ts

import { Accept } from 'react-dropzone';
import { Observable } from 'rxjs';

import { toDataFrame } from '@grafana/data';
import { readSpreadsheet } from 'app/core/utils/sheet';

import { FileImportResult } from './types';

function getFileExtensions(acceptedFiles: Accept) {
  const fileExtentions = new Set<string>();
  Object.keys(acceptedFiles).forEach((v) => {
    acceptedFiles[v].forEach((extension) => {
      fileExtentions.add(extension);
    });
  });
  return fileExtentions;
}

export function formatFileTypes(acceptedFiles: Accept) {
  const fileExtentions = Array.from(getFileExtensions(acceptedFiles));
  if (fileExtentions.length === 1) {
    return fileExtentions[0];
  }
  return `${fileExtentions.slice(0, -1).join(', ')} or ${fileExtentions.slice(-1)}`;
}

export function filesToDataframes(files: File[]): Observable<FileImportResult> {
  return new Observable<FileImportResult>((subscriber) => {
    let completedFiles = 0;
    files.forEach((file) => {
      const reader = new FileReader();
      reader.readAsArrayBuffer(file);
      reader.onload = () => {
        const result = reader.result;
        if (result && result instanceof ArrayBuffer) {
          if (file.type === 'application/json') {
            const decoder = new TextDecoder('utf-8');
            const json = JSON.parse(decoder.decode(result));
            subscriber.next({ dataFrames: [toDataFrame(json)], file: file });
          } else {
            subscriber.next({ dataFrames: readSpreadsheet(result), file: file });
          }
          if (++completedFiles >= files.length) {
            subscriber.complete();
          }
        }
      };
    });
  });
}