Merge pull request #519 from erkinalp/master
Snowflake-based invite generation
2 files changed, 26 insertions, 4 deletions
diff --git a/api/src/util/RandomInviteID.ts b/api/src/util/RandomInviteID.ts
index b37994d8..45450b0d 100644
--- a/api/src/util/RandomInviteID.ts
+++ b/api/src/util/RandomInviteID.ts
@@ -1,3 +1,5 @@
+import { Snowflake } from "@fosscord/util";
+
export function random(length = 6) {
// Declare all characters
let chars = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789";
@@ -10,3 +12,20 @@ export function random(length = 6) {
return str;
}
+
+export function snowflakeBasedInvite() {
+ // Declare all characters
+ let chars = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789";
+
+ let snowflake = Snowflake.generateWorkerProcess();
+
+ // snowflakes hold ~10.75 characters worth of entropy;
+ // safe to generate a 8-char invite out of them
+ let str = "";
+ for (let i=0; i < 10; i++) {
+ str += chars.charAt((snowflake % chars.length));
+ snowflake /= chars.length;
+ }
+
+ return str.substr(3,8).reverse(); // little-endianise for entropy
+}
diff --git a/util/src/util/Snowflake.ts b/util/src/util/Snowflake.ts
index 3f6e3c63..134d526e 100644
--- a/util/src/util/Snowflake.ts
+++ b/util/src/util/Snowflake.ts
@@ -21,7 +21,7 @@ export class Snowflake {
}
/**
- * A Twitter snowflake, except the epoch is 2015-01-01T00:00:00.000Z
+ * A Twitter-like snowflake, except the epoch is 2015-01-01T00:00:00.000Z
* ```
* If we have a snowflake '266241948824764416' we can represent it as binary:
*
@@ -83,14 +83,17 @@ export class Snowflake {
return dec;
}
- static generate() {
+ static generateWorkerProcess() { // worker process - returns a number
var time = BigInt(Date.now() - Snowflake.EPOCH) << BigInt(22);
var worker = Snowflake.workerId << 17n;
var process = Snowflake.processId << 12n;
var increment = Snowflake.INCREMENT++;
- return (time | worker | process | increment).toString();
+ return BigInt(time | worker | process | increment);
+ }
+
+ static generate() {
+ return Snowflake.generateWorkerProcess().toString();
}
-
/**
* A deconstructed snowflake.
* @typedef {Object} DeconstructedSnowflake
|
