@uniqbit/mate-core 0.18.5 → 1.0.0-canary.0

package/src/tools/setup/plugins/guidance.ts

@@ -2,6 +2,8 @@ import fs from "node:fs/promises";
 import path from "node:path";
 
 import { FRAMEWORK_NAME } from "../../../framework";
+import { companionBlockMarkers } from "../../../runtime/managed-block";
+import { writeAtomicThroughLink } from "../../../runtime/write-atomic";
 import { removeGraphifySection } from "../capabilities/graphify-shared";
 
 // Marker names derive from the framework identity, never the invocation name.
@@ -9,7 +11,7 @@ const upperName = () => FRAMEWORK_NAME.toUpperCase();
 
 function allMateStarts(): string[] {
   return [
-    `<!-- ${upperName()}:COMPANION:START -->`,
+    companionBlockMarkers(FRAMEWORK_NAME).start,
     "<!-- COMPANION:GUIDANCE:START -->",
     `<!-- ${upperName()}:COMPANION:GUIDANCE:START -->`,
     `<!-- ${upperName()}:GRAPHIFY:START -->`,
@@ -18,7 +20,7 @@ function allMateStarts(): string[] {
 
 function allMateEnds(): string[] {
   return [
-    `<!-- ${upperName()}:COMPANION:END -->`,
+    companionBlockMarkers(FRAMEWORK_NAME).end,
     "<!-- COMPANION:GUIDANCE:END -->",
     `<!-- ${upperName()}:COMPANION:GUIDANCE:END -->`,
     `<!-- ${upperName()}:GRAPHIFY:END -->`,
@@ -57,7 +59,7 @@ export async function stripGuidanceBlock(filePath: string): Promise<void> {
 
   content = removeGraphifySection(content);
 
-  await fs.writeFile(filePath, content ? content + "\n" : "", "utf8");
+  await writeAtomicThroughLink(filePath, content ? content + "\n" : "");
 }
 
 /**
@@ -127,5 +129,6 @@ export async function refreshFromTemplate(
   if (normalizedOptions.ensureParentDir) {
     await fs.mkdir(path.dirname(filePath), { recursive: true });
   }
-  await fs.writeFile(filePath, result, "utf8");
+  /** A session-start hook may read the file while another session's prepare rewrites it. */
+  await writeAtomicThroughLink(filePath, result);
 }

package/src/tools/setup/providers/agent-file-sections.ts

@@ -3,6 +3,8 @@
 
 import fs from "node:fs/promises";
 
+import { removeEmptiedFile, writeAtomicThroughLink } from "../../../runtime/write-atomic";
+
 export interface RemoveHeadingSectionOptions {
   /** Matches the heading line that opens the section to remove. */
   isHeading(line: string): boolean;
@@ -71,8 +73,8 @@ export async function stripSectionFromFile(
   const stripped = removeHeadingSection(content, options);
   if (stripped === content) return;
   if (!stripped.trim()) {
-    await fs.unlink(filePath);
+    await removeEmptiedFile(filePath);
   } else {
-    await fs.writeFile(filePath, stripped, "utf8");
+    await writeAtomicThroughLink(filePath, stripped);
   }
 }

package/src/tools/setup/providers/claude-attribution.ts

@@ -0,0 +1,65 @@
+import fs from "node:fs";
+import os from "node:os";
+import path from "node:path";
+import { isDeepStrictEqual } from "node:util";
+
+/**
+ * Hides Claude Code's commit and PR attribution, so a session is never told to
+ * add a model `Co-Authored-By` trailer. The object form, never `false`: Claude
+ * Code before v2.1.281 rejects `false` and drops the whole settings file, and
+ * Mate's permissions and guard hooks with it.
+ */
+export const MANAGED_CLAUDE_ATTRIBUTION = Object.freeze({
+  commit: "",
+  pr: "",
+  sessionUrl: false,
+});
+
+/** Claude Code's user config directory: `CLAUDE_CONFIG_DIR`, else `~/.claude`. */
+export function claudeUserConfigDir(env: Record<string, string | undefined> = process.env): string {
+  const configured = env.CLAUDE_CONFIG_DIR;
+  if (configured) return path.normalize(configured);
+  return path.join(env.HOME || os.homedir(), ".claude");
+}
+
+export interface ClaudeAttributionScopes {
+  /** Defaults to {@link claudeUserConfigDir}. */
+  userConfigDir?: string;
+  /** The Working Repository, when known: adds its project and local settings. */
+  repoRoot?: string;
+}
+
+function readSettings(filePath: string): Record<string, unknown> | null {
+  try {
+    const parsed: unknown = JSON.parse(fs.readFileSync(filePath, "utf8"));
+    return parsed && typeof parsed === "object" ? (parsed as Record<string, unknown>) : null;
+  } catch {
+    return null;
+  }
+}
+
+/** Whether `value` is Mate's own attribution rather than one the user set. */
+export function isManagedClaudeAttribution(value: unknown): boolean {
+  return isDeepStrictEqual(value, MANAGED_CLAUDE_ATTRIBUTION);
+}
+
+/**
+ * Whether a checked scope sets attribution itself, so Mate's value must yield.
+ * Mate's own wrapped value sits in the same `settings.local.json` the user
+ * edits; deep-equality with it is what tells the two apart.
+ */
+export function definesUserClaudeAttribution(scopes: ClaudeAttributionScopes): boolean {
+  const files = [path.join(scopes.userConfigDir ?? claudeUserConfigDir(), "settings.json")];
+  if (scopes.repoRoot) {
+    files.push(
+      path.join(scopes.repoRoot, ".claude", "settings.json"),
+      path.join(scopes.repoRoot, ".claude", "settings.local.json"),
+    );
+  }
+  return files.some((file) => {
+    const settings = readSettings(file);
+    if (!settings) return false;
+    if ("includeCoAuthoredBy" in settings) return true;
+    return "attribution" in settings && !isManagedClaudeAttribution(settings.attribution);
+  });
+}