flake.nix

{
  description = "Brownyx - Browser GPU/CPU Monitoring & Benchmarking";

  inputs = {
    nixpkgs.url = "github:NixOS/nixpkgs/nixos-unstable";
    flake-utils.url = "github:numtide/flake-utils";
  };

  outputs = { self, nixpkgs, flake-utils }:
    flake-utils.lib.eachDefaultSystem (system:
      let
        pkgs = import nixpkgs {
          inherit system;
          config.allowUnfree = true;
        };

        pythonEnv = pkgs.python3.withPackages (ps: with ps; [
          psutil
          influxdb-client
        ]);

        # Script para iniciar InfluxDB localmente
        startInfluxDB = pkgs.writeShellScriptBin "start-influxdb" ''
          export INFLUXD_CONFIG_PATH="$PWD/.nix-influxdb/config.yml"
          export INFLUXD_BOLT_PATH="$PWD/.nix-influxdb/influxd.bolt"
          export INFLUXD_ENGINE_PATH="$PWD/.nix-influxdb/engine"

          mkdir -p .nix-influxdb

          if [ ! -f "$INFLUXD_BOLT_PATH" ]; then
            echo "Setting up InfluxDB for the first time..."
            ${pkgs.influxdb2}/bin/influxd --bolt-path="$INFLUXD_BOLT_PATH" --engine-path="$INFLUXD_ENGINE_PATH" &
            INFLUXD_PID=$!
            sleep 5

            # Setup inicial
            ${pkgs.influxdb2}/bin/influx setup \
              --username admin \
              --password adminpassword \
              --org brownyx \
              --bucket browser-metrics \
              --token brownyx-super-secret-token \
              --force \
              --host http://localhost:8086

            kill $INFLUXD_PID
            wait $INFLUXD_PID 2>/dev/null
          fi

          echo "Starting InfluxDB on http://localhost:8086..."
          exec ${pkgs.influxdb2}/bin/influxd \
            --bolt-path="$INFLUXD_BOLT_PATH" \
            --engine-path="$INFLUXD_ENGINE_PATH"
        '';

        # Script para parar InfluxDB
        stopInfluxDB = pkgs.writeShellScriptBin "stop-influxdb" ''
          pkill -f influxd || echo "InfluxDB not running"
        '';

        # Script para iniciar o backend
        startBackend = pkgs.writeShellScriptBin "start-backend" ''
          echo "Installing Node.js dependencies..."
          npm install
          echo "Starting backend server..."
          npm run dev
        '';

        # Script para iniciar o frontend
        startFrontend = pkgs.writeShellScriptBin "start-frontend" ''
          cd frontend
          echo "Installing frontend dependencies..."
          npm install
          echo "Starting frontend server..."
          npm run dev
        '';

        # Script para instalar browsers do Playwright
        installBrowsers = pkgs.writeShellScriptBin "install-browsers" ''
          echo "Installing Playwright browsers..."
          npx playwright install --with-deps chromium firefox webkit
        '';

        # Script para testar o coletor de métricas
        testCollector = pkgs.writeShellScriptBin "test-collector" ''
          echo "Testing metrics collector..."
          ${pythonEnv}/bin/python3 collector/metrics_collector.py
        '';

        # Script all-in-one para iniciar tudo
        startAll = pkgs.writeShellScriptBin "start-all" ''
          set -e

          echo "🚀 Starting Brownyx Development Environment"
          echo ""

          # Check if InfluxDB is already running
          if pgrep -f influxd > /dev/null; then
            echo "✓ InfluxDB is already running"
          else
            echo "⚙ Starting InfluxDB..."
            start-influxdb &
            sleep 5
          fi

          echo ""
          echo "📊 Services will be available at:"
          echo "   - InfluxDB: http://localhost:8086"
          echo "   - Grafana: http://localhost:3002 (configure externally)"
          echo "   - API: http://localhost:3000"
          echo "   - Frontend: http://localhost:3003"
          echo ""
          echo "🔧 To start backend and frontend, run in separate terminals:"
          echo "   Terminal 1: start-backend"
          echo "   Terminal 2: start-frontend"
          echo ""
          echo "Press Ctrl+C to stop InfluxDB"

          wait
        '';

        # Script para limpar dados do InfluxDB
        cleanInfluxDB = pkgs.writeShellScriptBin "clean-influxdb" ''
          echo "Stopping InfluxDB..."
          stop-influxdb
          echo "Removing InfluxDB data..."
          rm -rf .nix-influxdb
          echo "InfluxDB data cleaned. Run 'start-influxdb' to reinitialize."
        '';

      in
      {
        devShells.default = pkgs.mkShell {
          buildInputs = with pkgs; [
            # Node.js e gerenciadores de pacotes
            nodejs_20
            nodePackages.npm
            nodePackages.typescript
            nodePackages.typescript-language-server

            # Python e pacotes
            pythonEnv

            # InfluxDB
            influxdb2
            influxdb2-cli

            # Ferramentas de sistema para métricas de GPU
            pciutils
            lshw

            # Para NVIDIA GPUs (opcional, se tiver)
            # linuxPackages.nvidia_x11

            # Playwright dependencies
            playwright-driver.browsers

            # Ferramentas de desenvolvimento
            git
            curl
            jq

            # Scripts customizados
            startInfluxDB
            stopInfluxDB
            startBackend
            startFrontend
            installBrowsers
            testCollector
            startAll
            cleanInfluxDB
          ];

          shellHook = ''
            echo "🎨 Brownyx Development Environment"
            echo "=================================="
            echo ""
            echo "Available commands:"
            echo "  start-all          - Start InfluxDB (all-in-one)"
            echo "  start-influxdb     - Start InfluxDB manually"
            echo "  stop-influxdb      - Stop InfluxDB"
            echo "  start-backend      - Install deps and start backend"
            echo "  start-frontend     - Install deps and start frontend"
            echo "  install-browsers   - Install Playwright browsers"
            echo "  test-collector     - Test Python metrics collector"
            echo "  clean-influxdb     - Clean InfluxDB data"
            echo ""
            echo "Quick Start:"
            echo "  1. Run: start-all (in one terminal)"
            echo "  2. Run: start-backend (in another terminal)"
            echo "  3. Run: start-frontend (in another terminal)"
            echo ""
            echo "InfluxDB Credentials:"
            echo "  URL: http://localhost:8086"
            echo "  Username: admin"
            echo "  Password: adminpassword"
            echo "  Token: brownyx-super-secret-token"
            echo "  Org: brownyx"
            echo "  Bucket: browser-metrics"
            echo ""

            # Set environment variables
            export INFLUXDB_URL="http://localhost:8086"
            export INFLUXDB_TOKEN="brownyx-super-secret-token"
            export INFLUXDB_ORG="brownyx"
            export INFLUXDB_BUCKET="browser-metrics"
            export API_PORT="3000"
            export WS_PORT="3001"

            # Create .env if it doesn't exist
            if [ ! -f .env ]; then
              cp .env.example .env 2>/dev/null || true
            fi

            # Playwright configuration
            export PLAYWRIGHT_BROWSERS_PATH="${pkgs.playwright-driver.browsers}"
            export PLAYWRIGHT_SKIP_VALIDATE_HOST_REQUIREMENTS=true

            echo "Environment variables set ✓"
            echo ""
          '';
        };

        # Também disponibiliza os scripts como apps
        apps = {
          start-influxdb = {
            type = "app";
            program = "${startInfluxDB}/bin/start-influxdb";
          };
          start-backend = {
            type = "app";
            program = "${startBackend}/bin/start-backend";
          };
          start-frontend = {
            type = "app";
            program = "${startFrontend}/bin/start-frontend";
          };
        };
      }
    );
}