Include annotation support

packages/sdk/src/annotations/poster.ts

@@ -19,7 +19,7 @@ export const encodePost = (content: string) => {
 }
 
 export const encodeAnnotationsPost = (
-  rolesMod: `0x${string}`,
+  rolesMod: string,
   roleKey: string,
   updatePost: UpdateAnnotationsPost
 ) => {

packages/sdk/src/calls/encodeCalls.ts

@@ -1,11 +1,15 @@
 import { Roles__factory } from "../../../evm/typechain-types"
+import { encodeAnnotationsPost } from "../annotations/poster"
 import { flattenCondition } from "../conditions"
 
 import { Call } from "./types"
 
 const rolesInterface = Roles__factory.createInterface()
 
-export const encodeCalls = (calls: Call[]): `0x${string}`[] => {
+export const encodeCalls = (
+  calls: Call[],
+  rolesMod: string
+): `0x${string}`[] => {
   return calls.map((call) => {
     switch (call.call) {
       case "allowTarget": {
@@ -76,6 +80,11 @@ export const encodeCalls = (calls: Call[]): `0x${string}`[] => {
           call.timestamp,
         ])
       }
+
+      case "postAnnotations": {
+        const { roleKey, body } = call
+        return encodeAnnotationsPost(rolesMod, roleKey, body)
+      }
     }
   }) as `0x${string}`[]
 }

packages/sdk/src/calls/logCall.ts

@@ -65,8 +65,23 @@ export const logCall = (
       break
     }
 
-    // default:
-    //   log(`Unhandled call ${call.call}`)
+    case "postAnnotations": {
+      const { addAnnotations, removeAnnotations } = call.body
+      const addCount =
+        addAnnotations?.reduce((acc, add) => acc + add.uris.length, 0) || 0
+      const removeCount = removeAnnotations?.length || 0
+      const message = [
+        addCount > 0 ? "add " + pluralize(addCount, "annotation") : undefined,
+        removeCount > 0
+          ? "remove " + pluralize(removeCount, "annotation")
+          : undefined,
+      ]
+        .filter(Boolean)
+        .join(", ")
+
+      log(`💬 ${message[0].toUpperCase()}${message.slice(1)}`)
+      break
+    }
   }
 }
 
@@ -76,3 +91,14 @@ const ExecutionOptionLabel = {
   [ExecutionOptions.Send]: "call, send",
   [ExecutionOptions.Both]: "call, delegatecall, send",
 }
+
+const pluralize = (
+  count: number,
+  singular: string,
+  plural = `${singular}s`
+) => {
+  if (count === 1) {
+    return `1 ${singular}`
+  }
+  return `${count} ${plural}`
+}

packages/sdk/src/calls/types.ts

@@ -59,6 +59,18 @@ interface SetAllowanceCall {
   timestamp: bigint
 }
 
+interface PostAnnotationsCall {
+  call: "postAnnotations"
+  roleKey: string
+  body: {
+    addAnnotations?: {
+      uris: string[]
+      schema: string
+    }[]
+    removeAnnotations?: string[]
+  }
+}
+
 export type Call =
   | AllowTargetCall
   | ScopeTargetCall
@@ -68,3 +80,4 @@ export type Call =
   | RevokeFunctionCall
   | AssignRolesCall
   | SetAllowanceCall
+  | PostAnnotationsCall

packages/sdk/src/diff/all.ts

@@ -8,8 +8,8 @@ export function diffAll({
   prev,
   next,
 }: {
-  prev?: { roles: Role[]; allowances: Allowance[] }
-  next?: { roles: Role[]; allowances: Allowance[] }
+  prev: { roles: Role[]; allowances: Allowance[] } | undefined | null
+  next: { roles: Role[]; allowances: Allowance[] } | undefined | null
 }): Diff {
   return merge(
     diffRoles({

packages/sdk/src/diff/annotations.ts

@@ -0,0 +1,60 @@
+import { Annotation } from "zodiac-roles-deployments"
+
+import { groupBy } from "../utils/groupBy"
+
+import { Diff } from "./helpers"
+
+export function diffAnnotations({
+  roleKey,
+  prev = [],
+  next = [],
+}: {
+  roleKey: string
+  prev?: Annotation[]
+  next?: Annotation[]
+}): Diff {
+  const removeAnnotations = prev
+    .map((annotation) => annotation.uri)
+    .filter((uri) => !next.some((nextAnnotation) => nextAnnotation.uri === uri))
+
+  const addAnnotations = groupAnnotations(
+    next.filter(
+      (annotation) =>
+        !prev.some(
+          (currentAnnotation) =>
+            currentAnnotation.uri === annotation.uri &&
+            currentAnnotation.schema === annotation.schema
+        )
+    )
+  )
+
+  return {
+    minus:
+      removeAnnotations.length > 0
+        ? [
+            {
+              call: "postAnnotations",
+              roleKey,
+              body: { removeAnnotations },
+            },
+          ]
+        : [],
+    plus: addAnnotations
+      ? [
+          {
+            call: "postAnnotations",
+            roleKey,
+            body: { addAnnotations },
+          },
+        ]
+      : [],
+  }
+}
+
+const groupAnnotations = (annotations: readonly Annotation[]) =>
+  Object.entries(groupBy(annotations, (annotation) => annotation.schema)).map(
+    ([schema, annotations]) => ({
+      schema,
+      uris: annotations.map((annotation) => annotation.uri),
+    })
+  )