fediglam/src/api/services/invites.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|
                @as(?DateTime, 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(account.id) AS times_used
        \\FROM invite LEFT OUTER JOIN account
        \\  ON invite.id = 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,
    );
}