gethue/topNavViewModel.js

Hue is an Open source SQL Query Editor for Databases/Warehouses

// Licensed to Cloudera, Inc. under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  Cloudera, Inc. licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License.  You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

import * as ko from 'knockout';

import huePubSub from './utils/huePubSub';
import {
  CONFIG_REFRESHED_TOPIC,
  GET_KNOWN_CONFIG_TOPIC,
  REFRESH_CONFIG_TOPIC
} from './config/events';
import { findEditorConnector } from './config/hueConfig';
import { withLocalStorage } from './utils/storageUtils';

class TopNavViewModel {
  constructor(onePageViewModel) {
    const self = this;
    self.onePageViewModel = onePageViewModel;

    // TODO: Drop. Just for PoC
    self.pocClusterMode = ko.observable();
    withLocalStorage('topNav.multiCluster', self.pocClusterMode, 'dw');
    huePubSub.subscribe('set.multi.cluster.mode', self.pocClusterMode);

    self.hasJobBrowser = ko.observable(window.HAS_JOB_BROWSER);
    self.clusters = ko.observableArray();

    const configUpdated = config => {
      if (config && config.clusters) {
        self.clusters(config.clusters);
      }

      self.hasJobBrowser(
        window.HAS_JOB_BROWSER &&
          findEditorConnector(
            connector =>
              connector.dialect === 'yarn' ||
              connector.dialect === 'impala' ||
              connector.dialect === 'dataeng'
          )
      );
    };

    huePubSub.publish(GET_KNOWN_CONFIG_TOPIC, configUpdated);
    huePubSub.subscribe(CONFIG_REFRESHED_TOPIC, configUpdated);

    huePubSub.subscribe('hue.new.default.app', () => {
      huePubSub.publish(REFRESH_CONFIG_TOPIC);
    });
  }
}

export default TopNavViewModel;