summary refs log tree commit diff
path: root/src/api/util/utility/RandomInviteID.ts
blob: ec54a9fce4117fae175845178f03e3b87e0cd537 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
/*
	Spacebar: A FOSS re-implementation and extension of the Discord.com backend.
	Copyright (C) 2023 Spacebar and Spacebar Contributors
	
	This program is free software: you can redistribute it and/or modify
	it under the terms of the GNU Affero General Public License as published
	by the Free Software Foundation, either version 3 of the License, or
	(at your option) any later version.
	
	This program is distributed in the hope that it will be useful,
	but WITHOUT ANY WARRANTY; without even the implied warranty of
	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
	GNU Affero General Public License for more details.
	
	You should have received a copy of the GNU Affero General Public License
	along with this program.  If not, see <https://www.gnu.org/licenses/>.
*/

import { Snowflake } from "@spacebar/util";
import crypto from "node:crypto";

// TODO: 'random'? seriously? who named this?
// And why is this even here? Just use cryto.randomBytes?

export function randomString(length = 6) {
    // Declare all characters
    const chars = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789";

    // Pick characers randomly
    let str = "";
    for (let i = 0; i < length; i++) {
        str += chars.charAt(Math.floor(crypto.randomInt(chars.length)));
    }

    return str;
}

export function snowflakeBasedInvite() {
    // Declare all characters
    const chars = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789";
    const base = BigInt(chars.length);
    let snowflake = Snowflake.generateWorkerProcess();

    // snowflakes hold ~10.75 characters worth of entropy;
    // safe to generate a 8-char invite out of them
    const str = "";
    for (let i = 0; i < 10; i++) {
        str.concat(chars.charAt(Number(snowflake % base)));
        snowflake = snowflake / base;
    }

    return str
        .slice(3, 3 + 8)
        .split("")
        .reverse()
        .join("");
}

export function randomUpperString(length: number = 10) {
    const chars = "ABCDEFGHIJKLMNOPQRSTUVWXYZ";

    let result = "";
    for (let i = 0; i < length; i++) {
        result += chars.charAt(Math.floor(Math.random() * chars.length));
    }

    return result;
}