UNPKG

uuniq

Version:
105 lines (99 loc) 3.81 kB
// src/Snowflake.class.ts import _ from "lodash"; // src/defaults/SnowflakeOptions.default.ts var SnowflakeOptionsDefault = { epoch: "2025-01-01T00:00:00.000Z", place_id: 0 }; // src/Snowflake.class.ts var parts = { timestamp: 53, place_id: 4, sequence: 10 }; var calculateLimits = (parts2) => { const limits2 = {}; const keys = Object.keys(parts2); for (const key of keys) limits2[key] = Math.pow(2, parts2[key]) - 1; return limits2; }; var calculateShifts = (parts2) => { const shifts2 = {}; let shift = 0; const keys = ["sequence", "place_id", "timestamp"]; for (const key of keys) { shifts2[key] = shift; shift += parts2[key]; } return shifts2; }; var limits = calculateLimits(parts); var shifts = calculateShifts(parts); var Snowflake = class { constructor(options = SnowflakeOptionsDefault) { this.options = _.merge({}, SnowflakeOptionsDefault, options); this.epoch = this.options.epoch instanceof Date ? this.options.epoch.getTime() : typeof this.options.epoch === "string" || typeof this.options.epoch === "number" ? new Date(this.options.epoch).getTime() : (/* @__PURE__ */ new Date("2025-01-01T00:00:00.000Z")).getTime(); if ((this.options.place_id ?? 0) < 0 || (this.options.place_id ?? 0) > limits.place_id) throw new Error(`Field place_id must be between 0 and ${limits.place_id}`); this.options.place_id = (this.options.place_id ?? 0) & limits.place_id; this.sequence = 0; this.last_timestamp = -1; } currentTimestamp() { return Date.now() - this.epoch; } waitForNextTime(last_timestamp) { let timestamp = this.currentTimestamp(); while (last_timestamp >= timestamp) timestamp = this.currentTimestamp(); return timestamp; } generate() { let timestamp = this.currentTimestamp(); if (timestamp < this.last_timestamp) throw new Error("Clock moved backwards."); if (timestamp === this.last_timestamp) { this.sequence = this.sequence + 1 & limits.sequence; if (this.sequence === 0) timestamp = this.waitForNextTime(this.last_timestamp); } else this.sequence = 0; this.last_timestamp = timestamp; return (BigInt(timestamp) << BigInt(shifts.timestamp) | BigInt(this.options.place_id ?? 0) << BigInt(shifts.place_id) | BigInt(this.sequence)).toString(); } resolve(id) { const bigint_id = BigInt(id); return { created_at: new Date(this.epoch + Number(bigint_id >> BigInt(shifts.timestamp) & BigInt(limits.timestamp))).toISOString(), place_id: Number(bigint_id >> BigInt(shifts.place_id) & BigInt(limits.place_id)), sequence: Number(bigint_id & BigInt(limits.sequence)) }; } }; // src/Symbolic.class.ts import Anybase from "any-base"; import _2 from "lodash"; // src/defaults/SymbolicOptions.default.ts var SymbolicOptionsDefault = { epoch: "2025-01-01T00:00:00.000Z", place_id: 0, charset: "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz" }; // src/Symbolic.class.ts var Symbolic = class { constructor(options = SymbolicOptionsDefault) { this.options = _2.merge({}, SymbolicOptionsDefault, options); this.epoch = this.options.epoch instanceof Date ? this.options.epoch.getTime() : typeof this.options.epoch === "string" || typeof this.options.epoch === "number" ? new Date(this.options.epoch).getTime() : (/* @__PURE__ */ new Date("2025-01-01T00:00:00.000Z")).getTime(); this.Snowflake = new Snowflake({ epoch: this.epoch, place_id: this.options.place_id ?? 0 }); this.encode = Anybase(Anybase.DEC, this.options.charset ?? ""); this.decode = Anybase(this.options.charset ?? "", Anybase.DEC); } generate() { return this.encode(this.Snowflake.generate()); } resolve(id) { return this.Snowflake.resolve(this.decode(id)); } }; export { Snowflake, Symbolic };