GenDev 智能开发：Amazon Q Developer CLI 赋能Amazon Code Family实现代码审核

version: 0.2

phases:
  install:
    commands:
      - echo Installing prerequisites...
      - aws s3 sync s3://<请填入您的s3文件夹路径> ~/.local/share/amazon-q/
      - echo Amazon Q credentials synced.

  pre_build:
    commands:
      - echo Initializing environment
      - echo "=== Getting PR Information from EventBridge ==="
      - |
        # 从EventBridge传递的环境变量获取PR信息
        PR_ID="${PULL_REQUEST_ID:-}"
        REPO_NAME="${REPOSITORY_NAME:-unicorn-web-project}"
        # 请替换为您的服务所在的区域信息
        REGION="us-east-1"  
        
        echo "PR ID from EventBridge: $PR_ID"
        echo "Repository Name: $REPO_NAME"
        
        if [ -z "$PR_ID" ]; then
          echo "ERROR: No PR ID provided from EventBridge"
          exit 1
        fi
        
        echo "=== Getting PR Details ==="
        # 直接使用EventBridge传递的PR ID获取PR详细信息
        echo "Getting PR details for PR: $PR_ID"
        aws codecommit get-pull-request \
          --pull-request-id "$PR_ID" \
          --region "$REGION" > pr_details.json
        
        if [ $? -ne 0 ]; then
          echo "ERROR: Failed to get PR details for PR ID: $PR_ID"
          exit 1
        fi
        
        # 提取sourceCommit和destinationCommit
        PR_SOURCE_COMMIT=$(grep -o '"sourceCommit"[[:space:]]*:[[:space:]]*"[^"]*"' pr_details.json | sed 's/.*"sourceCommit"[[:space:]]*:[[:space:]]*"\([^"]*\)".*/\1/' | head -1)
        PR_DEST_COMMIT=$(grep -o '"destinationCommit"[[:space:]]*:[[:space:]]*"[^"]*"' pr_details.json | sed 's/.*"destinationCommit"[[:space:]]*:[[:space:]]*"\([^"]*\)".*/\1/' | head -1)
        
        echo "✓ Successfully retrieved PR details:"
        echo "✓ PR ID: $PR_ID"
        echo "✓ Source commit: $PR_SOURCE_COMMIT"
        echo "✓ Destination commit: $PR_DEST_COMMIT"
        
        # 验证commit信息
        if [ -z "$PR_SOURCE_COMMIT" ] || [ -z "$PR_DEST_COMMIT" ] || [ "$PR_SOURCE_COMMIT" = "null" ] || [ "$PR_DEST_COMMIT" = "null" ]; then
          echo "ERROR: Invalid commit information retrieved"
          echo "Source commit: '$PR_SOURCE_COMMIT'"
          echo "Destination commit: '$PR_DEST_COMMIT'"
          exit 1
        fi
        
        echo "=== Getting Changed Files via AWS CLI ==="
        # 使用AWS CLI获取变更文件列表，替代git clone + git diff
        echo "Getting differences between commits via AWS CodeCommit API"
        aws codecommit get-differences \
          --repository-name "$REPO_NAME" \
          --before-commit-specifier "$PR_DEST_COMMIT" \
          --after-commit-specifier "$PR_SOURCE_COMMIT" \
          --region "$REGION" \
          --query 'differences[].afterBlob.path' \
          --output text > changes.txt
        
        if [ $? -ne 0 ]; then
          echo "ERROR: Failed to get differences for repository: $REPO_NAME"
          exit 1
        fi
        
        echo "Changed files:"
        cat changes.txt
        
        # 确保changes.txt不为空
        if [ ! -s changes.txt ]; then
          echo "No changes detected in PR"
          echo "No files to review" > changes.txt
        fi
        
        echo "=== Downloading Changed Files for Review ==="
        # 创建目录存放要审核的文件
        mkdir -p review-files
        
        # 下载每个变更文件的内容
        while IFS= read -r file_path; do
          if [ -n "$file_path" ] && [ "$file_path" != "No files to review" ]; then
            echo "Downloading file: $file_path"
            # 创建目录结构
            mkdir -p "review-files/$(dirname "$file_path")" 2>/dev/null || true
            
            # 下载文件内容
            if aws codecommit get-file \
              --repository-name "$REPO_NAME" \
              --commit-specifier "$PR_SOURCE_COMMIT" \
              --file-path "$file_path" \
              --region "$REGION" \
              --query 'fileContent' \
              --output text | base64 -d > "review-files/$file_path"; then
              echo "✓ Downloaded: $file_path"
            else
              echo "⚠ Failed to download: $file_path"
            fi
          fi
        done < changes.txt
        
        echo "=== Files ready for review ==="
        find review-files -type f 2>/dev/null | head -10

  build:
    commands:
      - echo Build started on `date`
      - echo "=== Running Amazon Q Developer CLI for code review ==="
      - |
        # 运行Amazon Q审查（使用预创建的prompt文件）
        echo "Running Amazon Q with standardized prompt..."
        echo "Prompt file size: $(wc -c < code-review-prompt.txt) characters"
        
        # 运行Q命令
        q chat -a -- "$(cat code-review-prompt.txt)"   
        cp amazon_q_review.md amazon_q_review_${CODEBUILD_BUILD_NUMBER}.md

  post_build:
    commands:
      - echo Build completed on `date`
      - echo "=== Processing Review Results ==="
      - |
        if [ -f "amazon_q_review_${CODEBUILD_BUILD_NUMBER}.md" ]; then
          # 生成带时间戳的文件名
          TIMESTAMP=$(date +"%Y%m%d-%H%M%S")
          BUILD_NUM=${CODEBUILD_BUILD_NUMBER:-$(date +%s)}
          COMMIT_ID=$(git rev-parse --short HEAD)
          S3_KEY="review-results/${TIMESTAMP}-build${BUILD_NUM}-${COMMIT_ID}-review.md"
          
          # 上传到S3 - 可选
          aws s3 cp amazon_q_review_${CODEBUILD_BUILD_NUMBER}.md s3://code-family-code-review/${S3_KEY}
          echo "Review report uploaded to s3://code-family-code-review/${S3_KEY}"
          
          # 创建摘要文件
          cat > review_summary.md << EOF
        # 代码审查报告链接
        
        审查报告已上传到: s3://code-family-code-review/${S3_KEY}
        构建编号: ${BUILD_NUM}
        提交ID: ${COMMIT_ID}
        审查时间: $(date)
        EOF
          
          aws s3 cp review_summary.md s3://code-family-code-review/review-results/${TIMESTAMP}-build${BUILD_NUM}-${COMMIT_ID}-summary.md
          echo "Review summary uploaded to S3"
        else
          echo "Warning: amazon_q_review_${CODEBUILD_BUILD_NUMBER}.md not found, skipping S3 upload"
        fi
        
        echo "=== Adding Comment to Pull Request ==="
        if [ -f "amazon_q_review_${CODEBUILD_BUILD_NUMBER}.md" ]; then
          # 使用前面获取的PR变量
          REPO_NAME="${REPOSITORY_NAME:-unicorn-web-project}"
          
          if [ -n "$PR_ID" ] && [ "$PR_ID" != "" ] && [ -n "$PR_DEST_COMMIT" ] && [ -n "$PR_SOURCE_COMMIT" ]; then
            echo "Adding comment to Pull Request #$PR_ID in repository $REPO_NAME"
            
            # 准备元数据
            BUILD_NUM=${CODEBUILD_BUILD_NUMBER:-$(date +%s)}
            TIMESTAMP=$(date +"%Y%m%d-%H%M%S")
            COMMIT_ID=$(git rev-parse --short HEAD)
            S3_KEY="review-results/${TIMESTAMP}-build${BUILD_NUM}-${COMMIT_ID}-review.md"
            
            # 创建带有元数据的comment内容
            cat > comment_content.md << EOF
        ## 自动代码审查报告
        
        ---
        **生成时间:** $(date +"%Y-%m-%d %H:%M:%S")
        **构建编号:** ${BUILD_NUM}
        **提交ID:** ${COMMIT_ID}
        **审查范围:** ${PR_DEST_COMMIT:0:8}...${PR_SOURCE_COMMIT:0:8}
        **S3报告链接:** s3://<请填入您的s3桶名称>/${S3_KEY}
        **变更文件数:** $(wc -l < changes.txt)
        **Repository:** ${REPO_NAME}
        ---
        
        ### 变更文件列表:
        EOF
            
            while IFS= read -r file; do
              echo "- \`$file\`" >> comment_content.md
            done < changes.txt
            
            echo "" >> comment_content.md
            echo "---" >> comment_content.md
            echo "" >> comment_content.md
            
            # 添加审查内容
            cat amazon_q_review_${CODEBUILD_BUILD_NUMBER}.md >> comment_content.md
            
            # 使用AWS CLI添加comment到Pull Request
            echo "Posting comment to Pull Request: $PR_ID"
            echo "Repository: $REPO_NAME"
            echo "Before commit: $PR_DEST_COMMIT"
            echo "After commit: $PR_SOURCE_COMMIT"
            
            # 执行AWS CLI命令添加comment
            if aws codecommit post-comment-for-pull-request \
              --pull-request-id "$PR_ID" \
              --repository-name "$REPO_NAME" \
              --before-commit-id "$PR_DEST_COMMIT" \
              --after-commit-id "$PR_SOURCE_COMMIT" \
              --content "$(cat comment_content.md)" \
              --region us-east-1; then
              echo "Successfully added comment to Pull Request #$PR_ID"
            else
              echo "Failed to add comment to Pull Request #$PR_ID"
            fi
          else
            echo "PR information not available or incomplete, skipping comment"
            echo "PR_ID: '$PR_ID'"
            echo "PR_DEST_COMMIT: '$PR_DEST_COMMIT'"
            echo "PR_SOURCE_COMMIT: '$PR_SOURCE_COMMIT'"
          fi
        else
          echo "amazon_q_review_${CODEBUILD_BUILD_NUMBER}.md not found, skipping PR comment"
        fi

artifacts:
  files:
    - amazon_q_review_*.md
    - review_summary.md
    - comment_content.md
    - review-files/**/*
  discard-paths: no