marmot-logger 1.0.0

package/src/plugins/makefile.js

@@ -0,0 +1,68 @@
+const fs = require('fs');
+const path = require('path');
+const chalk = require('chalk');
+
+const MARMOT_MARKER = '# marmot';
+
+const MAKEFILE_SNIPPET = `# Marmot: Log make commands ${MARMOT_MARKER}
+MARMOT_LOG_DIR := logs ${MARMOT_MARKER}
+MARMOT_LOG_FILE := $(MARMOT_LOG_DIR)/file_events_$(shell date +%Y-%m-%d).log ${MARMOT_MARKER}
+$(shell mkdir -p $(MARMOT_LOG_DIR)) ${MARMOT_MARKER}
+$(shell marmot log make -d '{"target": "$(MAKECMDGOALS)"}' 2>/dev/null &) ${MARMOT_MARKER}`;
+
+async function enable(projectConfig) {
+  const projectDir = process.cwd();
+  const makefilePath = path.join(projectDir, 'Makefile');
+
+  console.log('');
+  console.log(chalk.bold('Makefile Plugin Setup:'));
+
+  if (fs.existsSync(makefilePath)) {
+    const content = fs.readFileSync(makefilePath, 'utf8');
+    if (content.includes(MARMOT_MARKER)) {
+      console.log(chalk.yellow('Makefile already contains Marmot logging.'));
+      return;
+    }
+  }
+
+  console.log('Add the following to the top of your Makefile (after any initial comments):');
+  console.log('');
+  console.log(chalk.cyan(MAKEFILE_SNIPPET));
+  console.log('');
+}
+
+async function disable(projectConfig) {
+  const projectDir = process.cwd();
+  const makefilePath = path.join(projectDir, 'Makefile');
+
+  console.log('');
+  console.log(chalk.bold('Makefile Plugin Disabled:'));
+
+  if (!fs.existsSync(makefilePath)) {
+    console.log('No Makefile found.');
+    return;
+  }
+
+  const content = fs.readFileSync(makefilePath, 'utf8');
+  const lines = content.split('\n');
+
+  // Filter out lines ending with # marmot marker
+  const filteredLines = lines.filter(line => !line.trimEnd().endsWith(MARMOT_MARKER));
+
+  // Clean up multiple consecutive blank lines
+  const cleanedContent = filteredLines.join('\n')
+    .replace(/\n{3,}/g, '\n\n')
+    .replace(/^\n+/, '');
+
+  if (content !== cleanedContent) {
+    fs.writeFileSync(makefilePath, cleanedContent);
+    console.log(chalk.green('Removed marmot logging lines from Makefile.'));
+  } else {
+    console.log('No marmot lines found in Makefile.');
+  }
+}
+
+module.exports = {
+  enable,
+  disable
+};

package/src/plugins/terminal.js

@@ -0,0 +1,136 @@
+const fs = require('fs');
+const path = require('path');
+const os = require('os');
+const chalk = require('chalk');
+const config = require('../core/config');
+
+const MARMOT_BASHRC_MARKER = '# marmot-terminal';
+
+function getBashrcPath() {
+  return path.join(os.homedir(), '.bashrc');
+}
+
+async function enable(projectConfig) {
+  const projectDir = process.cwd();
+  const marmotDir = config.getMarmotDir(projectDir);
+  const hookFile = path.join(marmotDir, 'terminal-hook.sh');
+
+  // Marmot dir should already exist from init, but ensure it does
+  if (!fs.existsSync(marmotDir)) {
+    fs.mkdirSync(marmotDir, { recursive: true });
+  }
+
+  // Generate bashrc hook script
+  const script = `#!/bin/bash
+# Marmot Terminal Command Logging
+# Source this file in your ~/.bashrc to log terminal commands
+# Generated by: marmot enable terminal
+
+MARMOT_PROJECT_DIR="${projectDir}"
+
+if [[ -d "$MARMOT_PROJECT_DIR" ]]; then
+  __marmot_last_cmd=""
+
+  __marmot_log_command() {
+    local cmd=$(history 1 | sed 's/^ *[0-9]* *//')
+
+    # Skip if same as last command
+    [[ "$cmd" == "$__marmot_last_cmd" ]] && return
+
+    # Skip if empty
+    [[ -z "$cmd" ]] && return
+
+    # Only log if in project directory
+    if [[ "$PWD/" == "$MARMOT_PROJECT_DIR/"* || "$PWD" == "$MARMOT_PROJECT_DIR" ]]; then
+      (cd "$MARMOT_PROJECT_DIR" && marmot log terminal -d "{\\"command\\": \\"$cmd\\", \\"path\\": \\"$PWD\\"}" 2>/dev/null &) 2>/dev/null
+      __marmot_last_cmd="$cmd"
+    fi
+  }
+
+  # Append to PROMPT_COMMAND
+  if [[ -z "$PROMPT_COMMAND" ]]; then
+    PROMPT_COMMAND="__marmot_log_command"
+  elif [[ "$PROMPT_COMMAND" != *"__marmot_log_command"* ]]; then
+    PROMPT_COMMAND="__marmot_log_command; $PROMPT_COMMAND"
+  fi
+fi
+`;
+
+  fs.writeFileSync(hookFile, script);
+  fs.chmodSync(hookFile, '755');
+
+  // Add source line to ~/.bashrc
+  const bashrcPath = getBashrcPath();
+  const sourceLine = `source "${hookFile}" ${MARMOT_BASHRC_MARKER}`;
+
+  let bashrcUpdated = false;
+  if (fs.existsSync(bashrcPath)) {
+    const bashrcContent = fs.readFileSync(bashrcPath, 'utf8');
+    if (!bashrcContent.includes(hookFile)) {
+      fs.appendFileSync(bashrcPath, `\n${sourceLine}\n`);
+      bashrcUpdated = true;
+    }
+  } else {
+    fs.writeFileSync(bashrcPath, `${sourceLine}\n`);
+    bashrcUpdated = true;
+  }
+
+  console.log('');
+  console.log(chalk.bold('Terminal Plugin Setup:'));
+  console.log(`  Hook script: ${chalk.cyan(hookFile)}`);
+
+  if (bashrcUpdated) {
+    console.log(`  Added to: ${chalk.cyan(bashrcPath)}`);
+    console.log('');
+    console.log('Reload your shell to activate:');
+    console.log(chalk.cyan('  source ~/.bashrc'));
+  } else {
+    console.log(`  Already in: ${chalk.cyan(bashrcPath)}`);
+  }
+}
+
+async function disable(projectConfig) {
+  const projectDir = process.cwd();
+  const marmotDir = config.getMarmotDir(projectDir);
+  const hookFile = path.join(marmotDir, 'terminal-hook.sh');
+  const bashrcPath = getBashrcPath();
+
+  console.log('');
+  console.log(chalk.bold('Terminal Plugin Disabled:'));
+
+  // Remove source line from ~/.bashrc
+  if (fs.existsSync(bashrcPath)) {
+    const bashrcContent = fs.readFileSync(bashrcPath, 'utf8');
+    const lines = bashrcContent.split('\n');
+
+    // Remove lines that source this hook file or have marmot marker
+    const filteredLines = lines.filter(line => {
+      const trimmed = line.trim();
+      if (trimmed.includes(hookFile)) return false;
+      if (trimmed.endsWith(MARMOT_BASHRC_MARKER)) return false;
+      return true;
+    });
+
+    const newContent = filteredLines.join('\n').replace(/\n{3,}/g, '\n\n');
+
+    if (newContent !== bashrcContent) {
+      fs.writeFileSync(bashrcPath, newContent);
+      console.log(`  Removed from: ${chalk.cyan(bashrcPath)}`);
+    }
+  }
+
+  // Remove hook script file
+  if (fs.existsSync(hookFile)) {
+    fs.unlinkSync(hookFile);
+    console.log(`  Removed: ${chalk.cyan(hookFile)}`);
+  }
+
+  console.log('');
+  console.log('Reload your shell:');
+  console.log(chalk.cyan('  source ~/.bashrc'));
+}
+
+module.exports = {
+  enable,
+  disable
+};