First Commit

This commit is contained in:
Benjamin Toby
2024-11-05 12:12:42 +01:00
commit 09ae0dd3fe
1078 changed files with 138432 additions and 0 deletions
+177
View File
@@ -0,0 +1,177 @@
// @ts-check
////////////////////////////////////////
////////////////////////////////////////
////////////////////////////////////////
// require("dotenv").config({ path: "../../.env" });
const noDatabaseDbHandler = require("../utils/noDatabaseDbHandler");
const dbHandler = require("../utils/dbHandler");
/** ****************************************************************************** */
/** ****************************************************************************** */
/** ****************************************************************************** */
/** ****************************************************************************** */
/** ****************************************************************************** */
/** ****************************************************************************** */
/**
* @typedef {object} GrantType
* @property {string} database - Database Name
* @property {string} table - Table Name
* @property {string[]} privileges - Privileges
*/
// const mariadbUserFlagIndex = process.argv.findIndex((arg) => arg == "--user");
// const mariadbUser =
// mariadbUserFlagIndex > 0 ? process.argv[mariadbUserFlagIndex + 1] : null;
// //////////////////////////////////////////////
// //////////////////////////////////////////////
// const mariadbUserHostFlagIndex = process.argv.findIndex(
// (arg) => arg == "--host"
// );
// const mariadbUserHost =
// mariadbUserHostFlagIndex > 0
// ? process.argv[mariadbUserHostFlagIndex + 1]
// : null;
// //////////////////////////////////////////////
// //////////////////////////////////////////////
// const newGrantsJSONFlagIndex = process.argv.findIndex(
// (arg) => arg == "--grants"
// );
// const newGrantsJSON =
// newGrantsJSONFlagIndex > 0
// ? process.argv[newGrantsJSONFlagIndex + 1]
// : null;
// //////////////////////////////////////////////
// //////////////////////////////////////////////
// if (!mariadbUser) {
// console.log(`No user provided. \x1b[35m\`--user\`\x1b[0m flag is required`);
// process.exit();
// }
// if (!mariadbUserHost) {
// console.log(`No Host provided. \x1b[35m\`--host\`\x1b[0m flag is required`);
// process.exit();
// }
// if (!newGrantsJSON) {
// console.log(
// `No grants JSON provided. \x1b[35m\`--grants\`\x1b[0m flag is required, and it must be a JSON string`
// );
// process.exit();
// }
//////////////////////////////////////////////
//////////////////////////////////////////////
/**
* Handle Grants for Users
* ==============================================================================
* @param {object} params - Single object params
* @param {string} params.username - Username
* @param {string} params.host - Host
* @param {GrantType[]} params.grants - Grants
*
* @returns {Promise<boolean>} success
*/
async function handleGrants({ username, host, grants }) {
let success = false;
console.log(`Handling Grants for User =>`, username, host);
if (!username) {
console.log(`No username provided.`);
return success;
}
if (!host) {
console.log(
`No Host provided. \x1b[35m\`--host\`\x1b[0m flag is required`
);
return success;
}
if (!grants) {
console.log(`No grants Array provided.`);
return success;
}
try {
const existingUser = await noDatabaseDbHandler(
`SELECT * FROM mysql.user WHERE User = '${username}' AND Host = '${host}'`
);
const isUserExisting = Boolean(existingUser?.[0]?.User);
if (isUserExisting) {
const userGrants = await noDatabaseDbHandler(
`SHOW GRANTS FOR '${username}'@'${host}'`
);
for (let i = 0; i < userGrants.length; i++) {
const grantObject = userGrants[i];
const grant = grantObject?.[Object.keys(grantObject)[0]];
if (grant?.match(/GRANT .* PRIVILEGES ON .* TO/)) {
const revokeGrantText = grant
.replace(/GRANT/, "REVOKE")
.replace(/ TO /, " FROM ");
const revokePrivilege = await noDatabaseDbHandler(
revokeGrantText
);
}
}
/**
* @type {GrantType[]}
*/
const grantsArray = grants;
for (let i = 0; i < grantsArray.length; i++) {
const grantObject = grantsArray[i];
const { database, table, privileges } = grantObject;
const tableText = table == "*" ? "*" : `\`${table}\``;
const databaseText = database == "*" ? "*" : `\`${database}\``;
const privilegesText = privileges.includes("ALL")
? "ALL PRIVILEGES"
: privileges.join(", ");
const grantText = `GRANT ${privilegesText} ON ${databaseText}.${tableText} TO '${username}'@'${host}'`;
const grantPriviledge = await noDatabaseDbHandler(grantText);
}
}
success = true;
//////////////////////////////////////////////
//////////////////////////////////////////////
//////////////////////////////////////////////
} catch (/** @type {any} */ error) {
console.log(`Error in adding SQL user =>`, error.message);
}
return success;
////////////////////////////////////////
////////////////////////////////////////
////////////////////////////////////////
}
/** ****************************************************************************** */
/** ****************************************************************************** */
/** ****************************************************************************** */
/** ****************************************************************************** */
/** ****************************************************************************** */
module.exports = handleGrants;
+293
View File
@@ -0,0 +1,293 @@
// @ts-check
////////////////////////////////////////
////////////////////////////////////////
////////////////////////////////////////
require("dotenv").config({ path: "../../.env" });
const generator = require("generate-password");
const noDatabaseDbHandler = require("../utils/noDatabaseDbHandler");
const dbHandler = require("../utils/dbHandler");
const encrypt = require("../../package-shared/functions/backend/encrypt");
const decrypt = require("../../package-shared/functions/backend/decrypt");
const { execSync } = require("child_process");
const handleGrants = require("./handleGrants");
/** ****************************************************************************** */
/** ****************************************************************************** */
/** ****************************************************************************** */
/** ****************************************************************************** */
/** ****************************************************************************** */
/** ****************************************************************************** */
const userIdFlagIndex = process.argv.findIndex((arg) => arg == "--userId");
const userId = userIdFlagIndex > 0 ? process.argv[userIdFlagIndex + 1] : null;
//////////////////////////////////////////////
//////////////////////////////////////////////
const mariadbUserHostFlagIndex = process.argv.findIndex(
(arg) => arg == "--host"
);
const mariadbUserHostArg =
mariadbUserHostFlagIndex > 0
? process.argv[mariadbUserHostFlagIndex + 1]
: null;
//////////////////////////////////////////////
//////////////////////////////////////////////
const mariadbUserFlagIndex = process.argv.findIndex(
(arg) => arg == "--username"
);
const mariadbUserArg =
mariadbUserFlagIndex > 0 ? process.argv[mariadbUserFlagIndex + 1] : null;
const sqlUserIDFlagIndex = process.argv.findIndex(
(arg) => arg == "--sql-user-id"
);
const sqlUserIDArg =
sqlUserIDFlagIndex > 0 ? process.argv[sqlUserIDFlagIndex + 1] : null;
//////////////////////////////////////////////
//////////////////////////////////////////////
const defaultMariadbUserHost = process.env.DSQL_DB_HOST || "127.0.0.1";
/**
* Create database from Schema Function
* ==============================================================================
* @param {object} params - Single object params
* @param {number|string|null} params.userId - User ID or null
*/
async function refreshUsersAndGrants() {
/**
* @description Users
* @type {*[] | null}
*/ // @ts-ignore
const users = await dbHandler({
query: `SELECT * FROM users`,
});
if (!users?.[0]) {
process.exit();
}
for (let i = 0; i < users.length; i++) {
const user = users[i];
if (!user) continue;
if (userId && user.id != userId) continue;
try {
const { mariadb_user, mariadb_host, mariadb_pass, id } = user;
const existingUser = await noDatabaseDbHandler(
`SELECT * FROM mysql.user WHERE User = '${mariadb_user}' AND Host = '${mariadb_host}'`
);
const existingMariaDBUserArray =
userId && sqlUserIDArg
? await dbHandler({
query: `SELECT * FROM mariadb_users WHERE id = ? AND user_id = ?`,
values: [sqlUserIDArg, userId],
})
: null;
/**
* @type {import("../../package-shared/types").MYSQL_mariadb_users_table_def | undefined}
*/
const activeMariadbUserObject = Array.isArray(
existingMariaDBUserArray
)
? existingMariaDBUserArray?.[0]
: undefined;
const isPrimary = activeMariadbUserObject
? String(activeMariadbUserObject.primary)?.match(/1/)
? true
: false
: false;
const isUserExisting = Boolean(existingUser?.[0]?.User);
const isThisPrimaryHost = Boolean(
mariadbUserHostArg == defaultMariadbUserHost
);
const dslUsername = `dsql_user_${id}`;
const dsqlPassword = activeMariadbUserObject?.password
? activeMariadbUserObject.password
: isUserExisting
? mariadb_pass
: generator.generate({
length: 16,
numbers: true,
symbols: true,
uppercase: true,
exclude: "*#.'`\"",
});
const encryptedPassword = activeMariadbUserObject?.password
? activeMariadbUserObject.password
: isUserExisting
? mariadb_pass
: encrypt(dsqlPassword);
if (
!isUserExisting &&
!sqlUserIDArg &&
!isPrimary &&
!mariadbUserHostArg &&
!mariadbUserArg
) {
await noDatabaseDbHandler(
`CREATE USER IF NOT EXISTS '${dslUsername}'@'${defaultMariadbUserHost}' IDENTIFIED BY '${dsqlPassword}' REQUIRE SSL`
);
console.log(
`User ${user.id}: ${user.first_name} ${user.last_name} SQL credentials successfully updated.`
);
const updateUser = await dbHandler({
query: `UPDATE users SET mariadb_user = ?, mariadb_host = ?, mariadb_pass = ? WHERE id = ?`,
values: [
dslUsername,
defaultMariadbUserHost,
encryptedPassword,
user.id,
],
});
}
if (isPrimary) {
const finalHost = mariadbUserHostArg
? mariadbUserHostArg
: mariadb_host;
const updateUser = await dbHandler({
query: `UPDATE users SET mariadb_user = ?, mariadb_host = ?, mariadb_pass = ? WHERE id = ?`,
values: [
dslUsername,
finalHost,
encryptedPassword,
user.id,
],
});
}
//////////////////////////////////////////////
//////////////////////////////////////////////
//////////////////////////////////////////////
/**
* @description Handle mariadb_users table
*/
const existingMariadbPrimaryUser = await dbHandler({
query: `SELECT * FROM mariadb_users WHERE user_id = ? AND \`primary\` = 1`,
values: [id],
});
const isPrimaryUserExisting = Boolean(
Array.isArray(existingMariadbPrimaryUser) &&
existingMariadbPrimaryUser?.[0]?.user_id
);
/** @type {import("./handleGrants").GrantType[]} */
const primaryUserGrants = [
{
database: "*",
table: "*",
privileges: ["ALL"],
},
];
if (!isPrimaryUserExisting) {
const insertPrimaryMariadbUser = await dbHandler({
query: `INSERT INTO mariadb_users (user_id, username, password, \`primary\`, grants) VALUES (?, ?, ?, ?, ?)`,
values: [
id,
dslUsername,
encryptedPassword,
"1",
JSON.stringify(primaryUserGrants),
],
});
}
//////////////////////////////////////////////
const existingExtraMariadbUsers = await dbHandler({
query: `SELECT * FROM mariadb_users WHERE user_id = ? AND \`primary\` != '1'`,
values: [id],
});
if (Array.isArray(existingExtraMariadbUsers)) {
for (let i = 0; i < existingExtraMariadbUsers.length; i++) {
const mariadbUser = existingExtraMariadbUsers[i];
const {
user_id,
username,
host,
password,
primary,
grants,
} = mariadbUser;
if (mariadbUserArg && username != mariadbUserArg) continue;
if (mariadbUserHostArg && host != mariadbUserHostArg)
continue;
const decrptedPassword = decrypt(password);
const existingExtraMariadbUser = await noDatabaseDbHandler(
`SELECT * FROM mysql.user WHERE User = '${username}' AND Host = '${host}'`
);
const isExtraMariadbUserExisting = Boolean(
existingExtraMariadbUser?.[0]?.User
);
if (!isExtraMariadbUserExisting) {
await noDatabaseDbHandler(
`CREATE USER IF NOT EXISTS '${username}'@'${host}' IDENTIFIED BY '${decrptedPassword}' REQUIRE SSL`
);
}
const isGrantHandled = await handleGrants({
username,
host,
grants:
grants && typeof grants == "string"
? JSON.parse(grants)
: [],
});
if (!isGrantHandled) {
console.log(
`Error in handling grants for user ${username}@${host}`
);
}
}
}
//////////////////////////////////////////////
//////////////////////////////////////////////
//////////////////////////////////////////////
} catch (/** @type {any} */ error) {
console.log(`Error in adding SQL user =>`, error.message);
}
}
process.exit();
////////////////////////////////////////
////////////////////////////////////////
////////////////////////////////////////
}
/** ****************************************************************************** */
/** ****************************************************************************** */
/** ****************************************************************************** */
/** ****************************************************************************** */
/** ****************************************************************************** */
refreshUsersAndGrants();
+100
View File
@@ -0,0 +1,100 @@
// @ts-check
////////////////////////////////////////
////////////////////////////////////////
////////////////////////////////////////
require("dotenv").config({ path: "../../.env" });
const generator = require("generate-password");
const noDatabaseDbHandler = require("../utils/noDatabaseDbHandler");
const dbHandler = require("../utils/dbHandler");
const encrypt = require("../../package-shared/functions/backend/encrypt");
/** ****************************************************************************** */
/** ****************************************************************************** */
/** ****************************************************************************** */
/** ****************************************************************************** */
/** ****************************************************************************** */
/** ****************************************************************************** */
/**
* Create database from Schema Function
* ==============================================================================
* @param {object} params - Single object params
* @param {number|string|null} params.userId - User ID or null
*/
async function resetSQLCredentialsPasswords() {
/**
* @description Users
* @type {*[] | null}
*/ // @ts-ignore
const users = await dbHandler({
query: `SELECT * FROM users`,
});
if (!users) {
process.exit();
}
for (let i = 0; i < users.length; i++) {
const user = users[i];
if (!user) continue;
try {
/**
* @type {any[]}
*/ // @ts-ignore
const maridbUsers = await dbHandler({
query: `SELECT * FROM mysql.user WHERE User = 'dsql_user_${user.id}'`,
});
for (let j = 0; j < maridbUsers.length; j++) {
const { User, Host } = maridbUsers[j];
const password = generator.generate({
length: 16,
numbers: true,
symbols: true,
uppercase: true,
exclude: "*#.'`\"",
});
const encryptedPassword = encrypt(password);
await noDatabaseDbHandler(
`SET PASSWORD FOR '${User}'@'${Host}' = PASSWORD('${password}')`
);
if (user.mariadb_user == User && user.mariadb_host == Host) {
const updateUser = await dbHandler({
query: `UPDATE users SET mariadb_pass = ? WHERE id = ?`,
values: [encryptedPassword, user.id],
});
}
console.log(
`User ${user.id}: ${user.first_name} ${user.last_name} Password Updated successfully added.`
);
}
} catch (/** @type {any} */ error) {
console.log(
`Error Updating User ${user.id} Password =>`,
error.message
);
}
}
process.exit();
////////////////////////////////////////
////////////////////////////////////////
////////////////////////////////////////
}
/** ****************************************************************************** */
/** ****************************************************************************** */
/** ****************************************************************************** */
/** ****************************************************************************** */
/** ****************************************************************************** */
resetSQLCredentialsPasswords();
+166
View File
@@ -0,0 +1,166 @@
// @ts-check
const fs = require("fs");
require("dotenv").config({ path: "../../../.env" });
const hashPassword = require("datasquirel/functions/hashPassword");
const path = require("path");
const { execSync } = require("child_process");
const encrypt = require("../../../package-shared/functions/backend/encrypt");
const addMariadbUser = require("../../../package-shared/functions/backend/addMariadbUser");
const DB_HANDLER = require("../../../package-shared/utils/backend/global-db/DB_HANDLER");
const addDbEntry = require("../../../package-shared/functions/backend/db/addDbEntry");
const updateDbEntry = require("../../../package-shared/functions/backend/db/updateDbEntry");
/**
* # Create New User
*/
async function createUser() {
/**
* Validate Form
*
* @description Check if request body is valid
*/
try {
const userObj = JSON.parse(
fs.readFileSync(path.resolve(__dirname, "./new-user.json"), "utf-8")
);
const ROOT_DIR = path.resolve(__dirname, "../../../");
/**
* Validate Form
*
* @description Check if request body is valid
*/
const first_name = userObj.first_name;
const last_name = userObj.last_name;
const email = userObj.email;
const password = userObj.password;
const username = userObj.username;
const id = userObj.id || undefined;
if (!email?.match(/.*@.*\..*/)) return false;
if (
!first_name?.match(/^[a-zA-Z]+$/) ||
!last_name?.match(/^[a-zA-Z]+$/)
)
return false;
if (password?.match(/ /)) return false;
if (username?.match(/ /)) return false;
let hashedPassword = hashPassword({
encryptionKey: process.env.DSQL_ENCRYPTION_PASSWORD || "",
password: password,
});
let existingUser = await DB_HANDLER(
`SELECT * FROM users WHERE email='${email}'`
);
if (existingUser?.[0]) {
console.log("User Exists");
return false;
}
const newUser = await addDbEntry({
dbFullName: "datasquirel",
tableName: "users",
data: { ...userObj, password: hashedPassword },
});
if (!newUser?.insertId) return false;
/**
* Add a Mariadb User for this User
*/
await addMariadbUser({ userId: newUser.insertId });
const STATIC_ROOT = process.env.DSQL_STATIC_SERVER_DIR;
if (!STATIC_ROOT) {
console.log("Static File ENV not Found!");
throw new Error("No Static Path");
}
/**
* Create new user folder and file
*
* @description Create new user folder and file
*/
let newUserSchemaFolderPath = `./jsonData/dbSchemas/users/user-${newUser.insertId}`;
let newUserMediaFolderPath = path.join(
STATIC_ROOT,
`images/user-images/user-${newUser.insertId}`
);
fs.mkdirSync(newUserSchemaFolderPath, { recursive: true });
fs.mkdirSync(newUserMediaFolderPath, { recursive: true });
fs.writeFileSync(
`${newUserSchemaFolderPath}/main.json`,
JSON.stringify([]),
"utf8"
);
let imagePath = path.join(
STATIC_ROOT,
`images/user-images/user-${newUser.insertId}/user-${newUser.insertId}-profile.jpg`
);
let imageThumbnailPath = path.join(
STATIC_ROOT,
`images/user-images/user-${newUser.insertId}/user-${newUser.insertId}-profile-thumbnail.jpg`
);
let prodImageUrl = imagePath.replace(
STATIC_ROOT,
process.env.DSQL_STATIC_HOST || ""
);
let prodImageThumbnailUrl = imageThumbnailPath.replace(
STATIC_ROOT,
process.env.DSQL_STATIC_HOST || ""
);
fs.copyFileSync(
path.join(ROOT_DIR, "/public/images/user_images/user-preset.png"),
imagePath
);
fs.copyFileSync(
path.join(
ROOT_DIR,
"/public/images/user_images/user-preset-thumbnail.png"
),
imageThumbnailPath
);
execSync(`chmod 644 ${imagePath} ${imageThumbnailPath}`);
const updateImages = await updateDbEntry({
dbFullName: "datasquirel",
tableName: "users",
identifierColumnName: "id",
identifierValue: newUser.insertId,
data: {
image: prodImageUrl,
image_thumbnail: prodImageThumbnailUrl,
},
});
return true;
} catch (/** @type {any} */ error) {
console.log(`Error in creating user => ${error.message}`);
return false;
}
}
createUser().then((res) => {
if (res) {
console.log("User Creation Success!!!");
} else {
console.log("User Creation Failed!");
}
process.exit();
});
+65
View File
@@ -0,0 +1,65 @@
// @ts-check
const fs = require("fs");
require("dotenv").config({ path: "../../../.env" });
const hashPassword = require("datasquirel/functions/hashPassword");
const path = require("path");
const { execSync } = require("child_process");
const encrypt = require("../../../package-shared/functions/backend/encrypt");
const addMariadbUser = require("../../../package-shared/functions/backend/addMariadbUser");
const DB_HANDLER = require("../../../package-shared/utils/backend/global-db/DB_HANDLER");
const addDbEntry = require("../../../package-shared/functions/backend/db/addDbEntry");
const updateDbEntry = require("../../../package-shared/functions/backend/db/updateDbEntry");
/**
* # Create New User
*/
async function createUser() {
/**
* Validate Form
*
* @description Check if request body is valid
*/
try {
const updateUserObj = JSON.parse(
fs.readFileSync(path.resolve(__dirname, "./new-user.json"), "utf-8")
);
let hashedPassword = updateUserObj.password
? hashPassword({
encryptionKey: process.env.DSQL_ENCRYPTION_PASSWORD || "",
password: updateUserObj.password,
})
: undefined;
let updatePayload = { ...updateUserObj };
if (hashedPassword) {
updatePayload["password"] = hashedPassword;
}
/** @type {any} */
const newUser = await updateDbEntry({
dbFullName: "datasquirel",
tableName: "users",
data: { ...updatePayload, id: undefined },
identifierColumnName: "id",
identifierValue: updatePayload.id,
});
if (!newUser?.affectedRows) return false;
return true;
} catch (/** @type {any} */ error) {
console.log(`Error in creating user => ${error.message}`);
return false;
}
}
createUser().then((res) => {
if (res) {
console.log("User Update Success!!!");
} else {
console.log("User Update Failed!");
}
process.exit();
});