114 lines
3.2 KiB
TypeScript
114 lines
3.2 KiB
TypeScript
import * as chalk from "chalk";
|
|
|
|
import { StateService } from "@bitwarden/common/abstractions/state.service";
|
|
|
|
import { Response } from "./models/response";
|
|
import { ListResponse } from "./models/response/listResponse";
|
|
import { MessageResponse } from "./models/response/messageResponse";
|
|
import { StringResponse } from "./models/response/stringResponse";
|
|
|
|
export abstract class BaseProgram {
|
|
constructor(
|
|
protected stateService: StateService,
|
|
private writeLn: (s: string, finalLine: boolean, error: boolean) => void
|
|
) {}
|
|
|
|
protected processResponse(
|
|
response: Response,
|
|
exitImmediately = false,
|
|
dataProcessor: () => string = null
|
|
) {
|
|
if (!response.success) {
|
|
if (process.env.BW_QUIET !== "true") {
|
|
if (process.env.BW_RESPONSE === "true") {
|
|
this.writeLn(this.getJson(response), true, false);
|
|
} else {
|
|
this.writeLn(chalk.redBright(response.message), true, true);
|
|
}
|
|
}
|
|
const exitCode = process.env.BW_CLEANEXIT ? 0 : 1;
|
|
if (exitImmediately) {
|
|
process.exit(exitCode);
|
|
} else {
|
|
process.exitCode = exitCode;
|
|
}
|
|
return;
|
|
}
|
|
|
|
if (process.env.BW_RESPONSE === "true") {
|
|
this.writeLn(this.getJson(response), true, false);
|
|
} else if (response.data != null) {
|
|
let out: string = dataProcessor != null ? dataProcessor() : null;
|
|
if (out == null) {
|
|
if (response.data.object === "string") {
|
|
const data = (response.data as StringResponse).data;
|
|
if (data != null) {
|
|
out = data;
|
|
}
|
|
} else if (response.data.object === "list") {
|
|
out = this.getJson((response.data as ListResponse).data);
|
|
} else if (response.data.object === "message") {
|
|
out = this.getMessage(response);
|
|
} else {
|
|
out = this.getJson(response.data);
|
|
}
|
|
}
|
|
|
|
if (out != null && process.env.BW_QUIET !== "true") {
|
|
this.writeLn(out, true, false);
|
|
}
|
|
}
|
|
if (exitImmediately) {
|
|
process.exit(0);
|
|
} else {
|
|
process.exitCode = 0;
|
|
}
|
|
}
|
|
|
|
protected getJson(obj: any): string {
|
|
if (process.env.BW_PRETTY === "true") {
|
|
return JSON.stringify(obj, null, " ");
|
|
} else {
|
|
return JSON.stringify(obj);
|
|
}
|
|
}
|
|
|
|
protected getMessage(response: Response): string {
|
|
const message = response.data as MessageResponse;
|
|
if (process.env.BW_RAW === "true") {
|
|
return message.raw;
|
|
}
|
|
|
|
let out = "";
|
|
if (message.title != null) {
|
|
if (message.noColor) {
|
|
out = message.title;
|
|
} else {
|
|
out = chalk.greenBright(message.title);
|
|
}
|
|
}
|
|
if (message.message != null) {
|
|
if (message.title != null) {
|
|
out += "\n";
|
|
}
|
|
out += message.message;
|
|
}
|
|
return out.trim() === "" ? null : out;
|
|
}
|
|
|
|
protected async exitIfAuthed() {
|
|
const authed = await this.stateService.getIsAuthenticated();
|
|
if (authed) {
|
|
const email = await this.stateService.getEmail();
|
|
this.processResponse(Response.error("You are already logged in as " + email + "."), true);
|
|
}
|
|
}
|
|
|
|
protected async exitIfNotAuthed() {
|
|
const authed = await this.stateService.getIsAuthenticated();
|
|
if (!authed) {
|
|
this.processResponse(Response.error("You are not logged in."), true);
|
|
}
|
|
}
|
|
}
|