typespec-rust-emitter 0.12.0 → 0.13.1

package/src/decorators/rust_self.ts

@@ -0,0 +1,42 @@
+import {
+  DecoratorContext,
+  getNamespaceFullName,
+  Type,
+} from "@typespec/compiler";
+import { rustSelfReceiverKey } from "../models/keys.js";
+
+export function $rustMut(context: DecoratorContext, target: Type) {
+  if (target.kind !== "Operation") {
+    context.program.reportDiagnostic({
+      code: "rust-mut-invalid-target",
+      message: `@rustMut can only be applied to operations`,
+      severity: "error",
+      target: context.decoratorTarget,
+    });
+    return;
+  }
+
+  const ns = target.namespace ? getNamespaceFullName(target.namespace) : "";
+  if (!ns.startsWith("TypeSpec")) {
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    (target as any)[rustSelfReceiverKey] = "&mut self";
+  }
+}
+
+export function $rustOwn(context: DecoratorContext, target: Type) {
+  if (target.kind !== "Operation") {
+    context.program.reportDiagnostic({
+      code: "rust-own-invalid-target",
+      message: `@rustOwn can only be applied to operations`,
+      severity: "error",
+      target: context.decoratorTarget,
+    });
+    return;
+  }
+
+  const ns = target.namespace ? getNamespaceFullName(target.namespace) : "";
+  if (!ns.startsWith("TypeSpec")) {
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    (target as any)[rustSelfReceiverKey] = "self";
+  }
+}