milliways-infosec/AikidoSec-safe-chain
7
0
Fork 0
mirror of https://github.com/AikidoSec/safe-chain.git synced 2026-05-26 12:10:49 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Check if directory exists before creating a new shell startup file

Browse source
This commit is contained in:
Sander Declerck 2025-09-17 15:26:06 +02:00
parent 846c62e4e0
commit f793bb8467
No known key found for this signature in database
2 changed files with 33 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

27
packages/safe-chain/src/shell-integration/helpers.js
View file

@ -1,6 +1,7 @@
import { spawnSync } from "child_process";
import * as os from "os";
import fs from "fs";
import path from "path";
export const knownAikidoTools = [
{ tool: "npm", aikidoCommand: "aikido-npm" },
@ -48,7 +49,12 @@ function shouldRemoveLine(line, pattern) {
return false;
}
if (line.includes("\n") || line.includes("\r") || line.includes("\u2028") || line.includes("\u2029")) {
if (
line.includes("\n") ||
line.includes("\r") ||
line.includes("\u2028") ||
line.includes("\u2029")
) {
// If the line contains newlines, something has gone wrong in splitting
// \u2028 and \u2029 are Unicode line separator characters (line and paragraph separators)
return false;
@ -58,11 +64,22 @@ function shouldRemoveLine(line, pattern) {
}
export function addLineToFile(filePath, line) {
if (!fs.existsSync(filePath)) {
fs.writeFileSync(filePath, "", "utf-8");
}
createFileIfNotExists(filePath);
const fileContent = fs.readFileSync(filePath, "utf-8");
const updatedContent = fileContent + os.EOL + line;
const updatedContent = fileContent + os.EOL + line + os.EOL;
fs.writeFileSync(filePath, updatedContent, "utf-8");
}
function createFileIfNotExists(filePath) {
if (fs.existsSync(filePath)) {
return;
}
const dir = path.dirname(filePath);
if (!fs.existsSync(dir)) {
fs.mkdirSync(dir, { recursive: true });
}
fs.writeFileSync(filePath, "", "utf-8");
}

11
packages/safe-chain/src/shell-integration/setup.js
View file

@ -56,11 +56,13 @@ export async function setup() {
*/
function setupShell(shell) {
let success = false;
let error;
try {
shell.teardown(knownAikidoTools); // First, tear down to prevent duplicate aliases
success = shell.setup(knownAikidoTools);
} catch {
} catch (err) {
success = false;
error = err;
}
if (success) {
@ -75,6 +77,13 @@ function setupShell(shell) {
"Setup failed"
)}. Please check your ${shell.name} configuration.`
);
if (error) {
let message = ` Error: ${error.message}`;
if (error.code) {
message += ` (code: ${error.code})`;
}
ui.writeError(message);
}
}
return success;