codeant-cli 0.1.0

package/src/utils/secretsApiHelper.js

@@ -0,0 +1,127 @@
+import GitDiffHelper from './gitDiffHelper.js';
+
+/**
+ * Transforms git diff data into the format expected by the secrets detection API
+ *
+ * API Input Format:
+ * {
+ *   "files": [
+ *     {
+ *       "file_path": str,
+ *       "code": str,
+ *       "diffs": [{ "start_line": int, "end_line": int }]
+ *     }
+ *   ]
+ * }
+ */
+class SecretsApiHelper {
+  constructor(workspacePath) {
+    this.workspacePath = workspacePath;
+    this.gitHelper = new GitDiffHelper(workspacePath);
+  }
+
+  async init() {
+    await this.gitHelper.init();
+  }
+
+  /**
+   * Get staged files formatted for the secrets API
+   * This is used for pre-commit hooks
+   */
+  async getStagedFilesForApi() {
+    const diffs = await this.gitHelper.getDiffBasedOnReviewConfig({ type: 'staged-only' });
+    return this._transformDiffsToApiFormat(diffs);
+  }
+
+  /**
+   * Get all changed files formatted for the secrets API
+   */
+  async getChangedFilesForApi() {
+    const diffs = await this.gitHelper.getDiffBasedOnReviewConfig({ type: 'branch-diff' });
+    return this._transformDiffsToApiFormat(diffs);
+  }
+
+  /**
+   * Get uncommitted files formatted for the secrets API
+   */
+  async getUncommittedFilesForApi() {
+    const diffs = await this.gitHelper.getDiffBasedOnReviewConfig({ type: 'uncommitted' });
+    return this._transformDiffsToApiFormat(diffs);
+  }
+
+  /**
+   * Get last commit files formatted for the secrets API
+   */
+  async getLastCommitFilesForApi() {
+    const diffs = await this.gitHelper.getDiffBasedOnReviewConfig({ type: 'last-commit' });
+    return this._transformDiffsToApiFormat(diffs);
+  }
+
+  /**
+   * Transform diff info array to API format
+   * Groups by file and extracts diff ranges
+   */
+  _transformDiffsToApiFormat(diffs) {
+    // Group diffs by filename
+    const fileMap = new Map();
+
+    for (const diff of diffs) {
+      const filePath = diff.filename_str;
+
+      if (!fileMap.has(filePath)) {
+        fileMap.set(filePath, {
+          file_path: filePath,
+          code: diff.head_file_str || '',
+          diffs: []
+        });
+      }
+
+      // Add diff range if available
+      if (diff.start_line_str && diff.end_line_str) {
+        const startLine = parseInt(diff.start_line_str, 10);
+        const endLine = parseInt(diff.end_line_str, 10);
+
+        if (!isNaN(startLine) && !isNaN(endLine)) {
+          fileMap.get(filePath).diffs.push({
+            start_line: startLine,
+            end_line: endLine
+          });
+        }
+      }
+    }
+
+    return Array.from(fileMap.values());
+  }
+
+  /**
+   * Build the complete request body for the secrets API
+   */
+  async buildSecretsApiRequest(type = 'staged-only') {
+    let files;
+
+    switch (type) {
+      case 'staged-only':
+        files = await this.getStagedFilesForApi();
+        break;
+      case 'branch-diff':
+        files = await this.getChangedFilesForApi();
+        break;
+      case 'uncommitted':
+        files = await this.getUncommittedFilesForApi();
+        break;
+      case 'last-commit':
+        files = await this.getLastCommitFilesForApi();
+        break;
+      default:
+        files = await this.getStagedFilesForApi();
+    }
+
+    return { files };
+  }
+
+  getGitRoot() {
+    return this.gitHelper.getGitRoot();
+  }
+}
+
+export default SecretsApiHelper;