lemmy/migrations/2021-02-14-041356_split_user_table/down.sql

-- user_ban
alter table person_ban rename to user_ban;
alter sequence person_ban_id_seq rename to user_ban_id_seq;
alter index person_ban_pkey rename to user_ban_pkey;
alter index person_ban_person_id_key rename to user_ban_user_id_key;
alter table user_ban rename constraint person_ban_person_id_fkey to user_ban_user_id_fkey;

-- user_mention
alter table person_mention rename to user_mention;
alter sequence person_mention_id_seq rename to user_mention_id_seq;
alter index person_mention_pkey rename to user_mention_pkey;
alter index person_mention_recipient_id_comment_id_key rename to user_mention_recipient_id_comment_id_key;
alter table user_mention rename constraint person_mention_comment_id_fkey to user_mention_comment_id_fkey;
alter table user_mention rename constraint person_mention_recipient_id_fkey to user_mention_recipient_id_fkey;

-- User aggregates table
alter table person_aggregates rename to user_aggregates;
alter sequence person_aggregates_id_seq rename to user_aggregates_id_seq;
alter table user_aggregates rename column person_id to user_id;

-- Indexes
alter index person_aggregates_pkey rename to user_aggregates_pkey;
alter index idx_person_aggregates_comment_score rename to idx_user_aggregates_comment_score;
alter index person_aggregates_person_id_key rename to user_aggregates_user_id_key;
alter table user_aggregates rename constraint person_aggregates_person_id_fkey to user_aggregates_user_id_fkey;

-- Redo the user_aggregates table
drop trigger person_aggregates_person on person;
drop trigger person_aggregates_post_count on post;
drop trigger person_aggregates_post_score on post_like;
drop trigger person_aggregates_comment_count on comment;
drop trigger person_aggregates_comment_score on comment_like;
drop function 
  person_aggregates_person, 
  person_aggregates_post_count,
  person_aggregates_post_score,
  person_aggregates_comment_count,
  person_aggregates_comment_score;

-- user_ table
-- Drop views
drop view person_alias_1, person_alias_2;

-- Rename indexes
alter index person__pkey rename to user__pkey;
alter index idx_person_actor_id rename to idx_user_actor_id;
alter index idx_person_inbox_url rename to idx_user_inbox_url;
alter index idx_person_lower_actor_id rename to idx_user_lower_actor_id;
alter index idx_person_published rename to idx_user_published;

-- Rename triggers
alter trigger site_aggregates_person_delete on person rename to site_aggregates_user_delete;
alter trigger site_aggregates_person_insert on person rename to site_aggregates_user_insert;

-- Rename the trigger functions
alter function site_aggregates_person_delete() rename to site_aggregates_user_delete;
alter function site_aggregates_person_insert() rename to site_aggregates_user_insert;

-- Rename the table back to user_
alter table person rename to user_;
alter sequence person_id_seq rename to user__id_seq;

-- Add the columns back in
alter table user_
  add column password_encrypted text not null default 'changeme',
  add column email text,
  add column admin boolean default false not null,
  add column show_nsfw boolean default false not null,
  add column theme character varying(20) default 'darkly'::character varying not null,
  add column default_sort_type smallint default 0 not null,
  add column default_listing_type smallint default 1 not null,
  add column lang character varying(20) default 'browser'::character varying not null,
  add column show_avatars boolean default true not null,
  add column send_notifications_to_email boolean default false not null,
  add column matrix_user_id text;

-- Update the user_ table with the local_user data
update user_ u set
  password_encrypted = lu.password_encrypted,
  email = lu.email,
  admin = lu.admin,
  show_nsfw = lu.show_nsfw,
  theme = lu.theme,
  default_sort_type = lu.default_sort_type,
  default_listing_type = lu.default_listing_type,
  lang = lu.lang,
  show_avatars = lu.show_avatars,
  send_notifications_to_email = lu.send_notifications_to_email,
  matrix_user_id = lu.matrix_user_id
from local_user lu
where lu.user_id = u.id;

create view user_alias_1 as select * from user_;
create view user_alias_2 as select * from user_;

drop table local_user;

-- Add the user_aggregates table triggers

-- initial user add
create function user_aggregates_user()
returns trigger language plpgsql
as $$
begin
  IF (TG_OP = 'INSERT') THEN
    insert into user_aggregates (user_id) values (NEW.id);
  ELSIF (TG_OP = 'DELETE') THEN
    delete from user_aggregates where user_id = OLD.id;
  END IF;
  return null;
end $$;

create trigger user_aggregates_user
after insert or delete on user_
for each row
execute procedure user_aggregates_user();

-- post count
create function user_aggregates_post_count()
returns trigger language plpgsql
as $$
begin
  IF (TG_OP = 'INSERT') THEN
    update user_aggregates 
    set post_count = post_count + 1 where user_id = NEW.creator_id;

  ELSIF (TG_OP = 'DELETE') THEN
    update user_aggregates 
    set post_count = post_count - 1 where user_id = OLD.creator_id;

    -- If the post gets deleted, the score calculation trigger won't fire, 
    -- so you need to re-calculate
    update user_aggregates ua
    set post_score = pd.score
    from (
      select u.id,
      coalesce(0, sum(pl.score)) as score
      -- User join because posts could be empty
      from user_ u 
      left join post p on u.id = p.creator_id
      left join post_like pl on p.id = pl.post_id
      group by u.id
    ) pd 
    where ua.user_id = OLD.creator_id;

  END IF;
  return null;
end $$;

create trigger user_aggregates_post_count
after insert or delete on post
for each row
execute procedure user_aggregates_post_count();

-- post score
create function user_aggregates_post_score()
returns trigger language plpgsql
as $$
begin
  IF (TG_OP = 'INSERT') THEN
    -- Need to get the post creator, not the voter
    update user_aggregates ua
    set post_score = post_score + NEW.score
    from post p
    where ua.user_id = p.creator_id and p.id = NEW.post_id;
    
  ELSIF (TG_OP = 'DELETE') THEN
    update user_aggregates ua
    set post_score = post_score - OLD.score
    from post p
    where ua.user_id = p.creator_id and p.id = OLD.post_id;
  END IF;
  return null;
end $$;

create trigger user_aggregates_post_score
after insert or delete on post_like
for each row
execute procedure user_aggregates_post_score();

-- comment count
create function user_aggregates_comment_count()
returns trigger language plpgsql
as $$
begin
  IF (TG_OP = 'INSERT') THEN
    update user_aggregates 
    set comment_count = comment_count + 1 where user_id = NEW.creator_id;
  ELSIF (TG_OP = 'DELETE') THEN
    update user_aggregates 
    set comment_count = comment_count - 1 where user_id = OLD.creator_id;

    -- If the comment gets deleted, the score calculation trigger won't fire, 
    -- so you need to re-calculate
    update user_aggregates ua
    set comment_score = cd.score
    from (
      select u.id,
      coalesce(0, sum(cl.score)) as score
      -- User join because comments could be empty
      from user_ u 
      left join comment c on u.id = c.creator_id
      left join comment_like cl on c.id = cl.comment_id
      group by u.id
    ) cd 
    where ua.user_id = OLD.creator_id;
  END IF;
  return null;
end $$;

create trigger user_aggregates_comment_count
after insert or delete on comment
for each row
execute procedure user_aggregates_comment_count();

-- comment score
create function user_aggregates_comment_score()
returns trigger language plpgsql
as $$
begin
  IF (TG_OP = 'INSERT') THEN
    -- Need to get the post creator, not the voter
    update user_aggregates ua
    set comment_score = comment_score + NEW.score
    from comment c
    where ua.user_id = c.creator_id and c.id = NEW.comment_id;
  ELSIF (TG_OP = 'DELETE') THEN
    update user_aggregates ua
    set comment_score = comment_score - OLD.score
    from comment c
    where ua.user_id = c.creator_id and c.id = OLD.comment_id;
  END IF;
  return null;
end $$;

create trigger user_aggregates_comment_score
after insert or delete on comment_like
for each row
execute procedure user_aggregates_comment_score();