hive.ko.js 16 KB

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