kata-cli
Version:
Kata AI Command Line Tools
536 lines • 23.8 kB
JavaScript
;
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
Object.defineProperty(exports, "__esModule", { value: true });
const merapi_1 = require("merapi");
const inquirer = require("inquirer");
const Table = require("cli-table");
const colors = require("colors");
class User extends merapi_1.Component {
constructor(helper, api) {
super();
this.helper = helper;
this.api = api;
}
login(options) {
return __awaiter(this, void 0, void 0, function* () {
try {
let currToken = this.helper.getCurrentToken().token;
if (options.token) {
if (!currToken) {
currToken = options.token;
}
this.api.authApi.apiClient.defaultHeaders.Authorization = `Bearer ${currToken}`;
let result = yield this.helper.toPromise(this.api.authApi, this.api.authApi.tokensTokenIdGet, options.token);
const tokenObj = result.data;
if (tokenObj.type === "user") {
this.setToken({ name: "user", type: tokenObj.type }, options.token);
}
else if (tokenObj.type === "team") {
result = yield this.helper.toPromise(this.api.userApi, this.api.userApi.usersUserIdGet, tokenObj.teamId);
const team = result.data;
this.setToken({ name: team.username, type: tokenObj.type }, options.token);
}
else {
throw new Error("Invalid token");
}
}
else {
let user = (options.user ? options.user : "");
let pass = (options.password ? options.password : "");
const answer = yield inquirer.prompt([
{
type: "input",
name: "user",
message: "username: ",
when() {
return !user;
},
validate(user) {
if (!user) {
return "Username cannot be empty";
}
return true;
}
},
{
type: "password",
name: "password",
message: "password: ",
mask: "*",
when() {
return !pass;
},
validate(password) {
if (!password) {
return "Password cannot be empty";
}
return true;
}
}
]);
user = answer.user || user;
pass = answer.password || pass;
const result = yield this.helper.toPromise(this.api.authApi, this.api.authApi.cliLoginPost, { username: user, password: pass });
if (!result.data.isLoggedIn && !result.data.id) {
console.log(`username or password is incorrect`);
}
else {
const token = result.data.id;
this.helper.setProp("first_login", { type: "user", username: user, id: result.data.userId });
this.setToken({ name: user, type: "user", namespace: "platform" }, token);
console.log(`Logged in as ${colors.green(user)}`);
}
}
}
catch (e) {
console.log(this.helper.wrapError(e));
}
});
}
logout() {
return __awaiter(this, void 0, void 0, function* () {
const answer = yield this.helper.inquirerPrompt([
{
type: "confirm",
name: "confirmation",
message: "Do you want to log out ?",
default: false
}
]);
if (!answer.confirmation) {
return;
}
try {
const result = yield this.helper.delete();
if (result) {
console.log("Logged out");
}
else {
console.log("Please log in first");
}
this.helper.clearCommandSession();
}
catch (e) {
console.log(this.helper.wrapError(e));
}
});
}
switch(type, name) {
return __awaiter(this, void 0, void 0, function* () {
try {
const firstLogin = this.helper.getProp("first_login");
const currentType = this.helper.getProp("current_user_type");
const currentLogin = this.helper.getProp("current_login");
const username = name ? name : currentLogin;
if (currentType === type && username === currentLogin) {
throw new Error(`Unable to switch : already on ${colors.green(currentLogin)} as ${colors.green(type)}`);
}
const isImpersonate = this.helper.getProp("isImpersonate");
if (type === "team") {
let userId;
if (isImpersonate === true) {
const currentLoginName = this.helper.getProp("current_login").toString();
userId = (yield this.getUserInfo(currentLoginName)).id;
}
else {
userId = firstLogin.id.toString();
}
const { response } = yield this.helper.toPromise(this.api.userApi, this.api.userApi.usersUserIdGet, userId);
if (!response) {
throw new Error(`Unable to switch team: ${colors.green(name)}`);
}
const teams = response && response.body ?
response.body.teams.filter((team) => team.username === name) : [];
if (teams.length > 0) {
const result = yield this.helper.toPromise(this.api.authApi, this.api.authApi.tokensPost, {
type: "team",
teamId: teams[0].teamId
});
const token = result.data.id;
this.setToken({ name, type: "team" }, token);
this.helper.setProp("current_login", name);
this.helper.setProp("current_user_type", "team");
console.log(`Switched to team: ${colors.green(name)}`);
}
else {
console.log(`Invalid team name. Unable to switch to team: ${colors.red(name)}`);
}
}
else if (type === "user") {
if (isImpersonate === true) {
// TODO: jika user login sbg admin, impersonate dewi, switch ke team, lalu mau switch ke user yg non-user tsb
const userTokenInfo = this.getUserTokenInfo(name);
if (!userTokenInfo || userTokenInfo === undefined) {
throw new Error(`${colors.red(name)}'s token is not found.`);
}
else if (userTokenInfo.userName === "admin") {
throw new Error(`Cannot switch to ${colors.red(name)}. Use unimpersonate.`);
}
else {
if (name === userTokenInfo.userName) {
this.helper.setProp("current_login", name);
this.helper.setProp("current_user_type", "user");
console.log(`Switched to user ${colors.green(name)}`);
}
else {
throw new Error(`Unable to switch to user ${colors.red(name)}.`);
}
}
}
else {
this.helper.setProp("current_login", firstLogin.username);
this.helper.setProp("current_user_type", "user");
console.log(`Switched to user ${colors.green(firstLogin.username)}`);
}
}
}
catch (e) {
console.log(this.helper.wrapError(e));
}
});
}
changePassword() {
return __awaiter(this, void 0, void 0, function* () {
try {
const passObj = yield this.getNewPasswordData();
if (passObj.newPass !== passObj.rePass) {
throw new Error("Invalid retype password");
}
const currentUser = this.helper.getProp("current_login");
const { data } = yield this.helper.toPromise(this.api.authApi, this.api.authApi.loginPost, { username: currentUser, password: passObj.oldPass });
if (data) {
const result = yield this.helper.toPromise(this.api.userApi, this.api.userApi.usersUserIdPut, currentUser, { password: passObj.newPass });
console.log("Password changed");
}
else {
console.log("Invalid password");
}
}
catch (error) {
console.log(this.helper.wrapError(error));
}
});
}
whoami(options) {
const currentLogin = this.helper.getProp("current_login");
const currentType = this.helper.getProp("current_user_type");
console.log(`Current login: ${colors.green(currentLogin)}, login type: ${colors.green(currentType)}`);
}
createTeam(name) {
return __awaiter(this, void 0, void 0, function* () {
try {
const currentLogin = this.helper.getProp("current_login");
const currentUserType = this.helper.getProp("current_user_type");
if (currentUserType !== "user") {
throw new Error("Must be on user to do this operation");
}
const { response } = yield this.helper.toPromise(this.api.teamApi, this.api.teamApi.teamsPost, { username: name, password: "", roleId: "teamAdmin" });
if (response && response.body.id) {
console.log(`Team ${colors.green(name)} created !`);
}
else {
console.log(`Team ${colors.red(name)} exist !`);
}
}
catch (error) {
console.log(this.helper.wrapError(error));
}
});
}
createUser(username, options) {
return __awaiter(this, void 0, void 0, function* () {
const password = yield this.helper.inquirerPrompt([
{
type: "password",
name: "answer",
message: "password: ",
mask: "*",
default: null
}
]);
const confirmPassword = yield this.helper.inquirerPrompt([
{
type: "password",
name: "answer",
message: "retype password: ",
mask: "*",
default: null
}
]);
try {
if (password.answer !== confirmPassword.answer) {
throw new Error("Invalid retype password");
}
let role;
if (options.admin) {
role = "admin";
}
else if (options.internal) {
role = "internalUser";
}
else {
role = "user";
}
const { data } = yield this.helper.toPromise(this.api.userApi, this.api.userApi.usersUserIdGet, username);
if (data.id) {
throw new Error(`Username ${username} exist !`);
}
const newUser = yield this.helper.toPromise(this.api.userApi, this.api.userApi.usersPost, { username, password: password.answer, roleId: role });
console.log(`New user ${colors.green(newUser.data.username)} created !`);
}
catch (error) {
console.log(this.helper.wrapError(error));
}
});
}
impersonate(userName) {
return __awaiter(this, void 0, void 0, function* () {
// TODO : dibuat seperti login, pake inquirer.
try {
const currentLogin = this.helper.getProp("current_login").toString();
if (currentLogin !== "admin") {
throw new Error(`Your login status is not superadmin. You are not authorized to impersonate a user`);
}
// set currentToken header bearer token
const currentToken = this.helper.getCurrentToken().token.toString();
this.api.authApi.apiClient.defaultHeaders.Authorization = `Bearer ${currentToken}`;
// get admin token
this.api.authApi.apiClient.defaultHeaders.Authorization = `Bearer ${currentToken}`;
// get user id from username
const id = (yield this.getUserInfo(userName)).id;
const name = (yield this.getUserInfo(userName)).name;
const email = (yield this.getUserInfo(userName)).email;
if (name && email) {
// userName is user collected input
if (userName !== name && userName !== email) {
throw new Error(`Sorry, username is not exist.`);
}
}
else {
throw new Error(`Sorry, username is not valid.`);
}
// impersonate function
const result = yield this.helper.toPromise(this.api.authApi, this.api.authApi.impersonatePost, {
userId: id,
namespace: "platform"
});
// set value on .katajson
const impersonateToken = result.data.id.toString();
const type = result.data.type.toString();
this.helper.setProp("isImpersonate", true);
this.setToken({ name, type }, impersonateToken);
console.log(`Succesfully impersonate as ${colors.green(name)}`);
}
catch (error) {
console.log(this.helper.wrapError(error));
}
});
}
// unimpersonate command
unimpersonate() {
return __awaiter(this, void 0, void 0, function* () {
try {
const userName = this.helper.getProp("current_login").toString();
this.helper.deleteKeyToken(userName);
const currentLogin = this.helper.getProp("current_login").toString();
console.log(`Succesfully unimpersonate user. Now your current login is ${colors.green(currentLogin)}`);
}
catch (error) {
console.log(this.helper.wrapError(error));
}
});
}
setToken(userInfo, token) {
this.helper.setProp("current_login", userInfo.name);
this.helper.setProp("current_user_type", userInfo.type);
this.helper.setProp("namespace", userInfo.namespace);
const tokenProp = (this.helper.getProp("token") || {});
tokenProp[userInfo.name] = token;
this.helper.setProp("token", tokenProp);
}
getUserTokenInfo(name) {
const tokenProp = (this.helper.getProp("token") || {});
if (name in tokenProp) {
const userName = name.toString();
const token = tokenProp[name].toString();
return {
userName,
token
};
}
}
getUserInfo(userName) {
return __awaiter(this, void 0, void 0, function* () {
// get userId from currentlogin
const { response } = yield this.helper.toPromise(this.api.userApi, this.api.userApi.usersGetInfoKeyGet, userName);
const user = response.body;
const email = user.email ? user.email.toString() : "";
const name = user.username.toString();
const id = user.userId.toString();
return {
id,
name,
email,
};
});
}
getNewPasswordData() {
return __awaiter(this, void 0, void 0, function* () {
let oldPass;
let newPass;
let rePass;
const answer = yield inquirer.prompt([
{
type: "password",
name: "oldPass",
message: "current password: ",
mask: "*",
when() {
return !oldPass;
},
validate(user) {
if (!user) {
return "Password cannot be empty";
}
return true;
}
},
{
type: "password",
name: "newPass",
message: "new password: ",
mask: "*",
when() {
return !newPass;
},
validate(password) {
if (!password) {
return "Password cannot be empty";
}
return true;
}
},
{
type: "password",
name: "rePass",
message: "retype new password: ",
mask: "*",
when() {
return !rePass;
},
validate(password, answer) {
if (!password) {
return "Password cannot be empty";
}
if (password !== answer.newPass) {
return "Invalid retype password";
}
return true;
}
},
]);
return answer;
});
}
forgot(username) {
return __awaiter(this, void 0, void 0, function* () {
try {
const current_login = this.helper.getProp("current_login");
if (!current_login) {
const { response } = yield this.helper.toPromise(this.api.authApi, this.api.authApi.forgotPost, { username });
if (response && response.body && response.body.message) {
console.log("Please check your email to reset your password.");
}
}
else {
console.log(`Please log out first`);
}
}
catch (e) {
console.error(this.helper.wrapError(e));
}
});
}
listTeam() {
return __awaiter(this, void 0, void 0, function* () {
try {
const current_login = this.helper.getProp("current_login");
if (current_login) {
const { response } = yield this.helper.toPromise(this.api.userApi, this.api.userApi.usersUserIdTeamsGet, current_login);
const table = new Table({
head: ["Team Name", "Projects", "Members", "Bots"],
colWidths: [50, 15, 15, 15]
});
response.body.forEach((team) => {
table.push([team.username, team.projects, team.members, team.bots]);
});
console.log(table.toString());
}
else {
console.log("Please log in first");
}
}
catch (e) {
console.error(this.helper.wrapError(e));
}
});
}
listTeamUser(teamName) {
return __awaiter(this, void 0, void 0, function* () {
try {
const current_login = this.helper.getProp("current_login");
if (current_login) {
const dataTeams = yield this.helper.toPromise(this.api.userApi, this.api.userApi.usersUserIdTeamsGet, current_login);
const teams = dataTeams.response.body;
const choices = teams.map((team) => ({
name: team.username,
value: team.teamId
}));
let teamId = null;
if (teamName) {
const sameName = choices.find((choice) => choice.name === teamName);
if (sameName) {
teamId = sameName.value;
}
}
else {
const choice = yield inquirer.prompt([
{
type: "list",
name: "teamId",
message: "Team:",
choices: choices
}
]);
teamId = choice.teamId;
}
const { response } = yield this.helper.toPromise(this.api.teamApi, this.api.teamApi.teamsTeamIdUsersGet, teamId);
const table = new Table({
head: ["Username", "Role"],
colWidths: [50, 25]
});
response.body.forEach((user) => {
table.push([user.username, user.roleName]);
});
console.log(table.toString());
}
else {
console.log("Please log in first");
}
}
catch (e) {
console.error(this.helper.wrapError(e));
}
});
}
}
exports.default = User;
//# sourceMappingURL=user.js.map