Skip to content

add records resolution#1974

Open
sevenzing wants to merge 15 commits into
mainfrom
ll/omnigraph-resolution-api
Open

add records resolution#1974
sevenzing wants to merge 15 commits into
mainfrom
ll/omnigraph-resolution-api

Conversation

@sevenzing
Copy link
Copy Markdown
Member

@sevenzing sevenzing commented Apr 21, 2026

Summary

  • Add Domain.records to Omnigraph for live ENS forward resolution (texts, addresses, contenthash, pubkey, ABI, interfaces, etc.), driven by the GraphQL field selection.
  • Add Account.primaryNames for ENSIP-19 multichain primary name resolution, with optional chainIds filtering.
  • Add example queries for the new fields

Why

Omnigraph previously exposed only indexed ENS state. Resolution is important thing for ENS developers so this PR adds resolution alongside that data so GraphQL clients can fetch live records and primary names in the same query shape as the rest of the API! It's cool, is it?


Testing

  • Unit tests: build-records-selection.test.ts (GraphQL selection → ResolverRecordsSelection), validate-primary-names-chain-ids.test.ts.
  • Integration tests: domain.integration.test.ts (Domain.records for address/text and all record types), account.integration.test.ts (Account.primaryNames per-chain, all chains, and invalid chain id rejection).
  • SDK example queries added for domain-records and account-primary-names.
  • Run ensnode against devnet and tested queries in ensadmin locally 👍

Notes for Reviewer (Optional)

  • Record types to resolve are inferred from the GraphQL selection set on Domain.records (e.g. records { texts(keys: [...]) addresses(coinTypes: [...]) })
  • Both resolution fields accept disableAcceleration; acceleration is gated by request context (canAccelerate). Since we want to enable acceleration by default decided to rename old accelecate param to disableAcceleration
  • Please @shrugs take a closer look at apps/ensapi/src/omnigraph-api/lib/build-records-selection.ts. This is my first dive into graphql query world, still not an expert in this.

Pre-Review Checklist (Blocking)

  • This PR does not introduce significant changes and is low-risk to review quickly.
  • Relevant changesets are included (or are not required)

@sevenzing sevenzing requested a review from a team as a code owner April 21, 2026 17:03
@vercel
Copy link
Copy Markdown
Contributor

vercel Bot commented Apr 21, 2026

The latest updates on your projects. Learn more about Vercel for GitHub.

Project Deployment Actions Updated (UTC)
admin.ensnode.io Ready Ready Preview, Comment May 20, 2026 8:04pm
enskit-react-example.ensnode.io Ready Ready Preview, Comment May 20, 2026 8:04pm
ensnode.io Ready Ready Preview, Comment May 20, 2026 8:04pm
ensrainbow.io Ready Ready Preview, Comment May 20, 2026 8:04pm

@changeset-bot
Copy link
Copy Markdown

changeset-bot Bot commented Apr 21, 2026

🦋 Changeset detected

Latest commit: cf7cf01

The changes in this PR will be included in the next version bump.

This PR includes changesets to release 22 packages
Name Type
ensapi Patch
ensindexer Patch
ensadmin Patch
ensrainbow Patch
fallback-ensapi Patch
enssdk Patch
enscli Patch
enskit Patch
ensskills Patch
@ensnode/datasources Patch
@ensnode/ensrainbow-sdk Patch
@ensnode/ensdb-sdk Patch
@ensnode/ensnode-sdk Patch
@ensnode/integration-test-env Patch
@ensnode/ponder-sdk Patch
@ensnode/ponder-subgraph Patch
@ensnode/shared-configs Patch
@docs/ensnode Patch
@docs/ensrainbow Patch
@namehash/ens-referrals Patch
@namehash/namehash-ui Patch
@ensnode/ensindexer-perf-testing Patch

Not sure what this means? Click here to learn what changesets are.

Click here if you're a maintainer who wants to add another changeset to this PR

@sevenzing sevenzing marked this pull request as draft April 21, 2026 17:03
@coderabbitai
Copy link
Copy Markdown
Contributor

coderabbitai Bot commented Apr 21, 2026

Review Change Stack

Warning

Rate limit exceeded

@sevenzing has exceeded the limit for the number of commits that can be reviewed per hour. Please wait 19 minutes and 48 seconds before requesting another review.

You’ve run out of usage credits. Purchase more in the billing tab.

⌛ How to resolve this issue?

After the wait time has elapsed, a review can be triggered using the @coderabbitai review command as a PR comment. Alternatively, push new commits to this PR.

We recommend that you space out your commits to avoid hitting the rate limit.

🚦 How do rate limits work?

CodeRabbit enforces hourly rate limits for each developer per organization.

Our paid plans have higher rate limits than the trial, open-source and free plans. In all cases, we re-allow further reviews after a brief timeout.

Please see our FAQ for further information.

ℹ️ Review info
⚙️ Run configuration

Configuration used: Organization UI

Review profile: ASSERTIVE

Plan: Pro

Run ID: d12cd357-932e-4781-b6d7-a16f952713b3

📥 Commits

Reviewing files that changed from the base of the PR and between e1e5680 and cf7cf01.

⛔ Files ignored due to path filters (2)
  • packages/enssdk/src/omnigraph/generated/introspection.ts is excluded by !**/generated/**
  • packages/enssdk/src/omnigraph/generated/schema.graphql is excluded by !**/generated/**
📒 Files selected for processing (12)
  • .changeset/omnigraph-resolution-api.md
  • apps/ensapi/src/omnigraph-api/builder.ts
  • apps/ensapi/src/omnigraph-api/lib/build-records-selection.ts
  • apps/ensapi/src/omnigraph-api/lib/find-domains/find-domains-resolver.ts
  • apps/ensapi/src/omnigraph-api/schema/account.ts
  • apps/ensapi/src/omnigraph-api/schema/domain.ts
  • apps/ensapi/src/omnigraph-api/schema/resolution.ts
  • apps/ensapi/src/omnigraph-api/schema/scalars.ts
  • apps/ensapi/src/omnigraph-api/yoga.ts
  • package.json
  • packages/enssdk/src/omnigraph/graphql.ts
  • packages/integration-test-env/package.json
📝 Walkthrough

Walkthrough

Adds ENS resolution GraphQL types and InterfaceId scalar, parses GraphQL selections for records, introduces Domain.records and Account.primaryNames resolvers with tracing/acceleration controls, wires context/middlewares, and adds unit/integration tests.

Changes

Resolution Integration

Layer / File(s) Summary
Scalars and builder mappings
apps/ensapi/src/omnigraph-api/builder.ts, apps/ensapi/src/omnigraph-api/schema/scalars.ts, packages/enssdk/src/omnigraph/graphql.ts
Register InterfaceId scalar, add its mapping to BuilderScalars and OmnigraphScalars, and include scalar definition/validation.
Context, Yoga, and handler wiring
apps/ensapi/src/omnigraph-api/context.ts, apps/ensapi/src/omnigraph-api/yoga.ts, apps/ensapi/src/handlers/api/omnigraph/omnigraph-api.ts, apps/ensapi/src/omnigraph-api/lib/find-domains/find-domains-resolver.ts
Introduce Omnigraph server context type and createOmnigraphContext, wire Yoga to use it, add realtime/canAccelerate middlewares to the handler, and update resolver context types.
Resolved records GraphQL refs
apps/ensapi/src/omnigraph-api/schema/resolution.ts
Declare PrimaryNameByChainRef and resolved record refs (ResolvedTextRecordRef, ResolvedAddressRecordRef, ResolvedPubkeyRecordRef, ResolvedAbiRecordRef, ResolvedInterfaceRecordRef, ResolvedRecordsRef) with parametric list fields.
Records selection config & parser + tests
apps/ensapi/src/omnigraph-api/lib/records-selection-config.ts, apps/ensapi/src/omnigraph-api/lib/build-records-selection.ts, apps/ensapi/src/omnigraph-api/lib/build-records-selection.test.ts
Map GraphQL ResolvedRecords fields to selection keys, implement buildRecordsSelectionFromResolveInfo, and add tests for simple/parametric selections, merges, and empty/unknown selection errors.
Domain.records resolver and integration tests
apps/ensapi/src/omnigraph-api/schema.ts, apps/ensapi/src/omnigraph-api/schema/domain.ts, apps/ensapi/src/omnigraph-api/schema/domain.integration.test.ts
Side-effect import of resolution refs before schema build; add nullable Domain.records that derives a selection, runs resolveForward under runWithTrace with acceleration controls, and integration tests validating addresses/texts.
Account.primaryNames + validation + tests
apps/ensapi/src/omnigraph-api/schema/account.ts, apps/ensapi/src/omnigraph-api/schema/account.integration.test.ts, apps/ensapi/src/omnigraph-api/lib/validate-primary-names-chain-ids.ts, apps/ensapi/src/omnigraph-api/lib/validate-primary-names-chain-ids.test.ts
Add Account.primaryNames(chainIds?, disableAcceleration?) with validatePrimaryNamesChainIds, run resolvePrimaryNames under runWithTrace, and tests for valid/invalid chainIds.

Sequence Diagram

sequenceDiagram
  actor Client
  participant GQL as GraphQL Resolver
  participant Selection as Selection Parser
  participant Resolver as resolveForward / resolvePrimaryNames
  participant Trace as runWithTrace
  Client->>GQL: Query Domain.records / Account.primaryNames
  GQL->>Selection: build selection from GraphQLResolveInfo
  GQL->>Trace: runWithTrace(Resolver(name/address, selection, accelerate?))
  Trace->>Resolver: invoke resolveForward / resolvePrimaryNames
  Resolver-->>GQL: resolved records / primary names
  GQL-->>Client: Response (lists or null)
Loading

Estimated code review effort

🎯 3 (Moderate) | ⏱️ ~25 minutes

Possibly related PRs

Poem

🐰 I hopped the codebase, nose a-gleam,
Scalars, refs, and selections in a stream.
Records and primaries now answer the call,
Tests in the meadow ensure none will fall.
A little rabbit applauds this neat schema dream.

🚥 Pre-merge checks | ✅ 4 | ❌ 1

❌ Failed checks (1 warning)

Check name Status Explanation Resolution
Docstring Coverage ⚠️ Warning Docstring coverage is 44.44% which is insufficient. The required threshold is 80.00%. Write docstrings for the functions missing them to satisfy the coverage threshold.
✅ Passed checks (4 passed)
Check name Status Explanation
Title check ✅ Passed The title 'add records resolution' clearly describes the primary change: adding the Domain.records field and resolution APIs to the Omnigraph GraphQL API, matching the main objective.
Linked Issues check ✅ Passed The PR successfully implements the core requirements from #1802: adds Domain.records and Account.primaryNames fields, supports acceleration controls via disableAcceleration argument, implements runtime field selection introspection for parametric fields (texts/addresses), injects canAccelerate into context, and includes comprehensive integration tests.
Out of Scope Changes check ✅ Passed All changes are in scope for issue #1802. The PR adds resolution APIs (Domain.records, Account.primaryNames), infrastructure for acceleration control (canAccelerate middleware, context), field-level selection introspection (buildRecordsSelectionFromResolveInfo, records-selection-config), integration tests, and updates scalar/type mappings. No unrelated changes detected.
Description check ✅ Passed The PR description follows the template structure with Summary, Why, Testing, Notes for Reviewer, and Pre-Review Checklist sections all present and substantive.

✏️ Tip: You can configure your own custom pre-merge checks in the settings.

✨ Finishing Touches
🧪 Generate unit tests (beta)
  • Create PR with unit tests
  • Commit unit tests in branch ll/omnigraph-resolution-api

Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

Comment @coderabbitai help to get the list of available commands and usage tips.

@greptile-apps
Copy link
Copy Markdown
Contributor

greptile-apps Bot commented Apr 21, 2026

Greptile Summary

This PR adds live ENS forward resolution to Omnigraph via two new fields: Domain.records (texts, addresses, contenthash, pubkey, ABI, interfaces, etc.) and Account.primaryNames (ENSIP-19 multichain primary names). Resolution targets are inferred from the GraphQL field selection set at runtime, and both fields accept a disableAcceleration flag. Middleware is wired so acceleration is gated by whether the indexer is close to chain head.

  • Domain.records uses buildRecordsSelectionFromResolveInfo to translate the GraphQL sub-selection into a flat ResolverRecordsSelection, then delegates to resolveForward; returns null for non-canonical domains.
  • Account.primaryNames resolves ENSIP-19 primary names across all supported chains (or a caller-specified subset), validated by a thin validatePrimaryNamesChainIds guard.
  • Integration tests cover address/text resolution, all record types for test.eth, and primary-name lookup per-chain and for all chains.

Confidence Score: 5/5

Safe to merge — new fields are additive, all existing endpoints are unchanged, and the resolution logic is guarded by null checks and GraphQL validation.

The core resolution pipeline (selection extraction → resolveForward → child field mapping) is well-structured and backed by both unit and integration tests. Both new fields are nullable or return sensible fallbacks on every failure path. The two observations are edge-case inconsistencies that do not affect common-path behavior.

No files require special attention — the changes are localized to the new resolution layer and do not touch existing indexing or query paths.

Important Files Changed

Filename Overview
apps/ensapi/src/omnigraph-api/schema/domain.ts Adds Domain.records field on the DomainInterface; uses domain.canonicalName with null-guard before delegating to resolveForward. Resolution args flow cleanly through context.
apps/ensapi/src/omnigraph-api/schema/resolution.ts Defines ResolvedRecords type with child resolvers for texts, addresses, abi, interfaces, contenthash, pubkey, dnszonehash, version, and reverseName. Each field correctly handles missing (not-selected) data with null/[] fallbacks.
apps/ensapi/src/omnigraph-api/lib/build-records-selection.ts Translates a GraphQL resolve-info selection set into a flat ResolverRecordsSelection, expanding fragments and merging multiple field nodes. Correctly throws on empty or unknown-only selections.
apps/ensapi/src/omnigraph-api/schema/account.ts Adds Account.primaryNames field; delegates to resolvePrimaryNames after validating chainIds. Uses chainIds ?? undefined to convert Pothos null to undefined correctly.
apps/ensapi/src/handlers/api/omnigraph/omnigraph-api.ts Wires canAccelerateMiddleware and makeIsRealtimeMiddleware into the Omnigraph API route, gating acceleration on indexer proximity to chain head.

Sequence Diagram

sequenceDiagram
    participant Client
    participant GraphQL as GraphQL Yoga
    participant DomainResolver as Domain.records resolver
    participant BuildSelection as buildRecordsSelectionFromResolveInfo
    participant ResolveForward as resolveForward
    participant ChildResolvers as ResolvedRecords child resolvers

    Client->>GraphQL: "query { domain { records { texts(keys:[...]) addresses(coinTypes:[...]) } } }"
    GraphQL->>DomainResolver: resolve(domain, args, context, info)
    DomainResolver->>DomainResolver: domain.canonicalName null check
    DomainResolver->>BuildSelection: buildRecordsSelectionFromResolveInfo(info)
    BuildSelection->>BuildSelection: flatten fieldNodes, expand fragments
    BuildSelection->>BuildSelection: extract parametric/simple field args
    BuildSelection-->>DomainResolver: ResolverRecordsSelection
    DomainResolver->>ResolveForward: resolveForward(name, selection, options)
    ResolveForward-->>DomainResolver: result: ResolverRecordsResponseBase
    DomainResolver-->>GraphQL: result
    GraphQL->>ChildResolvers: texts resolver, addresses resolver, etc.
    ChildResolvers-->>Client: resolved records
Loading

Reviews (4): Last reviewed commit: "Merge branch 'main' into ll/omnigraph-re..." | Re-trigger Greptile

Comment thread apps/ensapi/src/omnigraph-api/schema/domain.ts Outdated
Comment thread apps/ensapi/src/omnigraph-api/schema/resolution.ts Outdated
Copy link
Copy Markdown
Contributor

@coderabbitai coderabbitai Bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Actionable comments posted: 2

🤖 Prompt for all review comments with AI agents
Verify each finding against the current code and only fix it if needed.

Inline comments:
In `@apps/ensapi/src/omnigraph-api/schema/domain.ts`:
- Around line 224-249: Domain.records currently lacks an accelerate arg and
always calls resolveForward with accelerate: false; add a GraphQL arg
"accelerate" to the Domain.records field (the field defined at records: t.field)
and forward its value along with context.canAccelerate into resolveForward
(i.e., call resolveForward(name, {...}, { accelerate: args.accelerate,
canAccelerate: context.canAccelerate })). Also wire up the acceleration
middleware chain to the omnigraph-api router by applying
makeIsRealtimeMiddleware, indexingStatusMiddleware and canAccelerateMiddleware
(same order used in resolution-api) and update the omnigraph-api context factory
to accept the initial context param that contains canAccelerate from the
middleware so context.canAccelerate is available in the resolver; keep
getDomainInterpretedName usage unchanged.

In `@apps/ensapi/src/omnigraph-api/schema/resolution.ts`:
- Around line 16-18: Update the description string for the texts field in
resolution.ts to fix the malformed example; locate the texts: t.stringList({ ...
}) declaration and replace the example snippet "Text record keys to resolve
(e.g. `avatar`, `description`, `com.)." with a correctly punctuated example such
as "Text record keys to resolve (e.g. `avatar`, `description`, `com`)." ensuring
matching backticks and proper parentheses/period placement.
🪄 Autofix (Beta)

Fix all unresolved CodeRabbit comments on this PR:

  • Push a commit to this branch (recommended)
  • Create a new PR with the fixes

ℹ️ Review info
⚙️ Run configuration

Configuration used: Organization UI

Review profile: ASSERTIVE

Plan: Pro

Run ID: 909769d3-67ea-4cf4-873d-e0d220e4bdd9

📥 Commits

Reviewing files that changed from the base of the PR and between d9ab6b0 and eae25f6.

⛔ Files ignored due to path filters (1)
  • packages/enssdk/src/omnigraph/generated/schema.graphql is excluded by !**/generated/**
📒 Files selected for processing (4)
  • apps/ensapi/src/omnigraph-api/schema.ts
  • apps/ensapi/src/omnigraph-api/schema/domain.integration.test.ts
  • apps/ensapi/src/omnigraph-api/schema/domain.ts
  • apps/ensapi/src/omnigraph-api/schema/resolution.ts

Comment thread apps/ensapi/src/omnigraph-api/schema/domain.ts Outdated
Comment thread apps/ensapi/src/omnigraph-api/schema/resolution.ts Outdated
Copy link
Copy Markdown
Member

@shrugs shrugs left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

So i mentioned this in the original issue, which is that i'm not sure we want to allow people to query the trace and other meta information, like whether it was accelerated or not via the graphql api because i'm not convinced the ergonomics of it makes sense. but if you find a format that feels good, we can totally do so. For example, we could make the signature

query GetRecords {
  domain(by: { name: "eth" }) {
    resolve([accelerate: True]) {
      records { # equivalent to a ResolverRecordsSelection
        name
        texts(keys: ["description"]) { key value }
        addresses(coinType: [60]) { coinType address }
      }
      acceleration { requested attempted }
      trace
    }
  }
}

query GetPrimaryNames {
  account(by: { address: "0xabcd" }) {
    resolve(accelerate: False) {
      primaryNames(coinTypes: [60]) { coinType name }
      acceleration { requested attempted }
      trace
    }
  }
}

then that feels good; let's do that! then the pattern is clear; Domain.resolve.records and Account.resolve.primaryNames — feels good to me!


Note that within enskit, we need to update the cache client to mark these fields as EMBEDDED_DATA because they don't have a global key. Also, feel free to update the example app to render these records. So maybe add a new component that fetches a name's records by Domain.id or (name from the params, whatever) and then plug it into the name page.


Overall, this is definitely the correct direction and the data loader usage is correct. Let's move that helper into its own library.


to access the pothos context type, let's

  1. define a type Context = ReturnType<typeof createContext> in apps/ensapi/src/omnigraph-api/context.ts
  2. use that everywhere we currently use context: ReturnType<typeof createContext>
  3. make sure the helper you extracted here uses said type as well.

no need to alias it to graphqlContext; within the omnigraph graphql server module, "context" is always the graphql context

Comment thread apps/ensapi/src/omnigraph-api/schema/domain.integration.test.ts Outdated
Comment thread apps/ensapi/src/omnigraph-api/schema/domain.integration.test.ts Outdated
Comment thread apps/ensapi/src/omnigraph-api/schema/domain.integration.test.ts Outdated
Comment thread apps/ensapi/src/omnigraph-api/schema/domain.ts Outdated
Comment thread apps/ensapi/src/omnigraph-api/schema/resolution.ts Outdated
Copy link
Copy Markdown
Contributor

@coderabbitai coderabbitai Bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Actionable comments posted: 2

♻️ Duplicate comments (1)
apps/ensapi/src/omnigraph-api/schema/domain.ts (1)

185-201: ⚠️ Potential issue | 🟠 Major | ⚡ Quick win

Domain.records still ignores the acceleration contract.

Line 197 hardcodes accelerate: false, canAccelerate: false, so clients cannot use the documented default accelerated path (accelerate default true) and context capability is bypassed.

🤖 Prompt for AI Agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

In `@apps/ensapi/src/omnigraph-api/schema/domain.ts` around lines 185 - 201, The
Domain.records resolver currently hardcodes acceleration by calling
resolveForward(name, selection, { accelerate: false, canAccelerate: false }),
which prevents using the default accelerated path and ignores context
capabilities; update the resolve function in the records field to stop forcing
those flags — either omit the options so resolveForward uses its defaults
(allowing accelerate default true) or explicitly pass accelerate from args and
canAccelerate from the GraphQL context (e.g., use _args.accelerate ?? true and
_context.canAccelerate) when invoking resolveForward so acceleration behavior
respects client requests and context capability checks.
🤖 Prompt for all review comments with AI agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

Inline comments:
In `@apps/ensapi/src/omnigraph-api/lib/build-records-selection.ts`:
- Around line 45-48: The code currently uses only the first
GraphQLResolveInfo.fieldNodes element (fieldNode) when building the selection
and thus drops merged selections; update the build logic in
build-records-selection (where fieldNode is referenced) to merge all
info.fieldNodes by iterating over info.fieldNodes, collecting each
node.selectionSet?.selections and concatenating them into a single combined
selection array (or combined selectionSet), then use that combined selectionSet
for further processing; if no selection sets are present across all nodes, still
throw the existing GraphQLError with EMPTY_RECORDS_SELECTION_MESSAGE.

In `@apps/ensapi/src/omnigraph-api/schema/account.ts`:
- Around line 75-95: The Account.primaryNames field currently accepts chainIds
and forces acceleration off; change its args to match the contract (accept
optional accelerate?: boolean and coinTypes?: ["CoinType"] or appropriate type)
and stop hard-disabling acceleration: call resolvePrimaryNames(account.id,
/*pass coinTypes or equivalent*/ coinTypes ?? undefined, { accelerate:
accelerate ?? context.canAccelerate, canAccelerate: context.canAccelerate })
instead of {accelerate:false, canAccelerate:false}; keep input validation
(replace or adapt validatePrimaryNamesChainIds to validate the new
coinTypes/args if necessary) and ensure runWithTrace still wraps the
resolvePrimaryNames call.

---

Duplicate comments:
In `@apps/ensapi/src/omnigraph-api/schema/domain.ts`:
- Around line 185-201: The Domain.records resolver currently hardcodes
acceleration by calling resolveForward(name, selection, { accelerate: false,
canAccelerate: false }), which prevents using the default accelerated path and
ignores context capabilities; update the resolve function in the records field
to stop forcing those flags — either omit the options so resolveForward uses its
defaults (allowing accelerate default true) or explicitly pass accelerate from
args and canAccelerate from the GraphQL context (e.g., use _args.accelerate ??
true and _context.canAccelerate) when invoking resolveForward so acceleration
behavior respects client requests and context capability checks.
🪄 Autofix (Beta)

Fix all unresolved CodeRabbit comments on this PR:

  • Push a commit to this branch (recommended)
  • Create a new PR with the fixes

ℹ️ Review info
⚙️ Run configuration

Configuration used: Organization UI

Review profile: ASSERTIVE

Plan: Pro

Run ID: df655b88-c1d7-4207-b684-04d8bbcb87a2

📥 Commits

Reviewing files that changed from the base of the PR and between eae25f6 and 713f1ee.

📒 Files selected for processing (13)
  • apps/ensapi/src/omnigraph-api/builder.ts
  • apps/ensapi/src/omnigraph-api/lib/build-records-selection.test.ts
  • apps/ensapi/src/omnigraph-api/lib/build-records-selection.ts
  • apps/ensapi/src/omnigraph-api/lib/records-selection-config.ts
  • apps/ensapi/src/omnigraph-api/lib/validate-primary-names-chain-ids.test.ts
  • apps/ensapi/src/omnigraph-api/lib/validate-primary-names-chain-ids.ts
  • apps/ensapi/src/omnigraph-api/schema.ts
  • apps/ensapi/src/omnigraph-api/schema/account.integration.test.ts
  • apps/ensapi/src/omnigraph-api/schema/account.ts
  • apps/ensapi/src/omnigraph-api/schema/domain.integration.test.ts
  • apps/ensapi/src/omnigraph-api/schema/domain.ts
  • apps/ensapi/src/omnigraph-api/schema/resolution.ts
  • apps/ensapi/src/omnigraph-api/schema/scalars.ts

Comment thread apps/ensapi/src/omnigraph-api/lib/build-records-selection.ts Outdated
Comment thread apps/ensapi/src/omnigraph-api/schema/account.ts
Copy link
Copy Markdown
Contributor

@coderabbitai coderabbitai Bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Actionable comments posted: 2

Caution

Some comments are outside the diff and can’t be posted inline due to platform limitations.

⚠️ Outside diff range comments (3)
apps/ensapi/src/omnigraph-api/schema/domain.integration.test.ts (1)

551-601: 🛠️ Refactor suggestion | 🟠 Major | ⚡ Quick win

Prefer await expect(request(...)).resolves.toMatchObject(...) in these async assertions.

Please switch the two updated test cases to the preferred resolves style for consistency.

As per coding guidelines, Prefer the await expect(...).resolves.* format over await-then-expect for async assertions.

🤖 Prompt for AI Agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

In `@apps/ensapi/src/omnigraph-api/schema/domain.integration.test.ts` around lines
551 - 601, Update the two async tests ("resolves address and text records for
example.eth" and "resolves every supported record type for test.eth") to use the
await expect(...).resolves.toMatchObject(...) pattern instead of awaiting the
request and then calling expect on the result; specifically, call
request<DomainRecordsResult>(DomainRecords, {...}) and
request<DomainAllRecordsResult>(DomainRecordsAll, {...}) inside expect(...) and
append .resolves.toMatchObject(...) so the assertions are written as await
expect(request(...)).resolves.toMatchObject(...).
apps/ensapi/src/omnigraph-api/schema/resolution.ts (1)

196-246: ⚠️ Potential issue | 🟠 Major | ⚡ Quick win

Parametric record resolvers should enforce argument-scoped output.

interfaces(ids), texts(keys), and addresses(coinTypes) ignore their args and return all entries in r.interfaces/r.texts/r.addresses. That breaks field-level scoping for these parametric fields and can leak supersets when selections are merged.

Proposed fix
       interfaces: t.field({
@@
-        resolve: (r) =>
-          r.interfaces
-            ? Object.entries(r.interfaces).map(([interfaceId, implementer]) => ({
-                interfaceId: interfaceId as InterfaceId,
-                implementer,
-              }))
-            : [],
+        resolve: (r, { ids }) =>
+          ids.map((interfaceId) => ({
+            interfaceId,
+            implementer: r.interfaces?.[interfaceId] ?? null,
+          })),
       }),
@@
       texts: t.field({
@@
-        resolve: (r) =>
-          r.texts ? Object.entries(r.texts).map(([key, value]) => ({ key, value })) : [],
+        resolve: (r, { keys }) =>
+          keys.map((key) => ({
+            key,
+            value: r.texts?.[key] ?? null,
+          })),
       }),
@@
       addresses: t.field({
@@
-        resolve: (r) =>
-          r.addresses
-            ? Object.entries(r.addresses).map(([coinType, address]) => ({
-                coinType: Number(coinType) as CoinType,
-                address,
-              }))
-            : [],
+        resolve: (r, { coinTypes }) =>
+          coinTypes.map((coinType) => ({
+            coinType,
+            address: r.addresses?.[coinType] ?? null,
+          })),
       }),
🤖 Prompt for AI Agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

In `@apps/ensapi/src/omnigraph-api/schema/resolution.ts` around lines 196 - 246,
The resolvers for interfaces, texts, and addresses return all records from
r.interfaces/r.texts/r.addresses instead of restricting results to the requested
args; update the resolve functions for interfaces (field interfaces with arg
ids), texts (field texts with arg keys), and addresses (field addresses with arg
coinTypes) to iterate over the provided args (ids, keys, coinTypes) and for each
arg lookup the corresponding value in r.interfaces/r.texts/r.addresses,
returning only entries found (or an explicit null/omit if missing) and
preserving the declared return shape (InterfaceId, key/value, CoinType/address)
so the field output is scoped to the arguments.
apps/ensapi/src/omnigraph-api/lib/records-selection-config.ts (1)

45-78: ⚠️ Potential issue | 🟠 Major | ⚡ Quick win

Parametric selection assignment is currently lossy (last-write-wins).

applyToRecordsSelection overwrites texts/addresses/interfaces/abi each time. If the same parametric field appears multiple times (aliases/fragments), earlier requested values are dropped.

Proposed fix
   {
@@
     applyToRecordsSelection: (recordsSelection, args) => {
-      recordsSelection.texts = args.keys as string[];
+      const next = args.keys as string[];
+      recordsSelection.texts = [...new Set([...(recordsSelection.texts ?? []), ...next])];
     },
   },
@@
     applyToRecordsSelection: (recordsSelection, args) => {
-      recordsSelection.addresses = args.coinTypes as CoinType[];
+      const next = args.coinTypes as CoinType[];
+      recordsSelection.addresses = [
+        ...new Set([...(recordsSelection.addresses ?? []), ...next]),
+      ];
     },
   },
@@
     applyToRecordsSelection: (recordsSelection, args) => {
-      recordsSelection.abi = args.contentTypeMask as ContentType;
+      const next = args.contentTypeMask as ContentType;
+      recordsSelection.abi = ((recordsSelection.abi ?? 0n) | next) as ContentType;
     },
   },
@@
     applyToRecordsSelection: (recordsSelection, args) => {
-      recordsSelection.interfaces = args.ids as InterfaceId[];
+      const next = args.ids as InterfaceId[];
+      recordsSelection.interfaces = [
+        ...new Set([...(recordsSelection.interfaces ?? []), ...next]),
+      ];
     },
   },
🤖 Prompt for AI Agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

In `@apps/ensapi/src/omnigraph-api/lib/records-selection-config.ts` around lines
45 - 78, The current RECORDS_SELECTION_PARAMETRIC_FIELDS entries'
applyToRecordsSelection handlers overwrite recordsSelection.texts / .addresses /
.interfaces / .abi on each call, losing earlier values; change each handler to
merge into existing values instead of replacing: for texts/addresses/interfaces,
if recordsSelection.<field> is unset assign args value, otherwise append new
items and deduplicate (preserve existing entries) so repeated param occurrences
accumulate; for abi (contentTypeMask) combine masks using bitwise OR rather than
assignment; update the applyToRecordsSelection lambdas inside
RECORDS_SELECTION_PARAMETRIC_FIELDS to perform these merge/dedupe steps when
reading args.keys, args.coinTypes, args.ids and args.contentTypeMask.
♻️ Duplicate comments (1)
apps/ensapi/src/omnigraph-api/schema/domain.ts (1)

190-207: ⚠️ Potential issue | 🟠 Major | ⚡ Quick win

Use accelerate (default true) instead of inverse disableAcceleration for this GraphQL field.

This inverts the contract described for Domain.records and makes the API less ergonomic/consistent with the issue objectives.

Proposed fix
       args: {
-        disableAcceleration: t.arg.boolean({
+        accelerate: t.arg.boolean({
           required: false,
-          defaultValue: false,
-          description: "When true, disables protocol acceleration feature.",
+          defaultValue: true,
+          description: "Whether to use protocol acceleration when available.",
         }),
       },
-      resolve: async (domain, { disableAcceleration }, context, info) => {
+      resolve: async (domain, { accelerate }, context, info) => {
@@
         const { result } = await runWithTrace(() =>
           resolveForward(name, recordsSelection, {
-            accelerate: !disableAcceleration,
+            accelerate,
             canAccelerate: context.canAccelerate,
           }),
         );
🤖 Prompt for AI Agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

In `@apps/ensapi/src/omnigraph-api/schema/domain.ts` around lines 190 - 207, The
GraphQL field currently accepts disableAcceleration and inverts it when calling
resolveForward, which is confusing; change the argument to accelerate:
t.arg.boolean({ required: false, defaultValue: true, description: "When true,
enables protocol acceleration feature." }) (replace the disableAcceleration arg
and its description) and in the resolve function use the incoming accelerate
value directly when calling resolveForward (i.e., pass accelerate instead of
!disableAcceleration), keeping context.canAccelerate unchanged and leaving
buildRecordsSelectionFromResolveInfo(info) and resolveForward intact.
🤖 Prompt for all review comments with AI agents
Verify each finding against current code. Fix only still-valid issues, skip the
rest with a brief reason, keep changes minimal, and validate.

Inline comments:
In `@apps/ensapi/src/handlers/api/omnigraph/omnigraph-api.ts`:
- Line 15: Add a short explanatory comment above the
MAX_REALTIME_DISTANCE_TO_ACCELERATE constant describing that the 60-second
threshold defines the realtime window used to decide whether to accelerate
processing (e.g., to use realtime data vs. batched processing), and include the
rationale or source for the value (user-experience/latency target, backend
polling interval, or a linked issue/config) so future maintainers understand why
60 seconds was chosen; update the comment near the constant declaration
MAX_REALTIME_DISTANCE_TO_ACCELERATE: Duration = 60 with this brief
justification.

In `@apps/ensapi/src/omnigraph-api/lib/validate-primary-names-chain-ids.ts`:
- Around line 6-18: Update the JSDoc for validatePrimaryNamesChainIds to state
that this function only enforces non-empty arrays and that individual ChainId
values are validated by the GraphQL scalar (so invalid chain IDs like 0 are
rejected upstream), referencing the function name validatePrimaryNamesChainIds,
the ChainId scalar, and the EMPTY_CHAIN_IDS_MESSAGE/GraphQLError behavior to
make the division of responsibility explicit for future maintainers.

---

Outside diff comments:
In `@apps/ensapi/src/omnigraph-api/lib/records-selection-config.ts`:
- Around line 45-78: The current RECORDS_SELECTION_PARAMETRIC_FIELDS entries'
applyToRecordsSelection handlers overwrite recordsSelection.texts / .addresses /
.interfaces / .abi on each call, losing earlier values; change each handler to
merge into existing values instead of replacing: for texts/addresses/interfaces,
if recordsSelection.<field> is unset assign args value, otherwise append new
items and deduplicate (preserve existing entries) so repeated param occurrences
accumulate; for abi (contentTypeMask) combine masks using bitwise OR rather than
assignment; update the applyToRecordsSelection lambdas inside
RECORDS_SELECTION_PARAMETRIC_FIELDS to perform these merge/dedupe steps when
reading args.keys, args.coinTypes, args.ids and args.contentTypeMask.

In `@apps/ensapi/src/omnigraph-api/schema/domain.integration.test.ts`:
- Around line 551-601: Update the two async tests ("resolves address and text
records for example.eth" and "resolves every supported record type for
test.eth") to use the await expect(...).resolves.toMatchObject(...) pattern
instead of awaiting the request and then calling expect on the result;
specifically, call request<DomainRecordsResult>(DomainRecords, {...}) and
request<DomainAllRecordsResult>(DomainRecordsAll, {...}) inside expect(...) and
append .resolves.toMatchObject(...) so the assertions are written as await
expect(request(...)).resolves.toMatchObject(...).

In `@apps/ensapi/src/omnigraph-api/schema/resolution.ts`:
- Around line 196-246: The resolvers for interfaces, texts, and addresses return
all records from r.interfaces/r.texts/r.addresses instead of restricting results
to the requested args; update the resolve functions for interfaces (field
interfaces with arg ids), texts (field texts with arg keys), and addresses
(field addresses with arg coinTypes) to iterate over the provided args (ids,
keys, coinTypes) and for each arg lookup the corresponding value in
r.interfaces/r.texts/r.addresses, returning only entries found (or an explicit
null/omit if missing) and preserving the declared return shape (InterfaceId,
key/value, CoinType/address) so the field output is scoped to the arguments.

---

Duplicate comments:
In `@apps/ensapi/src/omnigraph-api/schema/domain.ts`:
- Around line 190-207: The GraphQL field currently accepts disableAcceleration
and inverts it when calling resolveForward, which is confusing; change the
argument to accelerate: t.arg.boolean({ required: false, defaultValue: true,
description: "When true, enables protocol acceleration feature." }) (replace the
disableAcceleration arg and its description) and in the resolve function use the
incoming accelerate value directly when calling resolveForward (i.e., pass
accelerate instead of !disableAcceleration), keeping context.canAccelerate
unchanged and leaving buildRecordsSelectionFromResolveInfo(info) and
resolveForward intact.
🪄 Autofix (Beta)

Fix all unresolved CodeRabbit comments on this PR:

  • Push a commit to this branch (recommended)
  • Create a new PR with the fixes

ℹ️ Review info
⚙️ Run configuration

Configuration used: Organization UI

Review profile: ASSERTIVE

Plan: Pro

Run ID: e0c05acb-eec2-4985-9af4-2595aba12a18

📥 Commits

Reviewing files that changed from the base of the PR and between 713f1ee and e1e5680.

⛔ Files ignored due to path filters (2)
  • packages/enssdk/src/omnigraph/generated/introspection.ts is excluded by !**/generated/**
  • packages/enssdk/src/omnigraph/generated/schema.graphql is excluded by !**/generated/**
📒 Files selected for processing (17)
  • apps/ensapi/src/handlers/api/omnigraph/omnigraph-api.ts
  • apps/ensapi/src/omnigraph-api/builder.ts
  • apps/ensapi/src/omnigraph-api/context.ts
  • apps/ensapi/src/omnigraph-api/lib/build-records-selection.test.ts
  • apps/ensapi/src/omnigraph-api/lib/build-records-selection.ts
  • apps/ensapi/src/omnigraph-api/lib/find-domains/find-domains-resolver.ts
  • apps/ensapi/src/omnigraph-api/lib/records-selection-config.ts
  • apps/ensapi/src/omnigraph-api/lib/validate-primary-names-chain-ids.test.ts
  • apps/ensapi/src/omnigraph-api/lib/validate-primary-names-chain-ids.ts
  • apps/ensapi/src/omnigraph-api/schema/account.integration.test.ts
  • apps/ensapi/src/omnigraph-api/schema/account.ts
  • apps/ensapi/src/omnigraph-api/schema/domain.integration.test.ts
  • apps/ensapi/src/omnigraph-api/schema/domain.ts
  • apps/ensapi/src/omnigraph-api/schema/resolution.ts
  • apps/ensapi/src/omnigraph-api/schema/scalars.ts
  • apps/ensapi/src/omnigraph-api/yoga.ts
  • packages/enssdk/src/omnigraph/graphql.ts
💤 Files with no reviewable changes (3)
  • packages/enssdk/src/omnigraph/graphql.ts
  • apps/ensapi/src/omnigraph-api/lib/validate-primary-names-chain-ids.test.ts
  • apps/ensapi/src/omnigraph-api/schema/scalars.ts

Comment thread apps/ensapi/src/handlers/api/omnigraph/omnigraph-api.ts
@sevenzing
Copy link
Copy Markdown
Member Author

@greptile review

import { indexingStatusMiddleware } from "@/middleware/indexing-status.middleware";
import { makeIsRealtimeMiddleware } from "@/middleware/is-realtime.middleware";

const MAX_REALTIME_DISTANCE_TO_ACCELERATE: Duration = 60;
Copy link
Copy Markdown
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

maybe use the same constant from resolution in ensapi? not sure since dont have a lot of context for perfect realtime distance

Copy link
Copy Markdown
Member

@shrugs shrugs left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I still prefer accelerate: false (with defaultValue: true) but i can see either.

I also see the value in renaming the name record, since it's famously confusing. so i support reverseName but a part of me wants it to be 1:1 with the protocol anyway.

need to mark these new entities as EMBEDDED_DATA in cache-exchange.ts since they don't have a cacheable unique id. actually i think we need to add all of those new fields, yikes. so PrimaryNameByChain, ResolvedTextRecord, ResolvedAddressRecord etc...

let's move all of these lib files into an omnigraph-api/lib/resolution/ sub folder


great work on the graphql info parsing! i figured that would be the gnarliest of the steps for this; i trust the tests for this but the implementation looks functional to me as well

`;

it("resolves primary name for owner on chain 1", async () => {
const result = await request<AccountPrimaryNamesResult>(AccountPrimaryNames, {
Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

nit: i prefer the

await expect(promise).resolves.toEqual();

or

await expect(promise).resolves.toMatch();

pattern for these tests, see if claude can reformat for you

});

expect(result.account.primaryNames).toEqual(
expect.arrayContaining([{ chainId: 1, name: "test.eth" }]),
Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

like here it should be compressable into

await expect(request<AccountPrimaryNamesResult>(AccountPrimaryNamesAllChains, {
      address: accounts.owner.address,
    })).resolves.toMatch({ account: { primaryNames: [{ chainId: 1, name: "test.eth" }] } })

}),
},
resolve: async (account, { chainIds, disableAcceleration }, context) => {
validatePrimaryNamesChainIds(chainIds);
Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

chainIds: t.arg({
          type: ["ChainId"],
          required: false,
          description:
            "Chain ids to resolve primary names for. Omit to resolve all ENSIP-19 supported chains.",
          validate: { minLength: 1 },
        }),

should match the behavior; it becomes optional but if defined has minLength (i think! if it doesn't work like that, keep as-is)

);

return Object.entries(result).map(([chainId, name]) => ({
chainId: Number(chainId) as ChainId,
Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

add a comment that both of these are already correctly typed but Object.entries erases types so we re-cast for the typechecker

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

and i think we can drop the Number(chainId) and just do chainId: chainId as ChainId or at the worst chainId: chainId as number as ChainId, basically to prove that we're not doing any runtime coercion, it's already a ChainId

description:
"Resolve ENS records for this Domain via the ENS protocol. Only canonical, normalized names can be resolved. Returns null if the domain is not canonical.",
type: ResolvedRecordsRef,
nullable: true,
Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

add tracing: true so this field opts into tracing and will helpfully wrap the resolution spans with Domain.records

//////////////////////
// PrimaryNameByChain
//////////////////////
export const PrimaryNameByChainRef = builder
Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

this likely isn't an issue, but safest to break these into

export const MyRef = builder.objectRef<>();
MyRef.implement({ ... })

just to avoid any circular reference situations

});
}
})
.transform((val) => val.toLowerCase() as InterfaceId)
Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

in this case let's put a transform above the .check and the cast transform as-is

.transform((val) => val.toLowerCase()) // coerce to lowercase
.check() // validate
.transform((val) => val as InterfaceId) // safe cast

graphqlEndpoint: "*",
schema,
context,
context: ({ canAccelerate }) => createOmnigraphContext({ canAccelerate }),
Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

nice! maybe just context: createOmnigraphContext if the types are happy with that?

@@ -0,0 +1,17 @@
import { ensTestEnvChain } from "@ensnode/datasources";
Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

depending on how we resolve the --only devnet usage that was just merged perhaps this becomes unnecessary?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants