lemmy/migrations/2021-03-09-171136_split_user_table_2/down.sql

-- post_saved
alter table post_saved rename column person_id to user_id;
alter table post_saved rename constraint post_saved_post_id_person_id_key to post_saved_post_id_user_id_key;
alter table post_saved rename constraint post_saved_person_id_fkey to post_saved_user_id_fkey;

-- post_read
alter table post_read rename column person_id to user_id;
alter table post_read rename constraint post_read_post_id_person_id_key to post_read_post_id_user_id_key;
alter table post_read rename constraint post_read_person_id_fkey to post_read_user_id_fkey;

-- post_like
alter table post_like rename column person_id to user_id;
alter index idx_post_like_person rename to idx_post_like_user;
alter table post_like rename constraint post_like_post_id_person_id_key to post_like_post_id_user_id_key;
alter table post_like rename constraint post_like_person_id_fkey to post_like_user_id_fkey;

-- password_reset_request
delete from password_reset_request;
alter table password_reset_request drop column local_user_id;
alter table password_reset_request add column user_id integer not null references person(id) on update cascade on delete cascade;

-- mod_sticky_post
alter table mod_sticky_post rename column mod_person_id to mod_user_id; 
alter table mod_sticky_post rename constraint mod_sticky_post_mod_person_id_fkey to mod_sticky_post_mod_user_id_fkey;

-- mod_remove_post
alter table mod_remove_post rename column mod_person_id to mod_user_id; 
alter table mod_remove_post rename constraint mod_remove_post_mod_person_id_fkey to mod_remove_post_mod_user_id_fkey;

-- mod_remove_community
alter table mod_remove_community rename column mod_person_id to mod_user_id; 
alter table mod_remove_community rename constraint mod_remove_community_mod_person_id_fkey to mod_remove_community_mod_user_id_fkey;

-- mod_remove_comment
alter table mod_remove_comment rename column mod_person_id to mod_user_id; 
alter table mod_remove_comment rename constraint mod_remove_comment_mod_person_id_fkey to mod_remove_comment_mod_user_id_fkey;

-- mod_lock_post
alter table mod_lock_post rename column mod_person_id to mod_user_id; 
alter table mod_lock_post rename constraint mod_lock_post_mod_person_id_fkey to mod_lock_post_mod_user_id_fkey;

-- mod_add_community
alter table mod_ban_from_community rename column mod_person_id to mod_user_id; 
alter table mod_ban_from_community rename column other_person_id to other_user_id; 
alter table mod_ban_from_community rename constraint mod_ban_from_community_mod_person_id_fkey to mod_ban_from_community_mod_user_id_fkey;
alter table mod_ban_from_community rename constraint mod_ban_from_community_other_person_id_fkey to mod_ban_from_community_other_user_id_fkey;

-- mod_ban
alter table mod_ban rename column mod_person_id to mod_user_id; 
alter table mod_ban rename column other_person_id to other_user_id; 
alter table mod_ban rename constraint mod_ban_mod_person_id_fkey to mod_ban_mod_user_id_fkey;
alter table mod_ban rename constraint mod_ban_other_person_id_fkey to mod_ban_other_user_id_fkey;

-- mod_add_community
alter table mod_add_community rename column mod_person_id to mod_user_id; 
alter table mod_add_community rename column other_person_id to other_user_id; 
alter table mod_add_community rename constraint mod_add_community_mod_person_id_fkey to mod_add_community_mod_user_id_fkey;
alter table mod_add_community rename constraint mod_add_community_other_person_id_fkey to mod_add_community_other_user_id_fkey;

-- mod_add
alter table mod_add rename column mod_person_id to mod_user_id; 
alter table mod_add rename column other_person_id to other_user_id; 
alter table mod_add rename constraint mod_add_mod_person_id_fkey to mod_add_mod_user_id_fkey;
alter table mod_add rename constraint mod_add_other_person_id_fkey to mod_add_other_user_id_fkey;

-- community_user_ban
alter table community_person_ban rename to community_user_ban;
alter sequence community_person_ban_id_seq rename to community_user_ban_id_seq;
alter table community_user_ban rename column person_id to user_id;
alter table community_user_ban rename constraint community_person_ban_pkey to community_user_ban_pkey; 
alter table community_user_ban rename constraint community_person_ban_community_id_fkey to community_user_ban_community_id_fkey;
alter table community_user_ban rename constraint community_person_ban_community_id_person_id_key to community_user_ban_community_id_user_id_key;
alter table community_user_ban rename constraint community_person_ban_person_id_fkey to community_user_ban_user_id_fkey;

-- community_moderator
alter table community_moderator rename column person_id to user_id;
alter table community_moderator rename constraint community_moderator_community_id_person_id_key to community_moderator_community_id_user_id_key;
alter table community_moderator rename constraint community_moderator_person_id_fkey to community_moderator_user_id_fkey;

-- community_follower
alter table community_follower rename column person_id to user_id;
alter table community_follower rename constraint community_follower_community_id_person_id_key to community_follower_community_id_user_id_key;
alter table community_follower rename constraint community_follower_person_id_fkey to community_follower_user_id_fkey;

-- comment_saved
alter table comment_saved rename column person_id to user_id;
alter table comment_saved rename constraint comment_saved_comment_id_person_id_key to comment_saved_comment_id_user_id_key;
alter table comment_saved rename constraint comment_saved_person_id_fkey to comment_saved_user_id_fkey;

-- comment_like
alter table comment_like rename column person_id to user_id;
alter index idx_comment_like_person rename to idx_comment_like_user;
alter table comment_like rename constraint comment_like_comment_id_person_id_key to comment_like_comment_id_user_id_key;
alter table comment_like rename constraint comment_like_person_id_fkey to comment_like_user_id_fkey;

-- 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 column person_id to user_id;
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.person_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();

-- redo site aggregates trigger
create or replace function site_aggregates_activity(i text) returns integer
    language plpgsql
    as $$
declare
   count_ integer;
begin
  select count(*)
  into count_
  from (
    select c.creator_id from comment c
    inner join user_ u on c.creator_id = u.id
    where c.published > ('now'::timestamp - i::interval) 
    and u.local = true
    union
    select p.creator_id from post p
    inner join user_ u on p.creator_id = u.id
    where p.published > ('now'::timestamp - i::interval)
    and u.local = true
  ) a;
  return count_;
end;
$$;