name: Profiling Comparison for Specific File 2

on:
  push
jobs:
  profiler:
    runs-on: ubuntu-latest

    steps:
    # Checkout the code from the repository with full history
    - name: Checkout code
      uses: actions/checkout@v3
      with:
        fetch-depth: 0  # Fetch all history so we can checkout any commit

    # Set up Python environment
    - name: Set up Python
      uses: actions/setup-python@v4
      with:
        python-version: '3.10'

    - name: Install Poetry
      uses: snok/install-poetry@v1.3.2
      with:
        virtualenvs-create: true
        virtualenvs-in-project: true
        installer-parallel: true

    - name: Install dependencies
      run: |
        poetry install --no-interaction --all-extras
        poetry add scalene requests


    # Set environment variables for SHAs
    - name: Set environment variables
      run: |
        echo "BASE_SHA=${{ github.event.pull_request.base.sha }}" >> $GITHUB_ENV
        echo "HEAD_SHA=${{ github.event.pull_request.head.sha }}" >> $GITHUB_ENV

    # Run profiler on the base branch
    - name: Run profiler on base branch
      env:
        BASE_SHA: ${{ env.BASE_SHA }}
      run: |
        echo "Profiling the base branch for code_graph_pipeline.py"
        echo "Checking out base SHA: $BASE_SHA"
        git checkout $BASE_SHA
        echo "This is the working directory: $PWD"
        # Ensure the script is executable
        chmod +x cognee/api/v1/cognify/code_graph_pipeline.py
        # Run Scalene
        poetry run scalene --json --outfile base_results.json cognee/api/v1/cognify/code_graph_pipeline.py

    # Run profiler on head branch
    - name: Run profiler on head branch
      env:
        HEAD_SHA: ${{ env.HEAD_SHA }}
      run: |
        echo "Profiling the head branch for code_graph_pipeline.py"
        echo "Checking out head SHA: $HEAD_SHA"
        git checkout $HEAD_SHA
        echo "This is the working directory: $PWD"
        # Ensure the script is executable
        chmod +x cognee/api/v1/cognify/code_graph_pipeline.py
        # Run Scalene
        poetry run scalene --json --outfile head_results.json cognee/api/v1/cognify/code_graph_pipeline.py

    # Compare profiling results
    - name: Compare profiling results
      run: |
        python -c '
        import json
        try:
            with open("base_results.json") as f:
                base = json.load(f)
            with open("head_results.json") as f:
                head = json.load(f)
            cpu_diff = head.get("total_cpu_samples_python", 0) - base.get("total_cpu_samples_python", 0)
            memory_diff = head.get("malloc_samples", 0) - base.get("malloc_samples", 0)
            with open("profiling_diff.txt", "w") as f:
                f.write(f"CPU Usage Difference: {cpu_diff}\\n")
                f.write(f"Memory Usage Difference: {memory_diff} bytes\\n")
        except Exception as e:
            with open("profiling_diff.txt", "w") as f:
                f.write(f"Error comparing profiling results: {e}\\n")
        '

    # Post results to the pull request
    - name: Post profiling results to PR
      uses: actions/github-script@v6
      with:
        script: |
          const fs = require('fs');
          const diff = fs.readFileSync('profiling_diff.txt', 'utf-8');
          github.rest.issues.createComment({
            issue_number: context.issue.number,
            owner: context.repo.owner,
            repo: context.repo.repo,
            body: `### Profiling Results for code_graph_pipeline.py\n\`\`\`\n${diff || 'No differences found.'}\n\`\`\``
          });