Mirror/SponsorBlockServer
1
0
Fork 0
mirror of https://github.com/ajayyy/SponsorBlockServer.git synced 2025-12-10 13:37:01 +03:00
Code Issues Packages Projects Releases Wiki Activity

Auto-lock username when admin changes it

Browse Source 
https://github.com/ajayyy/SponsorBlockServer/issues/168
This commit is contained in:
Ajay Ramachandran
2021-06-28 13:20:57 -04:00
parent 07ab48da1f
commit 0ddde452e3
2 changed files with 23 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/routes/setUsername.ts
View File

@@ -55,14 +55,16 @@ export async function setUsername(req: Request, res: Response) {
try { try {
//check if username is already set //check if username is already set
let row = await db.prepare('get', `SELECT count(*) as count FROM "userNames" WHERE "userID" = ?`, [userID]); const row = await db.prepare('get', `SELECT count(*) as count FROM "userNames" WHERE "userID" = ?`, [userID]);
const locked = adminUserIDInput === undefined ? 0 : 1;
if (row.count > 0) { if (row.count > 0) {
//already exists, update this row //already exists, update this row
await db.prepare('run', `UPDATE "userNames" SET "userName" = ? WHERE "userID" = ?`, [userName, userID]); await db.prepare('run', `UPDATE "userNames" SET "userName" = ? WHERE "userID" = ?`, [userName, userID]);
await db.prepare('run', `UPDATE "userNames" SET "locked" = ? WHERE "userID" = ?`, [locked, userID]);
} else { } else {
//add to the db //add to the db
await db.prepare('run', `INSERT INTO "userNames"("userID", "userName") VALUES(?, ?)`, [userID, userName]); await db.prepare('run', `INSERT INTO "userNames"("userID", "userName", "locked") VALUES(?, ?, ?)`, [userID, userName, locked]);
} }
res.sendStatus(200); res.sendStatus(200);

34
test/cases/setUsername.ts
View File

@@ -23,12 +23,12 @@ async function addUsername(userID: string, userName: string, locked = 0) {
await db.prepare('run', 'INSERT INTO "userNames" ("userID", "userName", "locked") VALUES(?, ?, ?)', [userID, userName, locked]); await db.prepare('run', 'INSERT INTO "userNames" ("userID", "userName", "locked") VALUES(?, ?, ?)', [userID, userName, locked]);
} }
async function getUsername(userID: string) { async function getUsernameInfo(userID: string): Promise<{ userName: string, locked: string }> {
const row = await db.prepare('get', 'SELECT "userName" FROM "userNames" WHERE "userID" = ?', [userID]); const row = await db.prepare('get', 'SELECT "userName", "locked" FROM "userNames" WHERE "userID" = ?', [userID]);
if (!row) { if (!row) {
return null; return null;
} }
return row.userName; return row;
} }
describe('setUsername', () => { describe('setUsername', () => {
@@ -95,8 +95,8 @@ describe('setUsername', () => {
.then(async res => { .then(async res => {
if (res.status !== 200) done(`Status code was ${res.status}`); if (res.status !== 200) done(`Status code was ${res.status}`);
else { else {
const userName = await getUsername(getHash(user02PrivateUserID)); const userNameInfo = await getUsernameInfo(getHash(user02PrivateUserID));
if (userName === newUsername) { if (userNameInfo.userName === newUsername) {
done(`Username '${username02}' got changed to '${newUsername}'`); done(`Username '${username02}' got changed to '${newUsername}'`);
} }
else done(); else done();
@@ -111,8 +111,9 @@ describe('setUsername', () => {
method: 'POST', method: 'POST',
}) })
.then(async res => { .then(async res => {
const username = await getUsername(getHash(user03PrivateUserID)); const usernameInfo = await getUsernameInfo(getHash(user03PrivateUserID));
if (username !== newUsername) done(`Username did not change`); if (usernameInfo.userName !== newUsername) done(`Username did not change`);
if (usernameInfo.locked == "1") done(`Username was locked when it shouldn't have been`);
else done(); else done();
}) })
.catch(err => done(`couldn't call endpoint`)); .catch(err => done(`couldn't call endpoint`));
@@ -124,8 +125,9 @@ describe('setUsername', () => {
method: 'POST', method: 'POST',
}) })
.then(async res => { .then(async res => {
const username = await getUsername(getHash(user04PrivateUserID)); const usernameInfo = await getUsernameInfo(getHash(user04PrivateUserID));
if (username === newUsername) done(`Username '${username04}' got changed to '${username}'`); if (usernameInfo.userName === newUsername) done(`Username '${username04}' got changed to '${usernameInfo}'`);
if (usernameInfo.locked == "0") done(`Username was unlocked when it shouldn't have been`);
else done(); else done();
}) })
.catch(err => done(`couldn't call endpoint`)); .catch(err => done(`couldn't call endpoint`));
@@ -137,8 +139,8 @@ describe('setUsername', () => {
method: 'POST', method: 'POST',
}) })
.then(async res => { .then(async res => {
const username = await getUsername(getHash(user05PrivateUserID)); const usernameInfo = await getUsernameInfo(getHash(user05PrivateUserID));
if (username === newUsername) done(`Username contains unicode control characters`); if (usernameInfo.userName === newUsername) done(`Username contains unicode control characters`);
else done(); else done();
}) })
.catch(err => done(`couldn't call endpoint`)); .catch(err => done(`couldn't call endpoint`));
@@ -162,8 +164,9 @@ describe('setUsername', () => {
method: 'POST', method: 'POST',
}) })
.then(async res => { .then(async res => {
const username = await getUsername(getHash(user06PrivateUserID)); const usernameInfo = await getUsernameInfo(getHash(user06PrivateUserID));
if (username !== newUsername) done(`Failed to change username from '${username06}' to '${newUsername}'`); if (usernameInfo.userName !== newUsername) done(`Failed to change username from '${username06}' to '${newUsername}'`);
if (usernameInfo.locked == "0") done(`Username was not locked`);
else done(); else done();
}) })
.catch(err => done(`couldn't call endpoint`)); .catch(err => done(`couldn't call endpoint`));
@@ -175,8 +178,9 @@ describe('setUsername', () => {
method: 'POST', method: 'POST',
}) })
.then(async res => { .then(async res => {
const username = await getUsername(getHash(user06PrivateUserID)); const usernameInfo = await getUsernameInfo(getHash(user06PrivateUserID));
if (username !== newUsername) done(`Failed to change username from '${username06}' to '${newUsername}'`); if (usernameInfo.userName !== newUsername) done(`Failed to change username from '${username06}' to '${newUsername}'`);
if (usernameInfo.locked == "0") done(`Username was unlocked when it shouldn't have been`);
else done(); else done();
}) })
.catch(err => done(`couldn't call endpoint`)); .catch(err => done(`couldn't call endpoint`));