ambari/ambari-web/app/utils/db.js

/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF 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.
 */
var App = require('app');
App.db = {};

if (typeof Storage !== 'undefined') {
  Storage.prototype.setObject = function (key, value) {
    this.setItem(key, JSON.stringify(value));
  }

  Storage.prototype.getObject = function (key) {
    var value = this.getItem(key);
    return value && JSON.parse(value);
  }
} else {
  // stub for unit testing purposes
  window.localStorage = {};
  localStorage.setItem = function (key, val) {
    this[key] = val;
  }
  localStorage.getItem = function (key) {
    return this[key];
  }
  window.localStorage.setObject = function (key, value) {
    this[key] = value;
  };
  window.localStorage.getObject = function (key, value) {
    return this[key];
  };
}

App.db.cleanUp = function () {
  console.log('TRACE: Entering db:cleanup function');
  App.db.data = {
    'app': {
      'loginName': '',
      'authenticated': false
    }
  };
  console.log("In cleanup./..");
  localStorage.setObject('ambari', App.db.data);
}

// called whenever user logs in
if (localStorage.getObject('ambari') == null) {
  console.log('doing a cleanup');
  App.db.cleanUp();
}

/*
 * setter methods
 */

App.db.setLoginName = function (name) {
  console.log('TRACE: Entering db:setLoginName function');
  App.db.data = localStorage.getObject('ambari');
  App.db.data.app.loginName = name;
  localStorage.setObject('ambari', App.db.data);
}

// that works incorrectly
App.db.setUser = function (user) {
  console.log('TRACE: Entering db:setUser function');
  App.db.data = localStorage.getObject('ambari');
  App.db.data.app.user = user;
  localStorage.setObject('ambari', App.db.data);
}

App.db.setAuthenticated = function (authenticated) {
  console.log('TRACE: Entering db:setAuthenticated function');

  App.db.data = localStorage.getObject('ambari');
  console.log('present value of authentication is: ' + App.db.data.app.authenticated);
  console.log('desired value of authentication is: ' + authenticated);
  App.db.data.app.authenticated = authenticated;
  localStorage.setObject('ambari', App.db.data);
  App.db.data = localStorage.getObject('ambari');
  console.log('Now present value of authentication is: ' + App.db.data.app.authenticated);
}

App.db.setSection = function (section) {
  console.log('TRACE: Entering db:setSection function');
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  if (App.db.data[user] == undefined) {
    App.db.data[user] = {'name': user};
  }
  if (App.db.data[user].ClusterName == undefined) {
    App.db.data[user].ClusterName = {};
  }
  App.db.data[user].section = section;
  localStorage.setObject('ambari', App.db.data);
}

App.db.setInstallerCurrentStep = function (currentStep, completed) {
  console.log('TRACE: Entering db:setInstallerCurrentStep function');
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  if (App.db.data[user] == undefined) {
    console.log('In data[user] condition');
    App.db.data[user] = {'name': user};
    console.log('value of data[user].name: ' + App.db.data[user].name);
  }
  if (App.db.data[user].Installer == undefined) {
    App.db.data[user].Installer = {};
    console.log('');
  }
  App.db.data[user].Installer.currentStep = currentStep;
  App.db.data[user].Installer.completed = completed;
  localStorage.setObject('ambari', App.db.data);
}

App.db.setClusterName = function (name) {
  console.log('TRACE: Entering db:setClusterName function');
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  // all information from Installer.ClusterName will be transferred to clusters[ClusterName] when app migrates from installer to main
  if (App.db.data[user] == undefined) {
    App.db.data[user] = {'name': user};
  }
  if (App.db.data[user].clusters == undefined) {
    App.db.data[user].clusters = {};
  }

  if (App.db.data[user].Installer == undefined) {
    App.db.data[user].Installer = {};
  }
  App.db.data[user].Installer.ClusterName = name;
  localStorage.setObject('ambari', App.db.data);
}

App.db.setAllHostNames = function (hostNames) {
  console.log('TRACE: Entering db:setAllHostNames function');
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  if (App.db.data[user] == undefined) {
    App.db.data[user] = {'name': user};
  }
  App.db.data[user].Installer.hostNames = hostNames;
  localStorage.setObject('ambari', App.db.data);
}

App.db.setHosts = function (hostInfo) {
  console.log('TRACE: Entering db:setHosts function');
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  if (App.db.data[user] == undefined) {
    App.db.data[user] = {'name': user};
  }
  App.db.data[user].Installer.hostInfo = hostInfo;
  localStorage.setObject('ambari', App.db.data);
}

App.db.setInstallType = function (installType) {
  console.log('TRACE: Entering db:setInstallType function');
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  if (App.db.data[user] == undefined) {
    App.db.data[user] = {'name': user};
  }
  App.db.data[user].Installer.installType = installType;
  localStorage.setObject('ambari', App.db.data);
}

App.db.setSoftRepo = function (softRepo) {
  console.log('TRACE: Entering db:setSoftRepo function');
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  if (App.db.data[user] == undefined) {
    App.db.data[user] = {'name': user};
  }
  App.db.data[user].Installer.softRepo = softRepo;
  localStorage.setObject('ambari', App.db.data);
}

App.db.setBootStatus = function (status) {
  console.log('TRACE: Entering db:setService function');
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  App.db.data[user].Installer.bootStatus = status;
  localStorage.setObject('ambari', App.db.data);
}

App.db.removeHosts = function (hostInfo) {
  console.log('TRACE: Entering db:setSoftRepo function');
  var hostList = App.db.getHosts();
  hostInfo.forEach(function (_hostInfo) {
    var host = _hostInfo.hostName;
    delete hostList[host];
  });
  App.db.setHosts(hostList);
}

App.db.setService = function (serviceInfo) {
  console.log('TRACE: Entering db:setService function');
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  App.db.data[user].Installer.serviceInfo = serviceInfo;
  localStorage.setObject('ambari', App.db.data);
}

App.db.setSelectedServiceNames = function (serviceNames) {
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  App.db.data[user].Installer.selectedServiceNames = serviceNames;
  localStorage.setObject('ambari', App.db.data);
}

App.db.setClientsForSelectedServices = function(clientInfo) {
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  App.db.data[user].Installer.clientInfo = clientInfo;
  localStorage.setObject('ambari', App.db.data);
}

App.db.setMasterComponentHosts = function (masterComponentHosts) {
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  App.db.data[user].Installer.masterComponentHosts = masterComponentHosts;
  localStorage.setObject('ambari', App.db.data);
}

App.db.setHostToMasterComponent = function (hostToMasterComponent) {
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  App.db.data[user].Installer.hostToMasterComponent = hostToMasterComponent;
  localStorage.setObject('ambari', App.db.data);
}

App.db.setHostSlaveComponents = function (hostSlaveComponents) {
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  App.db.data[user].Installer.hostSlaveComponents = hostSlaveComponents;
  localStorage.setObject('ambari', App.db.data);
}

App.db.setSlaveComponentHosts = function (slaveComponentHosts) {
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  App.db.data[user].Installer.slaveComponentHosts = slaveComponentHosts;
  localStorage.setObject('ambari', App.db.data);
}

App.db.setServiceConfigs = function (serviceConfigs) {
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  App.db.data[user].Installer.serviceConfigs = serviceConfigs;
  localStorage.setObject('ambari', App.db.data);
}

App.db.setServiceConfigProperties = function (configProperties) {
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  App.db.data[user].Installer.configProperties = configProperties;
  localStorage.setObject('ambari', App.db.data);
}

App.db.setClusterStatus = function (status) {
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  App.db.data[user].Installer.clusterStatus = status;
  localStorage.setObject('ambari', App.db.data);
}

/*
 *  getter methods
 */

// that works incorrectly
App.db.getUser = function () {
  console.log('TRACE: Entering db:getUser function');
  App.db.data = localStorage.getObject('ambari');
  return App.db.data.app.user;
}

App.db.getLoginName = function () {
  console.log('Trace: Entering db:getLoginName function');
  App.db.data = localStorage.getObject('ambari');
  return App.db.data.app.loginName;
}

App.db.getAuthenticated = function () {
  console.log('Trace: Entering db:getAuthenticated function');
  App.db.data = localStorage.getObject('ambari');
  return App.db.data.app.authenticated;
}

App.db.getSection = function () {
  console.log('Trace: Entering db:getSection function');
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName
  if (App.db.data[user] == undefined || App.db.data[user] == '') {
    return 0;
  }
  return App.db.data[user].section;
}

App.db.getClusterName = function () {
  console.log('Trace: Entering db:getClusterName function');
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  if (user) {
    return App.db.data[user].Installer.ClusterName;
  }
};

/**
 * Return current step for specified Wizard Type
 * @param wizardType
 * @return {*}
 */
App.db.getWizardCurrentStep = function (wizardType) {
  console.log('Trace: Entering db:getWizardCurrentStep function for ', wizardType);
  var user = this.getUser();
  if (App.db.data[user] && App.db.data[user][wizardType.capitalize()]) {
    return App.db.data[user][wizardType.capitalize()].currentStep;
  }
  return 0;
};

/**
 * Set current step value for specified Wizard Type
 * @param wizardType
 * @param currentStep
 * @param completed
 */
App.db.setWizardCurrentStep = function (wizardType, currentStep, completed) {
  console.log('TRACE: Entering db:setWizardCurrentStep function');

  var user = this.getUser();
  if (!App.db.data[user]) {
    App.db.data[user] = {name: user};
    console.log('value of data[user].name: ' + App.db.data[user].name);
  }
  App.db.data[user][wizardType.capitalize()] = {
    currentStep: currentStep,
    completed: completed
  };
  localStorage.setObject('ambari', App.db.data);
};

App.db.getInstallerCurrentStep = function () {
  console.log('Trace: Entering db:getInstallerCurrentStep function');
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  if (App.db.data[user] == undefined || App.db.data[user] == '') {
    return 0;
  }
  return App.db.data[user].Installer.currentStep;
},

  App.db.getAllHostNames = function () {
    console.log('TRACE: Entering db:getHostNames function');
    App.db.data = localStorage.getObject('ambari');
    var user = App.db.data.app.loginName;
    return App.db.data[user].Installer.hostNames;
  },

  App.db.getInstallType = function () {
    console.log('TRACE: Entering db:getHostNames function');
    App.db.data = localStorage.getObject('ambari');
    var user = App.db.data.app.loginName;
    return App.db.data[user].Installer.installType;
  },

  App.db.getSoftRepo = function () {
    console.log('TRACE: Entering db:getSoftRepo function');
    App.db.data = localStorage.getObject('ambari');
    var user = App.db.data.app.loginName;
    return App.db.data[user].Installer.softRepo;
  }

App.db.isCompleted = function () {
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  return App.db.data[user].Installer.completed;
}

App.db.getHosts = function () {
  console.log('TRACE: Entering db:getHosts function');
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  if (App.db.data[user] == undefined || App.db.data[user] == '') {
    console.log('ERROR: loginName required for storing host info');
    return 0;
  }
  return App.db.data[user].Installer.hostInfo;
}

App.db.getBootStatus = function () {
  console.log('TRACE: Entering db:setService function');
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  return App.db.data[user].Installer.bootStatus;
}

App.db.getService = function () {
  console.log('TRACE: Entering db:getService function');
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  return App.db.data[user].Installer.serviceInfo;
}

App.db.getSelectedServiceNames = function () {
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  return App.db.data[user].Installer.selectedServiceNames;
}

App.db.getClientsForSelectedServices = function() {
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  return App.db.data[user].Installer.clientInfo;
}

App.db.getMasterComponentHosts = function () {
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  return App.db.data[user].Installer.masterComponentHosts;
}

App.db.getHostToMasterComponent = function () {
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  return App.db.data[user].Installer.hostToMasterComponent;
}

App.db.getHostSlaveComponents = function () {
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  return App.db.data[user].Installer.hostSlaveComponents;
}

App.db.getSlaveComponentHosts = function () {
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  return App.db.data[user].Installer.slaveComponentHosts;
}

App.db.getServiceConfigs = function () {
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  return App.db.data[user].Installer.serviceConfigs;
}

App.db.getServiceConfigProperties = function () {
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  return App.db.data[user].Installer.configProperties;
}

App.db.getClusterStatus = function () {
  console.log('TRACE: Entering db:setService function');
  App.db.data = localStorage.getObject('ambari');
  var user = App.db.data.app.loginName;
  return App.db.data[user].Installer.clusterStatus;
}

module.exports = App.db;