kata-cli/components/users/team.ts

Kata AI Command Line Tools

import { IHelper } from "interfaces/main";
import { Component, JsonObject } from "merapi";

export default class Team extends Component {

    constructor(private helper: IHelper, private api: any) {
        super();
    }

    public async addMember(username: string, options?: JsonObject) {
        try {
            const role = options.admin ? "teamAdmin" : "teamMember";
            let firstLogin = this.helper.getProp("first_login") as JsonObject;

            const { userInfo, teamInfo, teamMember, currentLogin } = await this.getInfo(
                username, 
                (firstLogin.username).toString() 
            );

            if (userInfo && userInfo.id) {
                if (this.checkUser(userInfo.id, teamMember)) {
                    throw new Error(`User ${username} already on this team`);
                }
                const { response } = await this.helper.toPromise(
                    this.api.teamApi,
                    this.api.teamApi.teamsTeamIdUsersUserIdPost, 
                    teamInfo.teamId, 
                    userInfo.username,
                    { 
                        roleId: role 
                    });
                if (!response.body) { 
                    throw new Error("Error adding user to team: invalid roleId");
                }
                console.log(`Success register ${username} to ${currentLogin}`);
            } else {
                console.log(`User ${username} not found`);
            }    
        } catch (error) {
            console.log(this.helper.wrapError(error));
        }
    }
    
    public async removeMember(username: string) {
        const answer = await this.helper.inquirerPrompt([
            {
                type: "confirm",
                name: "confirmation",
                message: `Do you want to remove ${username} ?`,
                default: false
            }
        ]);

        if (!answer.confirmation) {
            return;
        }

        try {
            const { userInfo, teamInfo, teamMember, currentLogin } = await this.getInfo(username);
             if (userInfo && userInfo.id) {
                if (!this.checkUser(userInfo.id, teamMember)) {
                    throw new Error(`User ${username} not a member of this team`);
                }

                const { response } = await this.helper.toPromise(this.api.teamApi, this.api.teamApi.teamsTeamIdUsersUserIdDelete, teamInfo.id, userInfo.id);

                console.log(`Success remove ${username} from ${currentLogin}`);
            } else {
                console.log(`User ${username} not found`);
            }   

        } catch (error) {
            console.log(this.helper.wrapError(error));
        }
    }

    private async getInfo(username: string, firstLoginUsername?: string) {
        const currentLogin = this.helper.getProp("current_login") as string;
        const currentUserType = this.helper.getProp("current_user_type") as string;

        if (currentUserType !== "team") {
            throw new Error("Must be on team to do this operation");
        }

        const requestTeamData =
            await this.helper.toPromise(
                this.api.userApi, 
                this.api.userApi.usersUserIdTeamsGet, 
                firstLoginUsername
            );

        let teamInfo: any;
        if (requestTeamData.response && requestTeamData.response.body) {
            const filterTeamBasedOnCurrentTeam = (requestTeamData.response.body).filter(
                (singleTeam: any) => ( singleTeam.username == currentLogin ));
            teamInfo = filterTeamBasedOnCurrentTeam[0];
        } else {
            throw new Error("Cannot add user to team");
        }
        // console.log('teamInfo ', teamInfo);
        
        const requestUserData =
        await this.helper.toPromise(
            this.api.userApi, 
            this.api.userApi.usersGetInfoKeyGet, 
            username
            );
        
        // console.log('requestUserData ', requestUserData);
        let userInfo: any;
        if (requestUserData && requestUserData.response) {
            userInfo = requestUserData.response.body;
        } else {
            throw new Error("Cannot add user to team");
        }

        // console.log('userInfo ', userInfo);
        const requestTeamMember =
            await this.helper.toPromise(
                this.api.teamApi, 
                this.api.teamApi.teamsTeamIdUsersGet, 
                teamInfo.teamId
            );

        let teamMember: any[];
        if (requestTeamMember && requestTeamMember.response) {
            teamMember = requestTeamMember.response.body;
        } else {
            throw new Error("Cannot add user to team");
        }

        // console.log('teamMember ', teamMember);
        return {
            teamInfo,
            userInfo,
            teamMember,
            currentLogin
        };
    }

    private checkUser(userId: string, member: any[]): boolean {
        const teamMember = member.map((x: JsonObject) => x.userId);
        if (teamMember.indexOf(userId) > -1) {
            return true;
        }

        return false;
    }
    
}