File size: 4,867 Bytes
0196580 bf802f2 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 | % SPDX-License-Identifier: LicenseRef-Sovereign-Leviathan-SL-AGPL3-001
% Licensing: see LICENSE in the repository root.
%%%-------------------------------------------------------------------
%%% @doc Wire-format conversions for every entity in the data model.
%%%
%%% This is the single place a record from `include/hyperkitty.hrl'
%%% is turned into a JSON-able map. Keeping it centralized means the
%%% API layer, the event payloads, and the audit view all render a
%%% given entity identically -- there is exactly one `#browser_tab{}'
%%% -> map projection, not one per call site.
%%% @end
%%%-------------------------------------------------------------------
-module(hk_schema).
-include("hyperkitty.hrl").
-export([to_map/1]).
to_map(#agent{} = A) ->
#{
agent_id => A#agent.agent_id,
objective => opt_map(A#agent.objective),
state => A#agent.state,
capabilities => A#agent.capabilities,
action_count => A#agent.action_count,
started_at_ms => A#agent.started_at_ms,
updated_at_ms => A#agent.updated_at_ms,
limits => A#agent.limits,
result => term_or_null(A#agent.result),
failure_reason => term_or_null(A#agent.failure_reason)
};
to_map(#agent_objective{} = O) ->
#{
objective_id => O#agent_objective.objective_id,
agent_id => O#agent_objective.agent_id,
description => O#agent_objective.description,
success_criteria => bin_or_null(O#agent_objective.success_criteria),
created_at_ms => O#agent_objective.created_at_ms
};
to_map(#browser_session{} = S) ->
#{
session_id => S#browser_session.session_id,
owner_agent_id => bin_or_null(S#browser_session.owner_agent_id),
profile => S#browser_session.profile,
status => S#browser_session.status,
tabs => S#browser_session.tabs,
created_at_ms => S#browser_session.created_at_ms,
updated_at_ms => S#browser_session.updated_at_ms
};
to_map(#browser_tab{} = T) ->
#{
tab_id => T#browser_tab.tab_id,
session_id => T#browser_tab.session_id,
url => bin_or_null(T#browser_tab.url),
title => bin_or_null(T#browser_tab.title),
status => T#browser_tab.status,
created_at_ms => T#browser_tab.created_at_ms,
updated_at_ms => T#browser_tab.updated_at_ms
};
to_map(#search_job{} = J) ->
#{
job_id => J#search_job.job_id,
owner_agent_id => bin_or_null(J#search_job.owner_agent_id),
query => J#search_job.query,
stage => J#search_job.stage,
max_iterations => J#search_job.max_iterations,
iteration => J#search_job.iteration,
results => J#search_job.results,
created_at_ms => J#search_job.created_at_ms,
updated_at_ms => J#search_job.updated_at_ms
};
to_map(#search_result{} = R) ->
#{
result_id => R#search_result.result_id,
job_id => R#search_result.job_id,
url => R#search_result.url,
normalized_url => R#search_result.normalized_url,
title => bin_or_null(R#search_result.title),
snippet => bin_or_null(R#search_result.snippet),
rank => term_or_null(R#search_result.rank),
source_document_id => bin_or_null(R#search_result.source_document_id),
created_at_ms => R#search_result.created_at_ms
};
to_map(#source_document{} = D) ->
#{
document_id => D#source_document.document_id,
url => D#source_document.url,
fetched_at_ms => D#source_document.fetched_at_ms,
content_type => bin_or_null(D#source_document.content_type),
extracted_text => bin_or_null(D#source_document.extracted_text),
citation_map => D#source_document.citation_map
};
to_map(#message{} = M) ->
#{
message_id => M#message.message_id,
sender => M#message.sender,
recipient => M#message.recipient,
timestamp => M#message.timestamp,
conversation_id => M#message.conversation_id,
message_type => M#message.message_type,
payload => M#message.payload,
status => M#message.status
};
to_map(#operation{} = Op) ->
#{
operation_id => Op#operation.operation_id,
request_id => Op#operation.request_id,
kind => Op#operation.kind,
status => Op#operation.status,
subject => case Op#operation.subject of
{Type, Id} -> #{type => Type, id => Id};
undefined -> null
end,
error => term_or_null(Op#operation.error),
created_at_ms => Op#operation.created_at_ms,
updated_at_ms => Op#operation.updated_at_ms
}.
%%% internal
opt_map(undefined) -> null;
opt_map(Record) -> to_map(Record).
bin_or_null(undefined) -> null;
bin_or_null(B) -> B.
term_or_null(undefined) -> null;
term_or_null(T) -> T.
|