diff --git a/ExampleBots/MediaModeratorPoC/PolicyEngine.cs b/ExampleBots/MediaModeratorPoC/PolicyEngine.cs
new file mode 100644
index 0000000..0a0a565
--- /dev/null
+++ b/ExampleBots/MediaModeratorPoC/PolicyEngine.cs
@@ -0,0 +1,86 @@
+using LibMatrix.EventTypes.Spec;
+using LibMatrix.Helpers;
+using LibMatrix.Homeservers;
+using LibMatrix.RoomTypes;
+using LibMatrix.Services;
+using MediaModeratorPoC.AccountData;
+using MediaModeratorPoC.StateEventTypes;
+using Microsoft.Extensions.Logging;
+
+namespace MediaModeratorPoC;
+
+public class PolicyEngine(AuthenticatedHomeserverGeneric hs, ILogger<MediaModBot> logger, MediaModBotConfiguration configuration,
+ HomeserverResolverService hsResolver) {
+ public List<PolicyList> ActivePolicyLists { get; set; } = new();
+ private GenericRoom? _logRoom;
+ private GenericRoom? _controlRoom;
+
+ public async Task ReloadActivePolicyLists() {
+ // first time init
+ if (_logRoom is null || _controlRoom is null) {
+ var botData = await hs.GetAccountDataAsync<BotData>("gay.rory.modbot_data");
+ _logRoom ??= hs.GetRoom(botData.LogRoom ?? botData.ControlRoom);
+ _controlRoom ??= hs.GetRoom(botData.ControlRoom);
+ }
+
+ await _controlRoom?.SendMessageEventAsync(MessageFormatter.FormatSuccess("Reloading policy lists!"))!;
+ await _logRoom?.SendMessageEventAsync(
+ new RoomMessageEventContent(
+ body: "Reloading policy lists!",
+ messageType: "m.text"))!;
+
+ await _controlRoom?.SendMessageEventAsync(MessageFormatter.FormatSuccess("0/? policy lists loaded"))!;
+
+ var policyLists = new List<PolicyList>();
+ var policyListAccountData = await hs.GetAccountDataAsync<Dictionary<string, PolicyList>>("gay.rory.modbot.policy_lists");
+ foreach (var (roomId, policyList) in policyListAccountData) {
+ _logRoom?.SendMessageEventAsync(
+ new RoomMessageEventContent(
+ body: $"Loading policy list {MessageFormatter.HtmlFormatMention(roomId)}!",
+ messageType: "m.text"));
+ var room = hs.GetRoom(roomId);
+
+ policyList.Room = room;
+
+ var stateEvents = room.GetFullStateAsync();
+ await foreach (var stateEvent in stateEvents) {
+ if (stateEvent != null && stateEvent.GetType.IsAssignableTo(typeof(BasePolicy))) {
+ policyList.Policies.Add(stateEvent);
+ }
+ }
+
+ //html table of policy count by type
+ var policyCount = policyList.Policies.GroupBy(x => x.Type).ToDictionary(x => x.Key, x => x.Count());
+ var policyCountTable = policyCount.Aggregate(
+ "<table><tr><th>Policy Type</th><th>Count</th></tr>",
+ (current, policy) => current + $"<tr><td>{policy.Key}</td><td>{policy.Value}</td></tr>");
+ policyCountTable += "</table>";
+
+ var policyCountTablePlainText = policyCount.Aggregate(
+ "Policy Type | Count\n",
+ (current, policy) => current + $"{policy.Key,-16} | {policy.Value}\n");
+ await _logRoom?.SendMessageEventAsync(
+ new RoomMessageEventContent() {
+ MessageType = "org.matrix.custom.html",
+ Body = $"Policy count for {roomId}:\n{policyCountTablePlainText}",
+ FormattedBody = $"Policy count for {MessageFormatter.HtmlFormatMention(roomId)}:\n{policyCountTable}",
+ })!;
+
+ await _logRoom?.SendMessageEventAsync(
+ new RoomMessageEventContent(
+ body: $"Loaded {policyList.Policies.Count} policies for {MessageFormatter.HtmlFormatMention(roomId)}!",
+ messageType: "m.text"))!;
+
+ policyLists.Add(policyList);
+
+ var progressMsgContent = MessageFormatter.FormatSuccess($"{policyLists.Count}/{policyListAccountData.Count} policy lists loaded");
+ //edit old message
+ progressMsgContent.RelatesTo = new() {
+
+ };
+ _controlRoom?.SendMessageEventAsync(progressMsgContent);
+ }
+
+ ActivePolicyLists = policyLists;
+ }
+}
|
