Split message/proxy data up in MessageContext and ProxyMember

PluralKit.Core/Database/Functions/DatabaseFunctionsExt.cs

@@ -0,0 +1,25 @@
+using System.Collections.Generic;
+using System.Data;
+using System.Threading.Tasks;
+
+using Dapper;
+
+namespace PluralKit.Core
+{
+    public static class DatabaseFunctionsExt
+    {
+        public static Task<MessageContext> QueryMessageContext(this IDbConnection conn, ulong account, ulong guild, ulong channel)
+        {
+            return conn.QueryFirstAsync<MessageContext>("message_context", 
+                new { account_id = account, guild_id = guild, channel_id = channel }, 
+                commandType: CommandType.StoredProcedure);
+        }  
+        
+        public static Task<IEnumerable<ProxyMember>> QueryProxyMembers(this IDbConnection conn, ulong account, ulong guild)
+        {
+            return conn.QueryAsync<ProxyMember>("proxy_members", 
+                new { account_id = account, guild_id = guild }, 
+                commandType: CommandType.StoredProcedure);
+        }  
+    }
+}

PluralKit.Core/Database/Functions/MessageContext.cs

@@ -0,0 +1,27 @@
+#nullable enable
+using System.Collections.Generic;
+
+using NodaTime;
+
+namespace PluralKit.Core
+{
+    /// <summary>
+    /// Model for the `message_context` PL/pgSQL function in `functions.sql`
+    /// </summary>
+    public class MessageContext
+    {
+        public int? SystemId { get; set; }
+        public ulong? LogChannel { get; set; }
+        public bool InBlacklist { get; set; }
+        public bool InLogBlacklist { get; set; }
+        public bool LogCleanupEnabled { get; set; }
+        public bool ProxyEnabled { get; set; }
+        public AutoproxyMode AutoproxyMode { get; set; }
+        public int? AutoproxyMember { get; set; }
+        public ulong? LastMessage { get; set; }
+        public int? LastMessageMember { get; set; }
+        public int LastSwitch { get; set; }
+        public IReadOnlyList<int> LastSwitchMembers { get; set; } = new int[0];
+        public Instant LastSwitchTimestamp { get; set; }
+    }
+}

PluralKit.Core/Database/Functions/ProxyMember.cs

@@ -0,0 +1,17 @@
+#nullable enable
+using System.Collections.Generic;
+
+namespace PluralKit.Core
+{
+    /// <summary>
+    /// Model for the `proxy_members` PL/pgSQL function in `functions.sql`
+    /// </summary>
+    public class ProxyMember
+    {
+        public int Id { get; set; }
+        public IReadOnlyCollection<ProxyTag> ProxyTags { get; set; } = new ProxyTag[0];
+        public bool KeepProxy { get; set; }
+        public string ProxyName { get; set; } = "";
+        public string? ProxyAvatar { get; set; }
+    }
+}

PluralKit.Core/Database/Functions/functions.sql

@@ -0,0 +1,73 @@
+create function message_context(account_id bigint, guild_id bigint, channel_id bigint)
+    returns table (
+        system_id int,
+        log_channel bigint,
+        in_blacklist bool,
+        in_log_blacklist bool,
+        log_cleanup_enabled bool,
+        proxy_enabled bool,
+        autoproxy_mode int,
+        autoproxy_member int,
+        last_message bigint,
+        last_message_member int,
+        last_switch int,
+        last_switch_members int[],
+        last_switch_timestamp timestamp
+    )
+as $$
+    with
+        system as (select systems.* from accounts inner join systems on systems.id = accounts.system where accounts.uid = account_id),
+        guild as (select * from servers where id = guild_id),
+        last_message as (select * from messages where messages.guild = guild_id and messages.sender = account_id order by mid desc limit 1)
+    select
+        system.id as system_id,
+        guild.log_channel,
+        (channel_id = any(guild.blacklist)) as in_blacklist,
+        (channel_id = any(guild.log_blacklist)) as in_log_blacklist,
+        guild.log_cleanup_enabled,
+        coalesce(system_guild.proxy_enabled, true) as proxy_enabled,
+        coalesce(system_guild.autoproxy_mode, 1) as autoproxy_mode,
+        system_guild.autoproxy_member,
+        last_message.mid as last_message,
+        last_message.member as last_message_member,
+        system_last_switch.switch as last_switch,
+        system_last_switch.members as last_switch_members,
+        system_last_switch.timestamp as last_switch_timestamp
+    from system
+        full join guild on true
+        left join last_message on true
+        
+        left join system_last_switch on system_last_switch.system = system.id
+        left join system_guild on system_guild.system = system.id and system_guild.guild = guild_id
+$$ language sql stable rows 1;
+
+
+-- Fetches info about proxying related to a given account/guild
+-- Returns one row per member in system, should be used in conjuction with `message_context` too
+create function proxy_members(account_id bigint, guild_id bigint)
+    returns table (
+        id int,
+        proxy_tags proxy_tag[],
+        keep_proxy bool,
+        proxy_name text,
+        proxy_avatar text
+    )
+as $$
+    select
+        -- Basic data
+        members.id as id,
+        members.proxy_tags as proxy_tags,
+        members.keep_proxy as keep_proxy,
+    
+        -- Proxy info
+        case
+            when systems.tag is not null then (coalesce(member_guild.display_name, members.display_name, members.name) || ' ' || systems.tag) 
+            else coalesce(member_guild.display_name, members.display_name, members.name)
+        end as proxy_name,
+        coalesce(member_guild.avatar_url, members.avatar_url, systems.avatar_url) as proxy_avatar
+    from accounts
+        inner join systems on systems.id = accounts.system
+        inner join members on members.system = systems.id
+        left join member_guild on member_guild.member = members.id and member_guild.guild = guild_id
+    where accounts.uid = account_id
+$$ language sql stable rows 10;