migrated file to ts (#10328)

public/app/features/plugins/datasource_srv.ts

@@ -0,0 +1,152 @@
+import _ from 'lodash';
+import coreModule from 'app/core/core_module';
+import config from 'app/core/config';
+// import pluginLoader from './plugin_loader';
+import { importPluginModule } from './plugin_loader';
+
+export class DatasourceSrv {
+  datasources: any;
+
+  constructor(private $q, private $injector, $rootScope, private templateSrv) {
+    this.init();
+  }
+
+  init() {
+    this.datasources = {};
+  }
+
+  get(name) {
+    if (!name) {
+      return this.get(config.defaultDatasource);
+    }
+
+    name = this.templateSrv.replace(name);
+
+    if (name === 'default') {
+      return this.get(config.defaultDatasource);
+    }
+
+    if (this.datasources[name]) {
+      return this.$q.when(this.datasources[name]);
+    }
+
+    return this.loadDatasource(name);
+  }
+
+  loadDatasource(name) {
+    var dsConfig = config.datasources[name];
+    if (!dsConfig) {
+      return this.$q.reject({ message: 'Datasource named ' + name + ' was not found' });
+    }
+
+    var deferred = this.$q.defer();
+    var pluginDef = dsConfig.meta;
+
+    importPluginModule(pluginDef.module)
+      .then(plugin => {
+        // check if its in cache now
+        if (this.datasources[name]) {
+          deferred.resolve(this.datasources[name]);
+          return;
+        }
+
+        // plugin module needs to export a constructor function named Datasource
+        if (!plugin.Datasource) {
+          throw new Error('Plugin module is missing Datasource constructor');
+        }
+
+        var instance = this.$injector.instantiate(plugin.Datasource, { instanceSettings: dsConfig });
+        instance.meta = pluginDef;
+        instance.name = name;
+        this.datasources[name] = instance;
+        deferred.resolve(instance);
+      })
+      .catch(function(err) {
+        this.$rootScope.appEvent('alert-error', [dsConfig.name + ' plugin failed', err.toString()]);
+      });
+
+    return deferred.promise;
+  }
+
+  getAll() {
+    return config.datasources;
+  }
+
+  getAnnotationSources() {
+    var sources = [];
+
+    this.addDataSourceVariables(sources);
+
+    _.each(config.datasources, function(value) {
+      if (value.meta && value.meta.annotations) {
+        sources.push(value);
+      }
+    });
+
+    return sources;
+  }
+
+  getMetricSources(options) {
+    var metricSources = [];
+
+    _.each(config.datasources, function(value, key) {
+      if (value.meta && value.meta.metrics) {
+        metricSources.push({ value: key, name: key, meta: value.meta });
+
+        if (key === config.defaultDatasource) {
+          metricSources.push({ value: null, name: 'default', meta: value.meta });
+        }
+      }
+    });
+
+    if (!options || !options.skipVariables) {
+      this.addDataSourceVariables(metricSources);
+    }
+
+    metricSources.sort(function(a, b) {
+      // these two should always be at the bottom
+      if (a.meta.id === 'mixed' || a.meta.id === 'grafana') {
+        return 1;
+      }
+      if (b.meta.id === 'mixed' || b.meta.id === 'grafana') {
+        return -1;
+      }
+      if (a.name.toLowerCase() > b.name.toLowerCase()) {
+        return 1;
+      }
+      if (a.name.toLowerCase() < b.name.toLowerCase()) {
+        return -1;
+      }
+      return 0;
+    });
+
+    return metricSources;
+  }
+
+  addDataSourceVariables(list) {
+    // look for data source variables
+    for (var i = 0; i < this.templateSrv.variables.length; i++) {
+      var variable = this.templateSrv.variables[i];
+      if (variable.type !== 'datasource') {
+        continue;
+      }
+
+      var first = variable.current.value;
+      if (first === 'default') {
+        first = config.defaultDatasource;
+      }
+
+      var ds = config.datasources[first];
+
+      if (ds) {
+        list.push({
+          name: '$' + variable.name,
+          value: '$' + variable.name,
+          meta: ds.meta,
+        });
+      }
+    }
+  }
+}
+
+coreModule.service('datasourceSrv', DatasourceSrv);