147 lines
3.5 KiB
Zig
147 lines
3.5 KiB
Zig
const std = @import("std");
|
|
const builtin = @import("builtin");
|
|
const util = @import("util");
|
|
|
|
const Uuid = util.Uuid;
|
|
const DateTime = util.DateTime;
|
|
|
|
// 9 random bytes = 12 random b64
|
|
const rand_len = 8;
|
|
const code_len = 12;
|
|
|
|
const Encoder = std.base64.url_safe.Encoder;
|
|
const Decoder = std.base64.url_safe.Decoder;
|
|
|
|
pub const Kind = enum {
|
|
system,
|
|
community_owner,
|
|
user,
|
|
|
|
pub const jsonStringify = util.jsonSerializeEnumAsString;
|
|
};
|
|
|
|
const InviteCount = u16;
|
|
pub const Invite = struct {
|
|
id: Uuid,
|
|
|
|
created_by: Uuid, // User ID
|
|
community_id: ?Uuid,
|
|
name: []const u8,
|
|
code: []const u8,
|
|
|
|
created_at: DateTime,
|
|
times_used: InviteCount,
|
|
|
|
expires_at: ?DateTime,
|
|
max_uses: ?InviteCount,
|
|
|
|
kind: Kind,
|
|
};
|
|
|
|
pub const InviteOptions = struct {
|
|
name: ?[]const u8 = null,
|
|
max_uses: ?InviteCount = null,
|
|
lifespan: ?DateTime.Duration = null,
|
|
kind: Kind = .user,
|
|
};
|
|
|
|
pub fn create(db: anytype, created_by: Uuid, community_id: ?Uuid, options: InviteOptions, alloc: std.mem.Allocator) !Uuid {
|
|
const id = Uuid.randV4(util.getThreadPrng());
|
|
|
|
var code_bytes: [rand_len]u8 = undefined;
|
|
util.getThreadPrng().bytes(&code_bytes);
|
|
|
|
const code = try alloc.alloc(u8, code_len);
|
|
defer alloc.free(code);
|
|
_ = Encoder.encode(code, &code_bytes);
|
|
|
|
const name = options.name orelse code;
|
|
const created_at = DateTime.now();
|
|
|
|
try db.insert(
|
|
"invite",
|
|
.{
|
|
.id = id,
|
|
|
|
.created_by = created_by,
|
|
.community_id = community_id,
|
|
.name = name,
|
|
.code = code,
|
|
|
|
.max_uses = options.max_uses,
|
|
.created_at = created_at,
|
|
.expires_at = if (options.lifespan) |lifespan|
|
|
created_at.add(lifespan)
|
|
else
|
|
null,
|
|
|
|
.kind = options.kind,
|
|
},
|
|
alloc,
|
|
);
|
|
|
|
return id;
|
|
}
|
|
|
|
pub const GetError = error{
|
|
NotFound,
|
|
DatabaseFailure,
|
|
};
|
|
|
|
// Helper fn for getting a single invite
|
|
fn doGetQuery(
|
|
db: anytype,
|
|
comptime where: []const u8,
|
|
query_args: anytype,
|
|
alloc: std.mem.Allocator,
|
|
) GetError!Invite {
|
|
// Generate list of fields from struct
|
|
const field_list = comptime util.comptimeJoinWithPrefix(
|
|
",",
|
|
"invite.",
|
|
&.{
|
|
"id",
|
|
"created_by",
|
|
"community_id",
|
|
"name",
|
|
"code",
|
|
"created_at",
|
|
"expires_at",
|
|
"max_uses",
|
|
"kind",
|
|
},
|
|
);
|
|
|
|
// times_used field is not stored directly in the DB, instead
|
|
// it is calculated based on the number of accounts that were created
|
|
// from it
|
|
const query = std.fmt.comptimePrint(
|
|
\\SELECT {s}, COUNT(local_account.account_id) AS times_used
|
|
\\FROM invite LEFT OUTER JOIN local_account
|
|
\\ ON invite.id = local_account.invite_id
|
|
\\WHERE {s}
|
|
\\GROUP BY invite.id
|
|
\\LIMIT 1
|
|
,
|
|
.{ field_list, where },
|
|
);
|
|
|
|
return db.queryRow(Invite, query, query_args, alloc) catch |err| switch (err) {
|
|
error.NoRows => error.NotFound,
|
|
else => error.DatabaseFailure,
|
|
};
|
|
}
|
|
|
|
pub fn get(db: anytype, id: Uuid, alloc: std.mem.Allocator) GetError!Invite {
|
|
return doGetQuery(db, "invite.id = $1", .{id}, alloc);
|
|
}
|
|
|
|
pub fn getByCode(db: anytype, code: []const u8, community_id: Uuid, alloc: std.mem.Allocator) GetError!Invite {
|
|
return doGetQuery(
|
|
db,
|
|
"invite.code = $1 AND invite.community_id = $2",
|
|
.{ code, community_id },
|
|
alloc,
|
|
);
|
|
}
|