typespec-rust-emitter 0.1.0 → 0.3.0

package/src/index.ts

@@ -1,3 +1,3 @@
 export { $onEmit } from "./emitter.js";
-export { $rustDerive, $rustDerives } from "./emitter.js";
+export { $rustDerive, $rustDerives, $rustAttr, $rustAttrs } from "./emitter.js";
 export { $lib } from "./lib.js";

package/src/lib.tsp

@@ -2,5 +2,7 @@ import "../dist/src/emitter.js";
 
 using TypeSpec.Reflection;
 
-extern dec rustDerive(target: Model, derive: valueof string);
-extern dec rustDerives(target: Model, ...derives: valueof string[]);
+extern dec rustDerive(target: Model | Enum, derive: valueof string);
+extern dec rustDerives(target: Model | Enum, ...derives: valueof string[]);
+extern dec rustAttr(target: Model | Enum, attr: valueof string);
+extern dec rustAttrs(target: Model | Enum, ...attrs: valueof string[]);

package/test/hello.test.ts

@@ -165,4 +165,104 @@ describe("Rust emitter", () => {
       true,
     );
   });
+
+  it("emits custom rustDerive on enum", async () => {
+    const results = await emit(`
+      import "typespec-rust-emitter";
+
+      @rustDerive("strum::Display")
+      enum Status {
+        active,
+        inactive,
+      }
+    `);
+    const output = results["types.rs"];
+    strictEqual(
+      output.includes(
+        "#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, serde::Serialize, serde::Deserialize, strum::Display)]",
+      ),
+      true,
+    );
+  });
+
+  it("emits multiple rustDerive macros on enum via $rustDerives", async () => {
+    const results = await emit(`
+      import "typespec-rust-emitter";
+
+      @rustDerives("strum::Display", "strum::EnumString")
+      enum Priority {
+        low,
+        high,
+      }
+    `);
+    const output = results["types.rs"];
+    strictEqual(
+      output.includes(
+        "#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash, serde::Serialize, serde::Deserialize, strum::Display, strum::EnumString)]",
+      ),
+      true,
+    );
+  });
+
+  it("emits custom rustAttr on model", async () => {
+    const results = await emit(`
+      import "typespec-rust-emitter";
+
+      @rustAttr("sqlx(type_name = \\"user\\")")
+      model User {
+        name: string;
+      }
+    `);
+    const output = results["types.rs"];
+    strictEqual(output.includes('#[sqlx(type_name = "user")]'), true);
+  });
+
+  it("emits multiple rustAttrs on model via $rustAttrs", async () => {
+    const results = await emit(`
+      import "typespec-rust-emitter";
+
+      @rustAttrs("sqlx(type_name = \\"user\\")", "cfg_attr(feature = \\"test\\", derive(Debug))")
+      model Person {
+        name: string;
+      }
+    `);
+    const output = results["types.rs"];
+    strictEqual(output.includes('#[sqlx(type_name = "user")]'), true);
+    strictEqual(
+      output.includes('#[cfg_attr(feature = "test", derive(Debug))]'),
+      true,
+    );
+  });
+
+  it("emits custom rustAttr on enum", async () => {
+    const results = await emit(`
+      import "typespec-rust-emitter";
+
+      @rustAttr("sqlx(type_name = \\"study_status\\")")
+      enum Status {
+        active,
+        inactive,
+      }
+    `);
+    const output = results["types.rs"];
+    strictEqual(output.includes('#[sqlx(type_name = "study_status")]'), true);
+  });
+
+  it("emits multiple rustAttrs on enum via $rustAttrs", async () => {
+    const results = await emit(`
+      import "typespec-rust-emitter";
+
+      @rustAttrs("sqlx(type_name = \\"priority\\")", "cfg_attr(feature = \\"test\\", derive(Debug))")
+      enum Priority {
+        low,
+        high,
+      }
+    `);
+    const output = results["types.rs"];
+    strictEqual(output.includes('#[sqlx(type_name = "priority")]'), true);
+    strictEqual(
+      output.includes('#[cfg_attr(feature = "test", derive(Debug))]'),
+      true,
+    );
+  });
 });