hive.ko.js 17 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585
  1. // Licensed to Cloudera, Inc. under one
  2. // or more contributor license agreements. See the NOTICE file
  3. // distributed with this work for additional information
  4. // regarding copyright ownership. Cloudera, Inc. licenses this file
  5. // to you under the Apache License, Version 2.0 (the
  6. // "License"); you may not use this file except in compliance
  7. // with the License. You may obtain a copy of the License at
  8. //
  9. // http://www.apache.org/licenses/LICENSE-2.0
  10. //
  11. // Unless required by applicable law or agreed to in writing, software
  12. // distributed under the License is distributed on an "AS IS" BASIS,
  13. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14. // See the License for the specific language governing permissions and
  15. // limitations under the License.
  16. var Privilege = function (vm, privilege) {
  17. var self = this;
  18. self.status = ko.observable(typeof privilege.status != "undefined" && privilege.status != null ? privilege.status : "");
  19. self.editing = ko.observable(typeof privilege.editing != "undefined" && privilege.editing != null ? privilege.editing : false);
  20. //self.privilegeScope = ko.observable(typeof privilege.privilegeScope != "undefined" && privilege.privilegeScope != null ? privilege.privilegeScope : "");
  21. self.serverName = ko.observable(typeof privilege.serverName != "undefined" && privilege.serverName != null ? privilege.serverName : "");
  22. self.serverName.subscribe(function () {
  23. if (self.status() == '') {
  24. self.status('modified');
  25. }
  26. });
  27. self.dbName = ko.observable(typeof privilege.dbName != "undefined" && privilege.dbName != null ? privilege.dbName : "");
  28. self.dbName.subscribe(function () {
  29. if (self.status() == '') {
  30. self.status('modified');
  31. }
  32. });
  33. self.tableName = ko.observable(typeof privilege.tableName != "undefined" && privilege.tableName != null ? privilege.tableName : "");
  34. self.tableName.subscribe(function () {
  35. if (self.status() == '') {
  36. self.status('modified');
  37. }
  38. });
  39. self.URI = ko.observable(typeof privilege.URI != "undefined" && privilege.URI != null ? privilege.URI : "");
  40. self.URI.subscribe(function () {
  41. if (self.status() == '') {
  42. self.status('modified');
  43. }
  44. });
  45. self.action = ko.observable(typeof privilege.action != "undefined" && privilege.action != null ? privilege.action : "");
  46. self.action.subscribe(function () {
  47. if (self.status() == '') {
  48. self.status('modified');
  49. }
  50. });
  51. // UI
  52. self.showAdvanced = ko.observable(false);
  53. self.path = ko.computed({
  54. read: function () {
  55. if (self.tableName().length > 0) {
  56. return self.dbName() + "." + self.tableName();
  57. } else {
  58. return self.dbName();
  59. }
  60. },
  61. write: function (value) {
  62. var lastSpacePos = value.lastIndexOf(".");
  63. if (lastSpacePos > 0) {
  64. this.dbName(value.substring(0, lastSpacePos));
  65. this.tableName(value.substring(lastSpacePos + 1));
  66. } else {
  67. this.dbName(value);
  68. this.tableName('');
  69. }
  70. },
  71. owner: self
  72. });
  73. self.privilegeScope = ko.computed(function() {
  74. if (self.tableName().length > 0) {
  75. return 'TABLE';
  76. } else if (self.dbName().length > 0) {
  77. return 'DATABASE';
  78. } else {
  79. return 'SERVER';
  80. }
  81. });
  82. self.availablePrivileges = ko.observableArray(['SERVER', 'DATABASE', 'TABLE']);
  83. self.availableActions = ko.observableArray(['SELECT', 'INSERT', 'ALL', '']);
  84. self.remove = function (privilege) {
  85. privilege.status('deleted');
  86. }
  87. }
  88. var Role = function (vm, role) {
  89. var self = this;
  90. self.name = ko.observable(typeof role.name != "undefined" && role.name != null ? role.name : "");
  91. self.grantorPrincipal = ko.observable(typeof role.grantorPrincipal != "undefined" && role.grantorPrincipal != null ? role.grantorPrincipal : "");
  92. self.groups = ko.observableArray();
  93. $.each(typeof role.groups != "undefined" && role.groups != null ? role.groups : [], function (index, group) {
  94. self.groups.push(group);
  95. });
  96. self.privileges = ko.observableArray(); // Not included in the API
  97. self.originalPrivileges = ko.observableArray();
  98. self.showPrivileges = ko.observable(false);
  99. self.privilegesChanged = ko.computed(function () {
  100. return $.grep(self.privileges(), function (privilege) {
  101. return ['new', 'deleted', 'modified'].indexOf(privilege.status()) != -1;
  102. });
  103. });
  104. self.reset = function () {
  105. self.name('');
  106. self.groups.removeAll();
  107. self.privileges.removeAll();
  108. self.originalPrivileges.removeAll();
  109. }
  110. self.addGroup = function () {
  111. self.groups.push('');
  112. }
  113. self.addPrivilege = function () {
  114. self.privileges.push(new Privilege(vm, {'serverName': vm.assist.server(), 'status': 'new', 'editing': true}));
  115. }
  116. self.create = function () {
  117. $(".jHueNotify").hide();
  118. $.post("/security/api/hive/create_role", {
  119. role: ko.mapping.toJSON(self)
  120. }, function (data) {
  121. if (data.status == 0) {
  122. $(document).trigger("info", data.message);
  123. vm.roles.unshift(new Role(vm, data.role));
  124. self.reset();
  125. vm.showCreateRole(false);
  126. } else {
  127. $(document).trigger("error", data.message);
  128. }
  129. }).fail(function (xhr, textStatus, errorThrown) {
  130. $(document).trigger("error", xhr.responseText);
  131. });
  132. }
  133. self.remove = function (role) {
  134. $(".jHueNotify").hide();
  135. $.post("/security/api/hive/drop_sentry_role", {
  136. roleName: role.name
  137. }, function (data) {
  138. if (data.status == 0) {
  139. $(document).trigger("info", data.message);
  140. vm.removeRole(role.name);
  141. }
  142. else {
  143. $(document).trigger("error", data.message);
  144. }
  145. }).fail(function (xhr, textStatus, errorThrown) {
  146. $(document).trigger("error", xhr.responseText);
  147. });
  148. }
  149. self.savePrivileges = function (role) {
  150. $(".jHueNotify").hide();
  151. $.post("/security/api/hive/save_privileges", {
  152. role: ko.mapping.toJSON(role)
  153. }, function (data) {
  154. if (data.status == 0) {
  155. vm.list_sentry_privileges_by_role(role); // Refresh all role privileges
  156. } else {
  157. $(document).trigger("error", data.message);
  158. }
  159. }).fail(function (xhr, textStatus, errorThrown) {
  160. $(document).trigger("error", xhr.responseText);
  161. });
  162. }
  163. }
  164. var Assist = function (vm) {
  165. var self = this;
  166. self.compareNames = function (a, b) {
  167. if (a.name.toLowerCase() < b.name.toLowerCase())
  168. return -1;
  169. if (a.name.toLowerCase() > b.name.toLowerCase())
  170. return 1;
  171. return 0;
  172. }
  173. self.path = ko.observable('');
  174. self.path.subscribe(function () {
  175. self.fetchHivePath();
  176. });
  177. self.server = ko.observable('');
  178. self.db = ko.computed(function () {
  179. return self.path().split(/[.]/)[0];
  180. });
  181. self.table = ko.computed(function () {
  182. return self.path().split(/[.]/)[1];
  183. });
  184. self.privileges = ko.observableArray();
  185. self.isDiffMode = ko.observable(false);
  186. self.isDiffMode = ko.observable(false);
  187. self.isDiffMode.subscribe(function () {
  188. //self.refreshTree();
  189. });
  190. self.isLoadingTree = ko.observable(false);
  191. self.treeAdditionalData = {};
  192. self.treeData = ko.observable({nodes: []});
  193. self.loadData = function (data) {
  194. self.treeData(new TreeNodeModel(data));
  195. };
  196. self.initialGrowingTree = {
  197. path: "__HUEROOT__",
  198. name: "__HUEROOT__",
  199. selected: false,
  200. nodes: []
  201. };
  202. self.growingTree = ko.observable(jQuery.extend(true, {}, self.initialGrowingTree));
  203. self.addDatabases = function (databases) {
  204. var _tree = self.growingTree();
  205. databases.forEach(function (db) {
  206. var _mainFound = false;
  207. _tree.nodes.forEach(function (node) {
  208. if (node.path == db) {
  209. _mainFound = true;
  210. }
  211. });
  212. if (!_mainFound) {
  213. var _item = {
  214. path: db,
  215. name: db,
  216. isDb: true,
  217. isTable: false,
  218. isColumn: false,
  219. isExpanded: false,
  220. nodes: []
  221. };
  222. _tree.nodes.push(_item);
  223. }
  224. });
  225. self.loadData(self.growingTree());
  226. }
  227. self.addTables = function (tables) {
  228. var _branch = self.growingTree();
  229. _branch.nodes.forEach(function (node) {
  230. if (node.path == self.path()) {
  231. _branch = node;
  232. }
  233. });
  234. tables.forEach(function (table) {
  235. var _mainFound = false;
  236. var _path = self.path() + "." + table;
  237. _branch.nodes.forEach(function (node) {
  238. if (node.path == _path) {
  239. _mainFound = true;
  240. }
  241. });
  242. if (!_mainFound) {
  243. var _item = {
  244. path: _path,
  245. name: table,
  246. isDb: false,
  247. isTable: true,
  248. isColumn: false,
  249. isExpanded: false,
  250. nodes: []
  251. };
  252. _branch.nodes.push(_item);
  253. }
  254. });
  255. self.loadData(self.growingTree());
  256. }
  257. self.addColumns = function (columns) {
  258. var _branch = self.growingTree();
  259. _branch.nodes.forEach(function (node) {
  260. if (node.path == self.path().split(".")[0]) {
  261. node.nodes.forEach(function (inode) {
  262. if (inode.path == self.path()) {
  263. _branch = inode;
  264. }
  265. });
  266. }
  267. });
  268. columns.forEach(function (column) {
  269. var _mainFound = false;
  270. var _path = self.path() + "." + column;
  271. _branch.nodes.forEach(function (node) {
  272. if (node.path == _path) {
  273. _mainFound = true;
  274. }
  275. });
  276. if (!_mainFound) {
  277. var _item = {
  278. path: _path,
  279. name: column,
  280. isDb: false,
  281. isTable: false,
  282. isColumn: true,
  283. isExpanded: false,
  284. nodes: []
  285. };
  286. _branch.nodes.push(_item);
  287. }
  288. });
  289. self.loadData(self.growingTree());
  290. }
  291. self.setPath = function (obj, toggle) {
  292. if (self.getTreeAdditionalDataForPath(obj.path()).loaded || (!obj.isExpanded() && !self.getTreeAdditionalDataForPath(obj.path()).loaded)) {
  293. if (typeof toggle == "boolean" && toggle) {
  294. obj.isExpanded(!obj.isExpanded());
  295. }
  296. else {
  297. obj.isExpanded(true);
  298. }
  299. self.updatePathProperty(self.growingTree(), obj.path(), "isExpanded", obj.isExpanded());
  300. }
  301. self.path(obj.path());
  302. }
  303. self.togglePath = function (obj) {
  304. self.setPath(obj, true);
  305. }
  306. self.getTreeAdditionalDataForPath = function (path) {
  307. if (typeof self.treeAdditionalData[path] == "undefined") {
  308. var _add = {
  309. loaded: false
  310. }
  311. self.treeAdditionalData[path] = _add;
  312. }
  313. return self.treeAdditionalData[path];
  314. }
  315. self.updatePathProperty = function (leaf, path, property, value) {
  316. if (leaf.path == path) {
  317. leaf[property] = value;
  318. }
  319. if (leaf.nodes.length > 0) {
  320. leaf.nodes.forEach(function (node) {
  321. self.updatePathProperty(node, path, property, value);
  322. });
  323. }
  324. return leaf;
  325. }
  326. self.updateTreeProperty = function (leaf, property, value) {
  327. leaf[property] = value;
  328. if (leaf.nodes.length > 0) {
  329. leaf.nodes.forEach(function (node) {
  330. self.updateTreeProperty(node, property, value);
  331. });
  332. }
  333. return leaf;
  334. }
  335. self.collapseTree = function () {
  336. self.updateTreeProperty(self.growingTree(), "isExpanded", false);
  337. self.updatePathProperty(self.growingTree(), "/", "isExpanded", true);
  338. self.loadData(self.growingTree());
  339. }
  340. self.collapseOthers = function () {
  341. self.updateTreeProperty(self.growingTree(), "isExpanded", false);
  342. self.updatePathProperty(self.growingTree(), "/", "isExpanded", true);
  343. var _path = self.path();
  344. var _crumb = "";
  345. for (var i = 0; i < _path.length; i++) {
  346. if ((_path[i] === "/" && _crumb != "")) {
  347. self.updatePathProperty(self.growingTree(), _crumb, "isExpanded", true);
  348. }
  349. _crumb += _path[i];
  350. }
  351. self.updatePathProperty(self.growingTree(), _path, "isExpanded", true);
  352. self.loadData(self.growingTree());
  353. }
  354. self.fetchHivePath = function () {
  355. if (self.path().split(".").length < 3) {
  356. var _path = self.path().replace('.', '/');
  357. var request = {
  358. url: '/beeswax/api/autocomplete/' + _path,
  359. dataType: 'json',
  360. type: 'GET',
  361. success: function (data) {
  362. if (data.databases) {
  363. self.addDatabases(data.databases);
  364. }
  365. else if (data.tables && data.tables.length > 0) {
  366. self.addTables(data.tables);
  367. var tables = $.map(data.tables, function (table, index) {
  368. return self.db() + '.' + table;
  369. });
  370. }
  371. else if (data.columns && data.columns.length > 0) {
  372. self.addColumns(data.columns);
  373. }
  374. vm.list_sentry_privileges_by_authorizable();
  375. },
  376. cache: false
  377. };
  378. $.ajax(request).fail(function (xhr, textStatus, errorThrown) {
  379. $(document).trigger("error", xhr.responseText);
  380. });
  381. }
  382. else {
  383. vm.list_sentry_privileges_by_authorizable();
  384. }
  385. };
  386. self.afterRender = function () {
  387. $(document).trigger("rendered.tree");
  388. }
  389. }
  390. var HiveViewModel = function (initial) {
  391. var self = this;
  392. // Models
  393. self.roles = ko.observableArray();
  394. self.availableHadoopGroups = ko.mapping.fromJS(initial.hadoop_groups);
  395. self.assist = new Assist(self);
  396. // Editing
  397. self.showCreateRole = ko.observable(false);
  398. self.role = new Role(self, {});
  399. self.privilege = new Privilege(self, {});
  400. self.doAs = ko.observable(initial.user);
  401. self.doAs.subscribe(function () {
  402. self.assist.fetchHivePath();
  403. });
  404. self.availableHadoopUsers = ko.observableArray();
  405. self.availableHadoopGroups = ko.observableArray();
  406. self.selectableHadoopUsers = ko.computed(function () {
  407. var _users = ko.utils.arrayMap(self.availableHadoopUsers(), function (user) {
  408. return user.username;
  409. });
  410. return _users.sort();
  411. }, self);
  412. self.selectableHadoopGroups = ko.computed(function () {
  413. var _users = ko.utils.arrayMap(self.availableHadoopGroups(), function (group) {
  414. return group.name;
  415. });
  416. return _users.sort();
  417. }, self);
  418. self.init = function () {
  419. self.fetchUsers();
  420. self.list_sentry_roles_by_group();
  421. self.assist.fetchHivePath();
  422. };
  423. self.removeRole = function (roleName) {
  424. $.each(self.roles(), function (index, role) {
  425. if (role.name == roleName) {
  426. self.roles.remove(role);
  427. return false;
  428. }
  429. });
  430. };
  431. self.list_sentry_roles_by_group = function () {
  432. $.getJSON('/security/api/hive/list_sentry_roles_by_group', function (data) {
  433. if (typeof data.status !== "undefined" && data.status == -1) {
  434. $(document).trigger("error", data.message);
  435. }
  436. else {
  437. $.each(data.roles, function (index, item) {
  438. self.roles.push(new Role(self, item));
  439. });
  440. }
  441. }).fail(function (xhr, textStatus, errorThrown) {
  442. $(document).trigger("error", xhr.responseText);
  443. });
  444. };
  445. self.list_sentry_privileges_by_role = function (role) {
  446. $.ajax({
  447. type: "POST",
  448. url: "/security/api/hive/list_sentry_privileges_by_role",
  449. data: {
  450. 'roleName': role.name
  451. },
  452. success: function (data) {
  453. if (typeof data.status !== "undefined" && data.status == -1) {
  454. $(document).trigger("error", data.message);
  455. }
  456. else {
  457. role.privileges.removeAll();
  458. role.originalPrivileges.removeAll();
  459. $.each(data.sentry_privileges, function (index, item) {
  460. var privilege = _create_ko_privilege(item);
  461. role.privileges.push(privilege);
  462. role.originalPrivileges.push(privilege);
  463. });
  464. role.showPrivileges(true);
  465. }
  466. }
  467. }).fail(function (xhr, textStatus, errorThrown) {
  468. $(document).trigger("error", xhr.responseText);
  469. });
  470. };
  471. function _create_ko_privilege(privilege) {
  472. var _privilege = new Privilege(self, {
  473. 'privilegeScope': privilege.scope,
  474. 'serverName': privilege.server,
  475. 'dbName': privilege.database,
  476. 'tableName': privilege.table,
  477. 'URI': privilege.URI,
  478. 'action': privilege.action
  479. });
  480. _privilege.properties = ko.mapping.fromJS(privilege);
  481. return _privilege;
  482. }
  483. self.list_sentry_privileges_by_authorizable = function () {
  484. if (self.assist.path() != ""){
  485. $.ajax({
  486. type: "POST",
  487. url: "/security/api/hive/list_sentry_privileges_by_authorizable",
  488. data: {
  489. groups: ko.mapping.toJSON(['sambashare', 'hadoop']),
  490. roleSet: ko.mapping.toJSON({all: true, roles: []}),
  491. authorizableHierarchy: ko.mapping.toJSON({
  492. 'server': self.assist.server(),
  493. 'db': self.assist.db(),
  494. 'table': self.assist.table()
  495. })
  496. },
  497. success: function (data) {
  498. self.assist.privileges.removeAll();
  499. $.each(data.privileges, function (index, item) {
  500. self.assist.privileges.push(_create_ko_privilege(item));
  501. });
  502. }
  503. }).fail(function (xhr, textStatus, errorThrown) {
  504. $(document).trigger("error", xhr.responseText);
  505. });
  506. }
  507. };
  508. self.fetchUsers = function () {
  509. $.getJSON('/desktop/api/users/autocomplete', {
  510. 'include_myself': true,
  511. 'extend_user': true
  512. }, function (data) {
  513. self.availableHadoopUsers(data.users);
  514. self.availableHadoopGroups(data.groups);
  515. $(document).trigger("loaded.users");
  516. });
  517. }
  518. };
  519. function logGA(page) {
  520. if (typeof trackOnGA == 'function') {
  521. trackOnGA('security/' + page);
  522. }
  523. }