feat(core): add location-scoped config loading (#29625)

packages/core/test/catalog.test.ts

@@ -5,14 +5,21 @@ import { EventV2 } from "@opencode-ai/core/event"
 import { Location } from "@opencode-ai/core/location"
 import { ModelV2 } from "@opencode-ai/core/model"
 import { PluginV2 } from "@opencode-ai/core/plugin"
+import { Policy } from "@opencode-ai/core/policy"
 import { ProviderV2 } from "@opencode-ai/core/provider"
+import { AbsolutePath } from "@opencode-ai/core/schema"
+import { location } from "./fixture/location"
 import { testEffect } from "./lib/effect"
 
-const locationLayer = Layer.succeed(Location.Service, Location.Service.of({ directory: "test" }))
+const locationLayer = Layer.succeed(
+  Location.Service,
+  Location.Service.of(location({ directory: AbsolutePath.make("test") })),
+)
 const it = testEffect(
   Catalog.layer.pipe(
     Layer.provideMerge(EventV2.defaultLayer),
     Layer.provideMerge(PluginV2.defaultLayer),
+    Layer.provideMerge(Policy.defaultLayer),
     Layer.provideMerge(locationLayer),
   ),
 )
@@ -22,9 +29,9 @@ describe("CatalogV2", () => {
     Effect.gen(function* () {
       const catalog = yield* Catalog.Service
       const providerID = ProviderV2.ID.make("test")
-      const load = yield* catalog.loader()
+      const transform = yield* catalog.transform()
 
-      yield* load((catalog) =>
+      yield* transform((catalog) =>
         catalog.provider.update(providerID, (provider) => {
           provider.endpoint = {
             type: "aisdk",
@@ -48,9 +55,9 @@ describe("CatalogV2", () => {
       const catalog = yield* Catalog.Service
       const providerID = ProviderV2.ID.make("test")
       const modelID = ModelV2.ID.make("model")
-      const load = yield* catalog.loader()
+      const transform = yield* catalog.transform()
 
-      yield* load((catalog) => {
+      yield* transform((catalog) => {
         catalog.provider.update(providerID, (provider) => {
           provider.endpoint = {
             type: "aisdk",
@@ -77,9 +84,9 @@ describe("CatalogV2", () => {
       const catalog = yield* Catalog.Service
       const providerID = ProviderV2.ID.make("test")
       const modelID = ModelV2.ID.make("model")
-      const load = yield* catalog.loader()
+      const transform = yield* catalog.transform()
 
-      yield* load((catalog) => {
+      yield* transform((catalog) => {
         catalog.provider.update(providerID, (provider) => {
           provider.endpoint = {
             type: "aisdk",
@@ -104,14 +111,14 @@ describe("CatalogV2", () => {
       const plugin = yield* PluginV2.Service
       const providerID = ProviderV2.ID.make("test")
       const seen: unknown[] = []
-      const load = yield* catalog.loader()
+      const transform = yield* catalog.transform()
 
       yield* plugin.add({
         id: PluginV2.ID.make("test"),
         effect: Effect.succeed({
           "catalog.transform": (evt) =>
             Effect.sync(() => {
-              const item = evt.data.find((record) => record.provider.id === providerID)
+              const item = evt.provider.get(providerID)
               if (!item) return
               seen.push(item.provider.endpoint.type)
               if (item?.provider.endpoint.type === "aisdk") seen.push(item.provider.endpoint.url)
@@ -119,7 +126,7 @@ describe("CatalogV2", () => {
             }),
         }),
       })
-      yield* load((catalog) =>
+      yield* transform((catalog) =>
         catalog.provider.update(providerID, (provider) => {
           provider.endpoint = { type: "aisdk", package: "@ai-sdk/openai-compatible" }
           provider.options.aisdk.provider.baseURL = "https://provider.example.com"
@@ -135,9 +142,9 @@ describe("CatalogV2", () => {
       const catalog = yield* Catalog.Service
       const plugin = yield* PluginV2.Service
       const providerID = ProviderV2.ID.make("test")
-      const load = yield* catalog.loader()
+      const transform = yield* catalog.transform()
 
-      yield* load((catalog) =>
+      yield* transform((catalog) =>
         catalog.provider.update(providerID, (provider) => {
           provider.name = "Before"
         }),
@@ -164,9 +171,9 @@ describe("CatalogV2", () => {
       const catalog = yield* Catalog.Service
       const providerID = ProviderV2.ID.make("test")
       const modelID = ModelV2.ID.make("model")
-      const load = yield* catalog.loader()
+      const transform = yield* catalog.transform()
 
-      yield* load((catalog) => {
+      yield* transform((catalog) => {
         catalog.provider.update(providerID, (provider) => {
           provider.options.headers.provider = "provider"
           provider.options.headers.shared = "provider"
@@ -194,9 +201,9 @@ describe("CatalogV2", () => {
     Effect.gen(function* () {
       const catalog = yield* Catalog.Service
       const providerID = ProviderV2.ID.make("test")
-      const load = yield* catalog.loader()
+      const transform = yield* catalog.transform()
 
-      yield* load((catalog) => {
+      yield* transform((catalog) => {
         catalog.provider.update(providerID, (provider) => {
           provider.enabled = { via: "custom", data: {} }
         })
@@ -212,13 +219,44 @@ describe("CatalogV2", () => {
     }),
   )
 
+  it.effect("uses a transform-provided default model until that transform is replaced", () =>
+    Effect.gen(function* () {
+      const catalog = yield* Catalog.Service
+      const providerID = ProviderV2.ID.make("test")
+      const old = ModelV2.ID.make("old")
+      const newest = ModelV2.ID.make("new")
+      const transform = yield* catalog.transform()
+
+      const models = (catalog: Catalog.Editor) => {
+        catalog.provider.update(providerID, (provider) => {
+          provider.enabled = { via: "custom", data: {} }
+        })
+        catalog.model.update(providerID, old, (model) => {
+          model.time.released = DateTime.makeUnsafe(1000)
+        })
+        catalog.model.update(providerID, newest, (model) => {
+          model.time.released = DateTime.makeUnsafe(2000)
+        })
+      }
+
+      yield* transform((catalog) => {
+        models(catalog)
+        catalog.model.default.set(providerID, old)
+      })
+      expect(Option.getOrUndefined(yield* catalog.model.default())?.id).toBe(old)
+
+      yield* transform(models)
+      expect(Option.getOrUndefined(yield* catalog.model.default())?.id).toBe(newest)
+    }),
+  )
+
   it.effect("small model prefers small keyword candidates before cost scoring", () =>
     Effect.gen(function* () {
       const catalog = yield* Catalog.Service
       const providerID = ProviderV2.ID.make("test")
-      const load = yield* catalog.loader()
+      const transform = yield* catalog.transform()
 
-      yield* load((catalog) => {
+      yield* transform((catalog) => {
         catalog.provider.update(providerID, () => {})
         catalog.model.update(providerID, ModelV2.ID.make("cheap-large"), (model) => {
           model.capabilities.input = ["text"]
@@ -237,4 +275,23 @@ describe("CatalogV2", () => {
       expect(Option.getOrUndefined(yield* catalog.model.small(providerID))?.id).toMatch("expensive-mini")
     }),
   )
+
+  it.effect("removes providers denied by policy after loading", () =>
+    Effect.gen(function* () {
+      const catalog = yield* Catalog.Service
+      const policy = yield* Policy.Service
+      const providerID = ProviderV2.ID.make("blocked")
+      const transform = yield* catalog.transform()
+
+      yield* policy.load([new Policy.Info({ effect: "deny", action: "provider.use", resource: "blocked" })])
+      yield* transform((catalog) => {
+        catalog.provider.update(providerID, () => {})
+        catalog.model.update(providerID, ModelV2.ID.make("model"), () => {})
+      })
+
+      expect(yield* catalog.provider.all()).toEqual([])
+      expect(yield* catalog.model.all()).toEqual([])
+      expect(yield* catalog.provider.get(providerID).pipe(Effect.option)).toEqual(Option.none())
+    }),
+  )
 })