yu-gstatic/src/auth.ts

Cli for Yu Static Site Hosting Service - GStatic

//This file contains code for Authectication mechanism i.e. Signup And Login

import * as prompts from "prompts";
import * as fs from "fs";
import * as chalk from "chalk";
import { apiBackend } from "./fixed";
import { errLogger } from "./utils";
import { homedir } from 'os';
import { join } from "path";
import fetch from "node-fetch";
let accessToken = "" //holds accestoken in memory

export const fetchAuthenticatedHeader = () => {
    return {
        "Authorization": getAccessToken(),
        "content-type": "application/json"
    };
}
export const fetchUnAuthenticateHeader = () => {
    return {
        "content-type": "application/json"

    }
}
export const fetchMultipartAuthenticatedHeader = () => {
    return {
        "Authorization": getAccessToken(),
        //"content-type": "multipart/form-data;boundary=-------".concat(Date.now().toString())
        //Don't need content type header here
    };
}
const tokenFile = join(homedir(), ".access_token");

export function setAccessToken(token: string) {
    //Sets access token

    fs.writeFileSync(tokenFile, token); //Writting token
    accessToken = token;
}

//Retrives accesstoken either from memory or from file
function getAccessToken() {
    if (accessToken.length == 0) {
        try {
            let fc = fs.readFileSync(tokenFile);
            accessToken = fc.toString()

        } catch (err) {
            console.log(chalk.red("Authentication Required !!"))
            process.exit(0); //Exiting app
        }

    }
    return "Bearer ".concat(accessToken);
}



export async function handleLogin() {
    const result = await prompts([
        {
            name: "email",

            type: "text",

            message: "Email ? "
        },
        {
            name: "password",
            validate: password => password.length < 8 ? "Password should be of atleast 8 characters" : true,
            type: "password",
            message: "Password ? "
        }], {
        onCancel: () => { process.exit(0) }
    }
    );

    const { email, password } = result;
    const json = JSON.stringify({ email, password });

    fetch([apiBackend, "login"].join("/"), {
        method: "post",
        body: json,
        headers: fetchUnAuthenticateHeader()
    }).then(res => {
        switch (res.status) {
            case 200: //Everything is okay
                return res.json()
                break;
            case 403://Invalid credentials
                throw Error("Authentiation Failed : Invalid Credentials");
                break;
            case 400:
                throw Error("Invalid Email/Password");
                break;
            default:
                throw Error("Internal Server Error");
                break;
        }
    })
        .then(data => {
            console.log("Welcome Again, ", data.email);
            setAccessToken(data.access_token);
        })
        .catch(errLogger);

}

export async function handleSignup() {

    //Let's signup user
    const result = await prompts([
        {
            name: "email",

            type: "text",
            message: "Email ? "
        },
        {
            name: "password",
            validate: password => password.length < 8 ? "Password should be of atleast 8 characters" : true,
            type: "password",
            message: "Password ? "
        }], {
        onCancel: () => { process.exit(0) }
    }
    );

    const { email, password } = result;
    const json = JSON.stringify({ email, password });
    fetch([apiBackend, "signup"].join("/"), {
        method: "post",
        body: json,
        headers: fetchUnAuthenticateHeader()
    }).then(res => {
        switch (res.status) {
            case 201:
                return res.json(); //Everything is perfect
                break;
            case 400:
                throw Error("Invalid Email/Password");
                break;
            case 403: //User Already exists but credentials are wrong    
                throw Error("Couldn't make your account : Account already exists, may be it doesn't belongs to you");
                break;
            case 200: //Account already exists and belongs to currnet user
                console.log(chalk.yellow("!! It seems you already have an account with this mail, please login"));
                process.exit(0)
                break;
            default:
                throw Error("Internal server error");
        }
    })
        .then(data => {
            console.log(chalk.green("Account Successfully Created!!"))
            setAccessToken(data.access_token);
        })
        .catch(errLogger)

}