[HTTP] adds getStreamMetadata API

.chronus/changes/get-stream-metadata-2024-10-19-16-52-8.md

@@ -0,0 +1,7 @@
+---
+changeKind: feature
+packages:
+  - "@typespec/http"
+---
+
+Adds getStreamMetadata JS API to simplify getting stream metadata from operation parameters and responses.

packages/http/src/index.ts

@@ -21,5 +21,7 @@ export * from "./route.js";
 
 export * from "./types.js";
 
+export * from "./streams/index.js";
+
 /** @internal */
 export { $decorators } from "./tsp-index.js";

packages/http/src/streams/index.ts

@@ -1,2 +1,81 @@
-// This is a placeholder for the streams submodule.
-export {};
+import { Model, ModelProperty, Program, Type } from "@typespec/compiler";
+import { HttpOperationParameters, HttpOperationResponseContent } from "../types.js";
+let getStreamOf: typeof import("@typespec/streams").getStreamOf;
+try {
+  // eslint-disable-next-line @typescript-eslint/ban-ts-comment
+  // @ts-ignore
+  getStreamOf = (await import("@typespec/streams")).getStreamOf;
+} catch {
+  getStreamOf = () => {
+    throw new Error("@typespec/streams was not found");
+  };
+}
+
+export interface StreamMetadata {
+  /**
+   * The `Type` of the property decorated with `@body`.
+   */
+  bodyType: Type;
+  /**
+   * The `Type` of the stream model.
+   * For example, an instance of `HttpStream`.
+   */
+  originalType: Type;
+  /**
+   * The `Type` of the streaming payload.
+   *
+   * For example, given `HttpStream<Foo, "application/jsonl">`,
+   * the `streamType` would be `Foo`.
+   */
+  streamType: Type;
+  /**
+   * The list of content-types that this stream supports.
+   */
+  contentTypes: string[];
+}
+
+/**
+ * Gets stream metadata for a given `HttpOperationParameters` or `HttpOperationResponseContent`.
+ */
+export function getStreamMetadata(
+  program: Program,
+  httpParametersOrResponse: HttpOperationParameters | HttpOperationResponseContent,
+): StreamMetadata | undefined {
+  const body = httpParametersOrResponse.body;
+  if (!body) return;
+
+  const contentTypes = body.contentTypes;
+  if (!contentTypes.length) return;
+
+  // @body is always explicitly set by HttpStream, so body.property will be defined.
+  const bodyProperty = body.property;
+  if (!bodyProperty) return;
+
+  const streamData = getStreamFromBodyProperty(program, bodyProperty);
+  if (!streamData) return;
+
+  return {
+    bodyType: body.type,
+    originalType: streamData.model,
+    streamType: streamData.streamOf,
+    contentTypes: contentTypes,
+  };
+}
+
+function getStreamFromBodyProperty(
+  program: Program,
+  bodyProperty: ModelProperty,
+): { model: Model; streamOf: Type } | undefined {
+  // Check the model first, then if we can't find it, fallback to the sourceProperty model.
+  const streamOf = bodyProperty.model ? getStreamOf(program, bodyProperty.model) : undefined;
+
+  if (streamOf) {
+    // if `streamOf` is defined, then we know that `bodyProperty.model` is defined.
+    return { model: bodyProperty.model!, streamOf };
+  }
+
+  if (bodyProperty.sourceProperty) {
+    return getStreamFromBodyProperty(program, bodyProperty.sourceProperty);
+  }
+  return;
+}

packages/http/test/streams/get-stream-metadata.test.ts

@@ -0,0 +1,196 @@
+import { Model, Program } from "@typespec/compiler";
+import {
+  createTestHost,
+  createTestWrapper,
+  expectDiagnosticEmpty,
+  type BasicTestRunner,
+} from "@typespec/compiler/testing";
+import { StreamsTestLibrary } from "@typespec/streams/testing";
+import { assert, beforeEach, describe, expect, it } from "vitest";
+import { getStreamMetadata } from "../../src/index.js";
+import { getAllHttpServices } from "../../src/operations.js";
+import { HttpTestLibrary } from "../../src/testing/index.js";
+import { HttpService } from "../../src/types.js";
+
+let runner: BasicTestRunner;
+let getHttpServiceWithProgram: (
+  code: string,
+) => Promise<{ service: HttpService; Thing: Model; program: Program }>;
+
+beforeEach(async () => {
+  const host = await createTestHost({
+    libraries: [StreamsTestLibrary, HttpTestLibrary],
+  });
+  runner = createTestWrapper(host, {
+    autoImports: [`@typespec/http/streams`, "@typespec/streams"],
+    autoUsings: ["TypeSpec.Http", "TypeSpec.Http.Streams", "TypeSpec.Streams"],
+  });
+  getHttpServiceWithProgram = async (code) => {
+    const { Thing } = await runner.compile(`
+      @test model Thing { id: string }
+      ${code}
+    `);
+    assert(Thing.kind === "Model");
+    const [services, diagnostics] = getAllHttpServices(runner.program);
+
+    expectDiagnosticEmpty(diagnostics);
+    return { service: services[0], Thing, program: runner.program };
+  };
+});
+
+describe("Operation Responses", () => {
+  it("can get stream metadata from HttpStream", async () => {
+    const { service, Thing, program } = await getHttpServiceWithProgram(`
+      @route("/")
+      op get(): HttpStream<Thing, "application/jsonl", string>;
+    `);
+
+    const operation = service.operations[0];
+    const streamMetadata = getStreamMetadata(program, operation.responses[0].responses[0]);
+
+    expect(streamMetadata?.bodyType).toMatchObject({ kind: "Scalar", name: "string" });
+    expect(streamMetadata?.contentTypes).toEqual(["application/jsonl"]);
+    expect(streamMetadata?.originalType).toMatchObject({ name: "HttpStream" });
+    expect(streamMetadata?.streamType).toBe(Thing);
+  });
+
+  it("can get stream metadata from JsonlStream", async () => {
+    const { service, Thing, program } = await getHttpServiceWithProgram(`
+      @route("/")
+      op get(): JsonlStream<Thing>;
+    `);
+
+    const operation = service.operations[0];
+    const streamMetadata = getStreamMetadata(program, operation.responses[0].responses[0]);
+
+    expect(streamMetadata?.bodyType).toMatchObject({ kind: "Scalar", name: "string" });
+    expect(streamMetadata?.contentTypes).toEqual(["application/jsonl"]);
+    expect(streamMetadata?.originalType).toMatchObject({ name: "JsonlStream" });
+    expect(streamMetadata?.streamType).toBe(Thing);
+  });
+
+  it("can get stream metadata from custom base Stream", async () => {
+    const { service, Thing, program } = await getHttpServiceWithProgram(`
+      @streamOf(Thing)
+      model CustomStream {
+        @header contentType: "custom/built-here";
+        @body body: bytes;
+      }
+
+      @route("/")
+      op get(): CustomStream;
+    `);
+
+    const operation = service.operations[0];
+    const streamMetadata = getStreamMetadata(program, operation.responses[0].responses[0]);
+
+    expect(streamMetadata?.bodyType).toMatchObject({ kind: "Scalar", name: "bytes" });
+    expect(streamMetadata?.contentTypes).toEqual(["custom/built-here"]);
+    expect(streamMetadata?.originalType).toMatchObject({ name: "CustomStream" });
+    expect(streamMetadata?.streamType).toBe(Thing);
+  });
+
+  it("can get stream metadata from intersection with stream", async () => {
+    const { service, Thing, program } = await getHttpServiceWithProgram(`
+      @route("/")
+      op get(): JsonlStream<Thing> & { @statusCode statusCode: 204; };
+    `);
+
+    const operation = service.operations[0];
+    const streamMetadata = getStreamMetadata(program, operation.responses[0].responses[0]);
+
+    expect(streamMetadata?.bodyType).toMatchObject({ kind: "Scalar", name: "string" });
+    expect(streamMetadata?.contentTypes).toEqual(["application/jsonl"]);
+    expect(streamMetadata?.originalType).toMatchObject({ name: "JsonlStream" });
+    expect(streamMetadata?.streamType).toBe(Thing);
+  });
+
+  it("can get stream metadata from each HttpResponseContent", async () => {
+    const { service, Thing, program } = await getHttpServiceWithProgram(`
+      @route("/")
+      op get(): JsonlStream<Thing> | HttpStream<Thing, "custom/json", string>;
+    `);
+
+    const operation = service.operations[0];
+
+    const jsonlStreamMetadata = getStreamMetadata(program, operation.responses[0].responses[0]);
+    expect(jsonlStreamMetadata?.bodyType).toMatchObject({ kind: "Scalar", name: "string" });
+    expect(jsonlStreamMetadata?.contentTypes).toEqual(["application/jsonl"]);
+    expect(jsonlStreamMetadata?.originalType).toMatchObject({ name: "JsonlStream" });
+    expect(jsonlStreamMetadata?.streamType).toBe(Thing);
+
+    const httpStreamMetadata = getStreamMetadata(program, operation.responses[0].responses[1]);
+    expect(httpStreamMetadata?.bodyType).toMatchObject({ kind: "Scalar", name: "string" });
+    expect(httpStreamMetadata?.contentTypes).toEqual(["custom/json"]);
+    expect(httpStreamMetadata?.originalType).toMatchObject({ name: "HttpStream" });
+    expect(httpStreamMetadata?.streamType).toBe(Thing);
+  });
+});
+
+describe("Operation Parameters", () => {
+  it("can get stream metadata from HttpStream", async () => {
+    const { service, Thing, program } = await getHttpServiceWithProgram(`
+      @route("/")
+      op get(stream: HttpStream<Thing, "application/jsonl", string>): void;
+    `);
+
+    const operation = service.operations[0];
+    const streamMetadata = getStreamMetadata(program, operation.parameters);
+
+    expect(streamMetadata?.bodyType).toMatchObject({ kind: "Scalar", name: "string" });
+    expect(streamMetadata?.contentTypes).toEqual(["application/jsonl"]);
+    expect(streamMetadata?.originalType).toMatchObject({ name: "HttpStream" });
+    expect(streamMetadata?.streamType).toBe(Thing);
+  });
+
+  it("can get stream metadata from JsonlStream", async () => {
+    const { service, Thing, program } = await getHttpServiceWithProgram(`
+      @route("/")
+      op get(stream: JsonlStream<Thing>): void;
+    `);
+
+    const operation = service.operations[0];
+    const streamMetadata = getStreamMetadata(program, operation.parameters);
+
+    expect(streamMetadata?.bodyType).toMatchObject({ kind: "Scalar", name: "string" });
+    expect(streamMetadata?.contentTypes).toEqual(["application/jsonl"]);
+    expect(streamMetadata?.originalType).toMatchObject({ name: "JsonlStream" });
+    expect(streamMetadata?.streamType).toBe(Thing);
+  });
+
+  it("can get stream metadata from custom base Stream", async () => {
+    const { service, Thing, program } = await getHttpServiceWithProgram(`
+      @streamOf(Thing)
+      model CustomStream {
+        @header contentType: "custom/built-here";
+        @body body: bytes;
+      }
+
+      @route("/")
+      op get(stream: CustomStream): void;
+    `);
+
+    const operation = service.operations[0];
+    const streamMetadata = getStreamMetadata(program, operation.parameters);
+
+    expect(streamMetadata?.bodyType).toMatchObject({ kind: "Scalar", name: "bytes" });
+    expect(streamMetadata?.contentTypes).toEqual(["custom/built-here"]);
+    expect(streamMetadata?.originalType).toMatchObject({ name: "CustomStream" });
+    expect(streamMetadata?.streamType).toBe(Thing);
+  });
+
+  it("can get stream metadata from spread parameters", async () => {
+    const { service, Thing, program } = await getHttpServiceWithProgram(`
+      @route("/")
+      op get(...JsonlStream<Thing>): void;
+    `);
+
+    const operation = service.operations[0];
+    const streamMetadata = getStreamMetadata(program, operation.parameters);
+
+    expect(streamMetadata?.bodyType).toMatchObject({ kind: "Scalar", name: "string" });
+    expect(streamMetadata?.contentTypes).toEqual(["application/jsonl"]);
+    expect(streamMetadata?.originalType).toMatchObject({ name: "JsonlStream" });
+    expect(streamMetadata?.streamType).toBe(Thing);
+  });
+});