iframe-proxy | Sunbelt Computer Software

189 lines (171 loc) · 6.85 KB
'use strict';
import fs from "fs";
import vscode from 'vscode';
import yaml from 'js-yaml';
import detectPort from "detect-port";
import {readYML} from './common';
import {showConnectRepo} from "./notifications";
import {createUserWithApi} from "./api_utils";
import {generateSettings} from "../settings";
import {contextVariables, ECONNREFUSED, getRepoInSyncMsg, HttpStatusCodes, NOTIFICATION, NOTIFICATION_BUTTON} from "../constants";
import { CodeSyncLogger } from "../logger";
import { pathUtils } from "./path_utils";
import { RepoState } from "./repo_state_utils";
import { UserState } from "./user_utils";
import { authHandler, reactivateAccountHandler } from "../handlers/user_commands";
import { createSystemDirectories, setupCodeSync } from "./setup_utils";
export const isPortAvailable = async (port: number) => {
    return detectPort(port)
        .then(_port => {
            return port === _port;
        .catch(err => {
            console.log(err);
            return false;
        });
const postSuccessLoginAddUser = (userEmail: string, accessToken: string) => {
    if (!userEmail) return;
    const settings = generateSettings();
    // Save access token of user against email in user.yml
    const users = readYML(settings.USER_PATH) || {};
    if (userEmail in users) {
        users[userEmail].access_token = accessToken;
        users[userEmail].is_active = true;
    } else {
        users[userEmail] = {
            access_token: accessToken,
            is_active: true
    fs.writeFileSync(settings.USER_PATH, yaml.dump(users));
export const postSuccessLogin = (userEmail: string, accessToken: string) => {
    postSuccessLoginAddUser(userEmail, accessToken);
    vscode.commands.executeCommand('setContext', contextVariables.showLogIn, false);
    vscode.commands.executeCommand('setContext', contextVariables.showReactivateAccount, false);
    const userState = new UserState();
    userState.setValidAccount();
    const repoPath = pathUtils.getRootPath() || "";
    if (!repoPath) return;
    const repoState = new RepoState(repoPath).get();
    const repoIsNotConnected = !repoState.IS_CONNECTED;
    vscode.commands.executeCommand('setContext', contextVariables.showConnectRepoView, repoIsNotConnected);
	if (repoIsNotConnected) {
        // Show notification to user to Sync the repo
        return showConnectRepo(repoPath, userEmail, accessToken);
    // Show notification that repo is in sync
    vscode.window.showInformationMessage(getRepoInSyncMsg(repoPath));
const postDeactivatedAccount = (userEmail: string, accessToken: string) => {
    postSuccessLoginAddUser(userEmail, accessToken);
    vscode.commands.executeCommand('setContext', contextVariables.showLogIn, false);
    vscode.commands.executeCommand('setContext', contextVariables.showReactivateAccount, true);
    vscode.commands.executeCommand('setContext', contextVariables.showConnectRepoView, false);
    const userState = new UserState();
    userState.setDeactivated();
const checkDeactivatedAccount = (email: string, accessToken: string, statusCode: number) => {
    if (statusCode !== HttpStatusCodes.USER_ACCOUNT_DEACTIVATED) return false;
    postDeactivatedAccount(email, accessToken);
    vscode.window.showErrorMessage(NOTIFICATION.ACCOUNT_DEACTIVATED, NOTIFICATION_BUTTON.REACTIVATE_ACCOUNT).then(selection => {
        if (!selection) return;
        if (selection === NOTIFICATION_BUTTON.REACTIVATE_ACCOUNT) {
            reactivateAccountHandler();
    return true;
export const isAccountActive = async (email: string, accessToken: string) => {
    const isValidAccount = true;
    const userState = new UserState();
    userState.setValidAccount();
    const userResponse = await createUserWithApi(accessToken);
    if (!userResponse.error) return isValidAccount;
    if (userResponse.statusCode === HttpStatusCodes.UNAUTHORIZED) {
        userState.setInvalidAccount();
        askAndTriggerSignUp();
        return false;
    const isDeactivated = checkDeactivatedAccount(email, accessToken, userResponse.statusCode);
	userState.setDeactivated(isDeactivated);
    if (isDeactivated) return false;
    return !userResponse.error.toString().includes(ECONNREFUSED);
export const createUser = async (accessToken: string, idToken: string) => {
    const auth0User = parseJwt(idToken);
    const userResponse = await createUserWithApi(accessToken);
    if (userResponse.error) {
        const isDeactivated = checkDeactivatedAccount(auth0User.email, accessToken, userResponse.statusCode);
        if (isDeactivated) return {
                success: true,
                isDeactivated: true
        vscode.window.showErrorMessage(NOTIFICATION.SIGNUP_FAILED);
        CodeSyncLogger.critical("Error creating user from API", userResponse.error);    
        return {
            success: false,
            isDeactivated: false
    const userEmail = userResponse.email;
    postSuccessLogin(userEmail, accessToken);
    return {
        success: true,
        isDeactivated: false
export const markUsersInactive = (notify=true) => {
    vscode.commands.executeCommand('setContext', contextVariables.showLogIn, true);
    // Mark all users as is_active=false in user.yml
    const settings = generateSettings();
    const users = readYML(settings.USER_PATH);
    Object.keys(users).forEach((email) => {
        users[email].is_active = false;
    fs.writeFileSync(settings.USER_PATH, yaml.dump(users));
    const userState = new UserState();
	userState.setInvalidAccount();
    if (!notify) return;
    vscode.window.showInformationMessage(NOTIFICATION.LOGGED_OUT_SUCCESSFULLY);
export const resetUserSession = async () => {
    const settings = generateSettings();
        fs.renameSync(settings.CODESYNC_ROOT, settings.BACKUP_ROOT);
    } catch (e) {
        // eslint-disable-next-line @typescript-eslint/ban-ts-comment
        // @ts-ignore
        CodeSyncLogger.error("Failed to reset root repo", e.stack);
    const repoPath = pathUtils.getRootPath();
    await setupCodeSync(repoPath);
export const askAndTriggerSignUp = () => {
    // Trigger sign up process
    vscode.window.showErrorMessage(
        NOTIFICATION.AUTHENTICATION_FAILED, ...[
        NOTIFICATION.LOGIN,
        NOTIFICATION.IGNORE
    ]).then(selection => {
        if (selection === NOTIFICATION.LOGIN) {
            authHandler(true);
const parseJwt = (token: string) => {
    return JSON.parse(Buffer.from(token.split('.')[1], 'base64').toString());
export const postSuccessLogout = async () => {
    markUsersInactive();
    // Rename system directory
    await resetUserSession();
    if ((global as any).socketConnection) {
        (global as any).socketConnection.close();
        (global as any).socketConnection = null;
    if ((global as any).websocketClient) {
        (global as any).websocketClient = null;
Sunbelt Computer Software

PL/B Language Development and Support

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

FilesExpand file tree

auth_utils.ts

Latest commit

History

auth_utils.ts

File metadata and controls
