alter type public."pull_result" add attribute "publications" json;

set check_function_bodies = off;

CREATE OR REPLACE FUNCTION public.pull_data(token_id uuid, client_group_id text)
 RETURNS pull_result
 LANGUAGE plpgsql
AS $function$DECLARE
    result pull_result;
BEGIN
    -- Get client group data as JSON array
    SELECT json_agg(row_to_json(rc))
    FROM replicache_clients rc
    WHERE rc.client_group = client_group_id
    INTO result.client_groups;

    -- Get facts as JSON array
    SELECT json_agg(row_to_json(f))
    FROM permission_tokens pt,
         get_facts(pt.root_entity) f
    WHERE pt.id = token_id
    INTO result.facts;

    -- Get publication data
    SELECT json_agg(row_to_json(lip))
    FROM leaflets_in_publications lip
    WHERE lip.leaflet = token_id
    INTO result.publications;

    RETURN result;
END;$function$
;