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