using Microsoft.EntityFrameworkCore.Migrations; using NodaTime; #nullable disable namespace Foxnouns.Backend.Database.Migrations { /// public partial class AddSids : Migration { /// protected override void Up(MigrationBuilder migrationBuilder) { migrationBuilder.AddColumn( name: "sid", table: "users", type: "text", nullable: true); migrationBuilder.AddColumn( name: "last_sid_reroll", table: "users", type: "timestamp with time zone", nullable: false, defaultValueSql: "now() - '1 hour'::interval"); migrationBuilder.AddColumn( name: "sid", table: "members", type: "text", nullable: true); migrationBuilder.CreateIndex( name: "ix_users_sid", table: "users", column: "sid", unique: true); migrationBuilder.CreateIndex( name: "ix_members_sid", table: "members", column: "sid", unique: true); migrationBuilder.Sql(@"create function generate_sid(len int) returns text as $$ select string_agg(substr('abcdefghijklmnopqrstuvwxyz', ceil(random() * 26)::integer, 1), '') from generate_series(1, len) $$ language sql volatile; "); migrationBuilder.Sql(@"create function find_free_user_sid() returns text as $$ declare new_sid text; begin loop new_sid := generate_sid(5); if not exists (select 1 from users where sid = new_sid) then return new_sid; end if; end loop; end $$ language plpgsql volatile; "); migrationBuilder.Sql(@"create function find_free_member_sid() returns text as $$ declare new_sid text; begin loop new_sid := generate_sid(6); if not exists (select 1 from members where sid = new_sid) then return new_sid; end if; end loop; end $$ language plpgsql volatile;"); } /// protected override void Down(MigrationBuilder migrationBuilder) { migrationBuilder.Sql("drop function find_free_member_sid;"); migrationBuilder.Sql("drop function find_free_user_sid;"); migrationBuilder.Sql("drop function generate_sid;"); migrationBuilder.DropIndex( name: "ix_users_sid", table: "users"); migrationBuilder.DropIndex( name: "ix_members_sid", table: "members"); migrationBuilder.DropColumn( name: "sid", table: "users"); migrationBuilder.DropColumn( name: "last_sid_reroll", table: "users"); migrationBuilder.DropColumn( name: "sid", table: "members"); } } }