Compare commits
96 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| 175f92542e | |||
| b448787f24 | |||
| 65e3da81e9 | |||
| f31e3a026e | |||
| 220652f27b | |||
| 349af82bfc | |||
| 15732d126d | |||
| 36b006108b | |||
| 029ae0d280 | |||
| 92b435eb80 | |||
| cacf9674c4 | |||
| f73260e3e6 | |||
| 3ddd6db7ca | |||
| 68499340e4 | |||
| acb06634be | |||
| 681e08557c | |||
| 15a7154aea | |||
| 434458a068 | |||
| 193954fdd7 | |||
| ab2d762658 | |||
| 876663cd1a | |||
| b2badf6d16 | |||
| 05fb2065b2 | |||
| 2042a5bf98 | |||
| 12da2b770c | |||
| a26ada9839 | |||
| 1328894afd | |||
| 85731f8360 | |||
| 1922352d86 | |||
| c0f31415a3 | |||
| 706ce04895 | |||
| 09be8e3068 | |||
| c6c1210fa0 | |||
| 0368512b9e | |||
| 9fb6135fd2 | |||
| bb78e5f94b | |||
| c668578c6f | |||
| 7f1566d9c2 | |||
| dd482566c2 | |||
| 57029c32a3 | |||
| 757d336475 | |||
| d03debab4b | |||
| a05829f781 | |||
| c8ba7940e3 | |||
| 710fdd0fa4 | |||
| 4f5ee28b8a | |||
| 806458b95a | |||
| 2c856e3337 | |||
| a93c34e61b | |||
| cd20491d22 | |||
| 1a6ce6728c | |||
| 3b39f2c8d8 | |||
| ec0eeb1d18 | |||
| 8ef805b9fc | |||
| 6e93fd9a72 | |||
| 9567d84442 | |||
| d79564db5e | |||
| a7a0e87fd8 | |||
| 7050b8de75 | |||
| 2fc3ddee16 | |||
| 284d9733dd | |||
| 94e2b5f6e0 | |||
| 03810d574e | |||
| f52e94c612 | |||
| 9444a0e208 | |||
| 2296060d04 | |||
| 458bfe18a0 | |||
| 4cfb9b5008 | |||
| 06542e382a | |||
| bcdf6ab5fb | |||
| 314f669f10 | |||
| a24275e21b | |||
| 872e620342 | |||
| 6d9c6fd2b1 | |||
| 008021df1c | |||
| d6bc0fdd64 | |||
| 8fd0328109 | |||
| a1f87ce118 | |||
| 3e7122611c | |||
| 9459803aaa | |||
| f74a75cfac | |||
| 16e04e7152 | |||
| 522779ef54 | |||
| d3d2dad025 | |||
| 66bf86f081 | |||
| 608322f026 | |||
| e3a3d416fb | |||
| f0e339f5c2 | |||
| 87d32763e9 | |||
| 671334f37d | |||
| 267ed3686f | |||
| ff1226a824 | |||
| e13c5eed00 | |||
| f2a1c3c1bb | |||
| e672deb934 | |||
| 70b365fca1 |
@@ -29,12 +29,12 @@ jobs:
|
|||||||
- name: Setup Node.js
|
- name: Setup Node.js
|
||||||
uses: actions/setup-node@v4
|
uses: actions/setup-node@v4
|
||||||
with:
|
with:
|
||||||
node-version: "20"
|
node-version: "24"
|
||||||
cache: "pnpm"
|
cache: "pnpm"
|
||||||
registry-url: "https://registry.npmjs.org"
|
registry-url: "https://registry.npmjs.org"
|
||||||
|
|
||||||
- name: Install dependencies
|
- name: Install dependencies
|
||||||
run: pnpm install --no-frozen-lockfile
|
run: pnpm install --frozen-lockfile
|
||||||
|
|
||||||
- name: Get package version
|
- name: Get package version
|
||||||
id: version
|
id: version
|
||||||
@@ -60,21 +60,6 @@ jobs:
|
|||||||
echo "✅ Tag ${{ steps.version.outputs.tag }} does not exist - will create release"
|
echo "✅ Tag ${{ steps.version.outputs.tag }} does not exist - will create release"
|
||||||
fi
|
fi
|
||||||
|
|
||||||
- name: Verify built files are up to date
|
|
||||||
if: steps.check_tag.outputs.exists == 'false'
|
|
||||||
run: |
|
|
||||||
# Check if there are any uncommitted changes
|
|
||||||
if [[ -n $(git status --porcelain) ]]; then
|
|
||||||
echo "❌ Error: There are uncommitted changes. Built files should be committed via pre-commit hook."
|
|
||||||
git status
|
|
||||||
exit 1
|
|
||||||
fi
|
|
||||||
echo "✅ All built files are up to date"
|
|
||||||
|
|
||||||
- name: Build for npm with zshy
|
|
||||||
if: steps.check_tag.outputs.exists == 'false'
|
|
||||||
run: pnpm build:npm
|
|
||||||
|
|
||||||
- name: Create and push tags
|
- name: Create and push tags
|
||||||
if: steps.check_tag.outputs.exists == 'false'
|
if: steps.check_tag.outputs.exists == 'false'
|
||||||
run: |
|
run: |
|
||||||
|
|||||||
+10
-3
@@ -1,6 +1,13 @@
|
|||||||
|
#!/usr/bin/env sh
|
||||||
|
. "$(dirname -- "$0")/_/husky.sh"
|
||||||
|
|
||||||
|
# Ensure lockfile is up to date
|
||||||
|
echo "🔒 Updating lockfile..."
|
||||||
|
pnpm install --lockfile-only
|
||||||
|
|
||||||
# Build the action before committing
|
# Build the action before committing
|
||||||
echo "🔨 Building action..."
|
echo "🔨 Building action..."
|
||||||
npm run build
|
pnpm build
|
||||||
|
|
||||||
# Add the built files to the commit
|
# Add the built files and lockfile to the commit
|
||||||
git add entry.cjs
|
git add entry.cjs pnpm-lock.yaml
|
||||||
|
|||||||
@@ -0,0 +1,299 @@
|
|||||||
|
# Claude Code Action Architecture & Flow
|
||||||
|
|
||||||
|
This document provides a comprehensive overview of how the official (Anthropic) Claude Code Action works, from token exchange through post-run cleanup.
|
||||||
|
|
||||||
|
## Overview
|
||||||
|
|
||||||
|
The Claude Code Action is a sophisticated GitHub automation platform that enables Claude to interact with GitHub repositories through secure token exchange, intelligent mode detection, and comprehensive GitHub API integration.
|
||||||
|
|
||||||
|
## High-Level Architecture
|
||||||
|
|
||||||
|
```mermaid
|
||||||
|
graph TD
|
||||||
|
Start([GitHub Action Triggered]) --> Setup[Setup Environment<br/>- Install Bun<br/>- Install Dependencies]
|
||||||
|
|
||||||
|
Setup --> ParseContext[Parse GitHub Context<br/>- Extract event data<br/>- Parse inputs]
|
||||||
|
|
||||||
|
ParseContext --> ModeDetection{Mode Detection}
|
||||||
|
|
||||||
|
ModeDetection -->|Has explicit prompt| AgentMode[AGENT MODE<br/>Direct automation]
|
||||||
|
ModeDetection -->|@claude mention/assignment/label| TagMode[TAG MODE<br/>Interactive response]
|
||||||
|
ModeDetection -->|No trigger| DefaultAgent[Default to Agent<br/>(won't trigger)]
|
||||||
|
|
||||||
|
%% Token Exchange Branch
|
||||||
|
AgentMode --> TokenExchange[Token Exchange Process]
|
||||||
|
TagMode --> TokenExchange
|
||||||
|
TokenExchange --> TokenMethod{Token Method}
|
||||||
|
|
||||||
|
TokenMethod -->|Custom token provided| UseCustom[Use Custom GitHub Token]
|
||||||
|
TokenMethod -->|No custom token| OIDC[Generate OIDC Token<br/>core.getIDToken()]
|
||||||
|
|
||||||
|
OIDC --> Exchange[Exchange OIDC for App Token<br/>api.anthropic.com/api/github/github-app-token-exchange]
|
||||||
|
Exchange --> CreateOctokit[Create Authenticated Octokit Client<br/>REST + GraphQL]
|
||||||
|
UseCustom --> CreateOctokit
|
||||||
|
|
||||||
|
%% Permission Checks
|
||||||
|
CreateOctokit --> PermCheck[Check Write Permissions<br/>Only for entity contexts]
|
||||||
|
PermCheck -->|No permissions| PermFail[❌ Exit: No write access]
|
||||||
|
PermCheck -->|Has permissions| TriggerCheck{Check Trigger Conditions}
|
||||||
|
|
||||||
|
%% Trigger Validation
|
||||||
|
TriggerCheck -->|Agent Mode| AgentTrigger{Has explicit prompt?}
|
||||||
|
TriggerCheck -->|Tag Mode| TagTrigger{Contains @claude mention<br/>or assignment/label?}
|
||||||
|
|
||||||
|
AgentTrigger -->|No prompt| NoTrigger[❌ Skip: No trigger found]
|
||||||
|
AgentTrigger -->|Has prompt| PrepareAgent[Prepare Agent Mode]
|
||||||
|
TagTrigger -->|No mention| NoTrigger
|
||||||
|
TagTrigger -->|Has mention| PrepareTag[Prepare Tag Mode]
|
||||||
|
|
||||||
|
%% Mode-Specific Preparation
|
||||||
|
PrepareAgent --> AgentPrep[Agent Mode Preparation<br/>- Create prompt file<br/>- Setup MCP servers<br/>- No tracking comment]
|
||||||
|
PrepareTag --> TagPrep[Tag Mode Preparation<br/>- Create tracking comment<br/>- Setup branches<br/>- Fetch GitHub data<br/>- Setup MCP servers]
|
||||||
|
|
||||||
|
%% Data Fetching (Tag Mode)
|
||||||
|
TagPrep --> DataFetch[Fetch GitHub Data<br/>GraphQL + REST API]
|
||||||
|
DataFetch --> FetchWhat{What to fetch?}
|
||||||
|
|
||||||
|
FetchWhat -->|Pull Request| PRData[PR Data:<br/>- Comments & reviews<br/>- Changed files + SHAs<br/>- Commit history<br/>- Author info]
|
||||||
|
FetchWhat -->|Issue| IssueData[Issue Data:<br/>- Comments<br/>- Issue details<br/>- Author info]
|
||||||
|
|
||||||
|
PRData --> ProcessImages[Process Images<br/>Download & convert to base64]
|
||||||
|
IssueData --> ProcessImages
|
||||||
|
ProcessImages --> SetupBranch[Setup Branch<br/>- Create Claude branch<br/>- Configure git auth]
|
||||||
|
|
||||||
|
%% MCP Server Setup
|
||||||
|
AgentPrep --> MCPSetup[Setup MCP Servers]
|
||||||
|
SetupBranch --> MCPSetup
|
||||||
|
|
||||||
|
MCPSetup --> MCPServers{MCP Servers}
|
||||||
|
MCPServers --> GitHubActions[GitHub Actions Server<br/>- Workflow data<br/>- CI results]
|
||||||
|
MCPServers --> GitHubComments[GitHub Comment Server<br/>- Comment operations]
|
||||||
|
MCPServers --> GitHubFiles[GitHub File Ops Server<br/>- File operations<br/>- Branch management]
|
||||||
|
MCPServers --> GitHubInline[GitHub Inline Comment Server<br/>- PR review comments]
|
||||||
|
|
||||||
|
GitHubActions --> PromptGen[Generate Prompt]
|
||||||
|
GitHubComments --> PromptGen
|
||||||
|
GitHubFiles --> PromptGen
|
||||||
|
GitHubInline --> PromptGen
|
||||||
|
|
||||||
|
%% Prompt Generation
|
||||||
|
PromptGen --> PromptType{Prompt Type}
|
||||||
|
PromptType -->|Agent Mode| AgentPrompt[Agent Prompt:<br/>- Direct user prompt<br/>- Minimal context]
|
||||||
|
PromptType -->|Tag Mode| TagPrompt[Tag Prompt:<br/>- Rich GitHub context<br/>- PR/Issue details<br/>- Changed files<br/>- Comments & reviews<br/>- Commit instructions]
|
||||||
|
|
||||||
|
AgentPrompt --> ClaudeRun[Run Claude Code]
|
||||||
|
TagPrompt --> ClaudeRun
|
||||||
|
|
||||||
|
%% Claude Execution
|
||||||
|
ClaudeRun --> ClaudeExec[Claude Code Execution<br/>base-action/src/index.ts]
|
||||||
|
ClaudeExec --> ClaudeArgs[Prepare Claude Args<br/>- Prompt file path<br/>- Custom claude_args<br/>- Output format: stream-json]
|
||||||
|
|
||||||
|
ClaudeArgs --> ClaudeProvider{Provider}
|
||||||
|
ClaudeProvider -->|Default| AnthropicAPI[Anthropic API<br/>ANTHROPIC_API_KEY]
|
||||||
|
ClaudeProvider -->|Bedrock| AWSBedrock[AWS Bedrock<br/>OIDC + AWS credentials]
|
||||||
|
ClaudeProvider -->|Vertex| GCPVertex[GCP Vertex AI<br/>OIDC + GCP credentials]
|
||||||
|
|
||||||
|
AnthropicAPI --> ClaudeProcess[Spawn Claude Process<br/>- Named pipe for input<br/>- Stream JSON output]
|
||||||
|
AWSBedrock --> ClaudeProcess
|
||||||
|
GCPVertex --> ClaudeProcess
|
||||||
|
|
||||||
|
ClaudeProcess --> ClaudeTools[Claude Tool Usage<br/>- MCP tools<br/>- File operations<br/>- GitHub API calls<br/>- Bash commands]
|
||||||
|
|
||||||
|
ClaudeTools --> ClaudeOutput[Claude Output Processing<br/>- Capture execution log<br/>- Parse JSON stream<br/>- Extract metrics]
|
||||||
|
|
||||||
|
%% Post-Run Actions
|
||||||
|
ClaudeOutput --> PostRun{Post-Run Actions}
|
||||||
|
|
||||||
|
PostRun -->|Success| Success[✅ Success Path]
|
||||||
|
PostRun -->|Failure| Failure[❌ Failure Path]
|
||||||
|
|
||||||
|
Success --> UpdateComment[Update Tracking Comment<br/>- Job run link<br/>- Branch link<br/>- PR link (if created)<br/>- Execution metrics]
|
||||||
|
Failure --> UpdateComment
|
||||||
|
|
||||||
|
UpdateComment --> BranchCleanup[Branch Cleanup<br/>- Check for changes<br/>- Delete empty branches<br/>- Keep branches with commits]
|
||||||
|
|
||||||
|
BranchCleanup --> FormatReport[Format Execution Report<br/>- Parse conversation turns<br/>- Format tool usage<br/>- Add to GitHub step summary]
|
||||||
|
|
||||||
|
FormatReport --> RevokeToken[Revoke App Token<br/>DELETE /installation/token]
|
||||||
|
|
||||||
|
RevokeToken --> End([Action Complete])
|
||||||
|
```
|
||||||
|
|
||||||
|
## Key Components
|
||||||
|
|
||||||
|
### 1. Token Exchange Process
|
||||||
|
|
||||||
|
The action uses a secure OIDC token exchange system:
|
||||||
|
|
||||||
|
1. **OIDC Token Generation**: `core.getIDToken("claude-code-github-action")`
|
||||||
|
2. **Token Exchange**: POST to `https://api.anthropic.com/api/github/github-app-token-exchange`
|
||||||
|
3. **Authentication**: Creates authenticated Octokit clients for GitHub API access
|
||||||
|
|
||||||
|
**Security Benefits:**
|
||||||
|
- Repository-scoped access
|
||||||
|
- Time-limited tokens
|
||||||
|
- Permission-limited (only configured GitHub App permissions)
|
||||||
|
- Automatic token masking in logs
|
||||||
|
|
||||||
|
### 2. Mode Detection
|
||||||
|
|
||||||
|
The action automatically detects the appropriate execution mode:
|
||||||
|
|
||||||
|
#### **Agent Mode**
|
||||||
|
- **Trigger**: Explicit `prompt` input provided
|
||||||
|
- **Use Case**: Direct automation, custom workflows
|
||||||
|
- **Behavior**: Minimal context, direct execution
|
||||||
|
- **Tracking**: No tracking comments
|
||||||
|
|
||||||
|
#### **Tag Mode**
|
||||||
|
- **Trigger**: @claude mentions, issue assignments, or labels
|
||||||
|
- **Use Case**: Interactive GitHub responses
|
||||||
|
- **Behavior**: Rich context, comprehensive GitHub data
|
||||||
|
- **Tracking**: Creates and updates tracking comments
|
||||||
|
|
||||||
|
### 3. Data Fetching (Tag Mode)
|
||||||
|
|
||||||
|
When in Tag Mode, the action fetches comprehensive GitHub context:
|
||||||
|
|
||||||
|
#### **Pull Request Data:**
|
||||||
|
- Comments and reviews (including inline comments)
|
||||||
|
- Changed files with SHAs
|
||||||
|
- Commit history and metadata
|
||||||
|
- Author information
|
||||||
|
- File diff data
|
||||||
|
|
||||||
|
#### **Issue Data:**
|
||||||
|
- Issue details and metadata
|
||||||
|
- All comments
|
||||||
|
- Author information
|
||||||
|
- Labels and assignments
|
||||||
|
|
||||||
|
#### **Image Processing:**
|
||||||
|
- Downloads images from GitHub
|
||||||
|
- Converts to base64 for Claude
|
||||||
|
- Maps original URLs to processed content
|
||||||
|
|
||||||
|
### 4. MCP Server Integration
|
||||||
|
|
||||||
|
The action sets up multiple MCP (Model Context Protocol) servers to provide Claude with GitHub capabilities:
|
||||||
|
|
||||||
|
#### **GitHub Actions Server**
|
||||||
|
- Access to workflow runs and CI data
|
||||||
|
- Build status and test results
|
||||||
|
- Artifact information
|
||||||
|
|
||||||
|
#### **GitHub Comment Server**
|
||||||
|
- Comment creation and updates
|
||||||
|
- Issue and PR comment management
|
||||||
|
|
||||||
|
#### **GitHub File Operations Server**
|
||||||
|
- File reading and writing
|
||||||
|
- Branch creation and management
|
||||||
|
- Commit operations
|
||||||
|
|
||||||
|
#### **GitHub Inline Comment Server**
|
||||||
|
- PR review comment operations
|
||||||
|
- Line-specific feedback
|
||||||
|
|
||||||
|
### 5. Prompt Generation
|
||||||
|
|
||||||
|
The action generates context-rich prompts based on the detected mode:
|
||||||
|
|
||||||
|
#### **Agent Mode Prompts:**
|
||||||
|
- Direct user prompt
|
||||||
|
- Minimal GitHub context
|
||||||
|
- Focused on specific task
|
||||||
|
|
||||||
|
#### **Tag Mode Prompts:**
|
||||||
|
- Comprehensive GitHub context
|
||||||
|
- PR/Issue details and history
|
||||||
|
- Changed files and diffs
|
||||||
|
- Comment threads and reviews
|
||||||
|
- Commit instructions and guidelines
|
||||||
|
|
||||||
|
### 6. Claude Execution
|
||||||
|
|
||||||
|
The action runs Claude Code through multiple provider options:
|
||||||
|
|
||||||
|
#### **Provider Support:**
|
||||||
|
- **Anthropic API** (default): Direct API access with API key
|
||||||
|
- **AWS Bedrock**: OIDC authentication with AWS credentials
|
||||||
|
- **GCP Vertex AI**: OIDC authentication with GCP credentials
|
||||||
|
|
||||||
|
#### **Execution Process:**
|
||||||
|
1. **Named Pipe Setup**: Creates pipe for prompt input
|
||||||
|
2. **Process Spawning**: Spawns Claude Code process
|
||||||
|
3. **Stream Processing**: Captures JSON stream output
|
||||||
|
4. **Tool Integration**: Enables MCP tools and GitHub operations
|
||||||
|
|
||||||
|
### 7. Post-Run Actions
|
||||||
|
|
||||||
|
After Claude execution, the action performs comprehensive cleanup and reporting:
|
||||||
|
|
||||||
|
#### **Comment Updates:**
|
||||||
|
- Updates tracking comments with results
|
||||||
|
- Adds job run links and execution metrics
|
||||||
|
- Includes branch and PR links when created
|
||||||
|
|
||||||
|
#### **Branch Management:**
|
||||||
|
- Checks for actual changes in Claude branches
|
||||||
|
- Deletes empty branches to avoid clutter
|
||||||
|
- Preserves branches with meaningful commits
|
||||||
|
|
||||||
|
#### **Report Generation:**
|
||||||
|
- Parses execution logs and conversation turns
|
||||||
|
- Formats tool usage and results
|
||||||
|
- Adds formatted report to GitHub step summary
|
||||||
|
|
||||||
|
#### **Security Cleanup:**
|
||||||
|
- Revokes GitHub App installation token
|
||||||
|
- Cleans up temporary files and processes
|
||||||
|
|
||||||
|
## Security Considerations
|
||||||
|
|
||||||
|
### **Access Control:**
|
||||||
|
- Repository-scoped permissions only
|
||||||
|
- Write access validation for actors
|
||||||
|
- Bot user controls and allowlists
|
||||||
|
|
||||||
|
### **Token Management:**
|
||||||
|
- Short-lived installation tokens
|
||||||
|
- Automatic token revocation after use
|
||||||
|
- Secure OIDC-based exchange
|
||||||
|
|
||||||
|
### **Permission Boundaries:**
|
||||||
|
- Limited to configured GitHub App permissions
|
||||||
|
- No cross-repository access
|
||||||
|
- Scoped to specific repository operations
|
||||||
|
|
||||||
|
## Integration Points
|
||||||
|
|
||||||
|
### **With Pullfrog:**
|
||||||
|
The Claude Code Action can be integrated with Pullfrog's workflow system, providing:
|
||||||
|
- Standardized agent interaction patterns
|
||||||
|
- Consistent GitHub integration
|
||||||
|
- Reusable authentication flows
|
||||||
|
- Common MCP server infrastructure
|
||||||
|
|
||||||
|
### **With GitHub:**
|
||||||
|
- Native GitHub Actions integration
|
||||||
|
- Comprehensive API coverage (REST + GraphQL)
|
||||||
|
- Proper webhook handling
|
||||||
|
- Standard GitHub UI integration
|
||||||
|
|
||||||
|
## Development Notes
|
||||||
|
|
||||||
|
### **Key Files:**
|
||||||
|
- `src/entrypoints/prepare.ts`: Main preparation logic
|
||||||
|
- `src/modes/`: Mode detection and handling
|
||||||
|
- `src/github/token.ts`: OIDC token exchange
|
||||||
|
- `src/mcp/`: MCP server implementations
|
||||||
|
- `base-action/`: Core Claude Code execution
|
||||||
|
|
||||||
|
### **Testing:**
|
||||||
|
- Unit tests for individual components
|
||||||
|
- Integration tests for full workflows
|
||||||
|
- Local testing with `act` tool
|
||||||
|
- Comprehensive fixture support
|
||||||
|
|
||||||
|
This architecture provides a robust, secure, and extensible foundation for Claude-GitHub integration while maintaining clear separation of concerns and comprehensive error handling.
|
||||||
@@ -2,6 +2,8 @@
|
|||||||
|
|
||||||
GitHub Action for running Claude Code and other agents via Pullfrog.
|
GitHub Action for running Claude Code and other agents via Pullfrog.
|
||||||
|
|
||||||
|
> **📖 Claude Code Action Architecture**: For a detailed technical overview of how the Claude Code Action works (token exchange, modes, data fetching, execution flow), see [CLAUDE-ACTION.md](./CLAUDE-ACTION.md).
|
||||||
|
|
||||||
## Quick Start
|
## Quick Start
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
@@ -10,69 +12,13 @@ pnpm install
|
|||||||
|
|
||||||
# Test with default prompt
|
# Test with default prompt
|
||||||
npm run play # Run locally on your machine
|
npm run play # Run locally on your machine
|
||||||
npm run play -- --act # Run in Docker (simulates GitHub Actions)
|
|
||||||
```
|
```
|
||||||
|
|
||||||
## Testing with play.ts
|
## Testing with play.ts
|
||||||
|
|
||||||
The `play.ts` script provides two ways to test the action:
|
|
||||||
|
|
||||||
### Local Mode (Default)
|
|
||||||
```bash
|
```bash
|
||||||
npm run play # Uses fixtures/play.txt
|
pnpm play # Uses fixtures/play.txt
|
||||||
npm run play fixtures/complex.txt # Custom prompt file
|
|
||||||
```
|
```
|
||||||
- Clones the scratch repository to `.temp`
|
- Clones the scratch repository to `.temp`
|
||||||
- Runs Claude Code directly on your machine
|
- Runs Claude Code directly on your machine
|
||||||
- Fast iteration for development
|
- Fast iteration for development
|
||||||
|
|
||||||
### Docker Mode (--act flag)
|
|
||||||
```bash
|
|
||||||
npm run play -- --act # Uses fixtures/play.txt
|
|
||||||
npm run play fixtures/simple.txt -- --act # Custom prompt file
|
|
||||||
```
|
|
||||||
- Builds fresh bundles with esbuild
|
|
||||||
- Creates minimal distribution without node_modules
|
|
||||||
- Runs in Docker container via `act`
|
|
||||||
- Simulates GitHub Actions environment
|
|
||||||
|
|
||||||
### Prompt Files
|
|
||||||
|
|
||||||
Supports `.txt`, `.json`, and `.ts` files:
|
|
||||||
```bash
|
|
||||||
npm run play prompt.txt # Plain text prompt
|
|
||||||
npm run play config.json # JSON configuration
|
|
||||||
npm run play dynamic.ts # TypeScript with default export
|
|
||||||
```
|
|
||||||
|
|
||||||
## Building
|
|
||||||
|
|
||||||
```bash
|
|
||||||
pnpm build # Production build (bundles & removes node_modules)
|
|
||||||
pnpm build:dev # Development build (keeps node_modules)
|
|
||||||
pnpm dev # Watch mode
|
|
||||||
```
|
|
||||||
|
|
||||||
The action is bundled into `entry.cjs` with all dependencies included, eliminating runtime dependency on node_modules.
|
|
||||||
|
|
||||||
## Environment Variables
|
|
||||||
|
|
||||||
Create `.env` in `/action`:
|
|
||||||
|
|
||||||
```bash
|
|
||||||
ANTHROPIC_API_KEY=sk-ant-api03-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx # Claude API key
|
|
||||||
```
|
|
||||||
|
|
||||||
## Architecture
|
|
||||||
|
|
||||||
- **entry.cjs**: Bundled action entry point (self-contained)
|
|
||||||
- **agents/**: Agent implementations (Claude, etc.)
|
|
||||||
- **utils/**: Utilities for subprocess, act, and formatting
|
|
||||||
- **fixtures/**: Test prompt files
|
|
||||||
|
|
||||||
## Why No node_modules?
|
|
||||||
|
|
||||||
pnpm uses symlinks that cause "invalid symlink" errors when `act` copies the action to Docker. Our solution:
|
|
||||||
1. Bundle everything into `entry.cjs`
|
|
||||||
2. Remove node_modules after building
|
|
||||||
3. Create minimal `.act-dist` for Docker testing
|
|
||||||
+1
-7
@@ -10,12 +10,6 @@ inputs:
|
|||||||
anthropic_api_key:
|
anthropic_api_key:
|
||||||
description: "Anthropic API key for Claude Code authentication"
|
description: "Anthropic API key for Claude Code authentication"
|
||||||
required: false
|
required: false
|
||||||
github_token:
|
|
||||||
description: "GitHub token for repository access"
|
|
||||||
required: false
|
|
||||||
github_installation_token:
|
|
||||||
description: "GitHub App installation token"
|
|
||||||
required: false
|
|
||||||
|
|
||||||
runs:
|
runs:
|
||||||
using: "node20"
|
using: "node20"
|
||||||
@@ -23,4 +17,4 @@ runs:
|
|||||||
|
|
||||||
branding:
|
branding:
|
||||||
icon: "code"
|
icon: "code"
|
||||||
color: "orange"
|
color: "green"
|
||||||
|
|||||||
+138
-399
@@ -1,406 +1,145 @@
|
|||||||
import { access, constants } from "node:fs/promises";
|
import { query, type SDKMessage } from "@anthropic-ai/claude-agent-sdk";
|
||||||
import * as core from "@actions/core";
|
import { log } from "../utils/cli.ts";
|
||||||
import { boxString, tableString } from "../utils";
|
import { type Agent, instructions } from "./shared.ts";
|
||||||
import { spawn } from "../utils/subprocess";
|
|
||||||
import type { Agent, AgentConfig, AgentResult } from "./types";
|
|
||||||
|
|
||||||
/**
|
export const claude: Agent = {
|
||||||
* Claude Code agent implementation
|
run: async ({ prompt, mcpServers, apiKey }) => {
|
||||||
*/
|
process.env.ANTHROPIC_API_KEY = apiKey;
|
||||||
export class ClaudeAgent implements Agent {
|
|
||||||
private apiKey: string;
|
|
||||||
public runStats = {
|
|
||||||
toolsUsed: 0,
|
|
||||||
turns: 0,
|
|
||||||
startTime: 0,
|
|
||||||
};
|
|
||||||
|
|
||||||
// $: ExecaMethod;
|
const queryInstance = query({
|
||||||
|
prompt: `${instructions}\n\n${prompt}`,
|
||||||
|
options: {
|
||||||
|
permissionMode: "bypassPermissions",
|
||||||
|
mcpServers,
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
constructor(config: AgentConfig) {
|
// Stream the results
|
||||||
if (!config.apiKey) {
|
for await (const message of queryInstance) {
|
||||||
throw new Error("Claude agent requires an API key");
|
const handler = messageHandlers[message.type];
|
||||||
}
|
await handler(message as never);
|
||||||
this.apiKey = config.apiKey;
|
|
||||||
// Removed execa dependency - using spawn utility instead
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Check if Claude Code CLI is already installed
|
|
||||||
*/
|
|
||||||
private async isClaudeInstalled(): Promise<boolean> {
|
|
||||||
try {
|
|
||||||
const claudePath = `${process.env.HOME}/.local/bin/claude`;
|
|
||||||
await access(claudePath, constants.F_OK | constants.X_OK);
|
|
||||||
return true;
|
|
||||||
} catch {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Install Claude Code CLI
|
|
||||||
*/
|
|
||||||
async install(): Promise<void> {
|
|
||||||
// Check if Claude Code is already installed
|
|
||||||
if (await this.isClaudeInstalled()) {
|
|
||||||
core.info("Claude Code is already installed, skipping installation");
|
|
||||||
return;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
core.info("Installing Claude Code...");
|
return {
|
||||||
try {
|
success: true,
|
||||||
// Use shell execution to properly handle the pipe
|
output: "",
|
||||||
const result = await spawn({
|
};
|
||||||
cmd: "bash",
|
},
|
||||||
args: [
|
};
|
||||||
"-c",
|
|
||||||
"curl -fsSL https://claude.ai/install.sh | bash -s 1.0.93",
|
type SDKMessageType = SDKMessage["type"];
|
||||||
|
|
||||||
|
type SDKMessageHandler<type extends SDKMessageType = SDKMessageType> = (
|
||||||
|
data: Extract<SDKMessage, { type: type }>
|
||||||
|
) => void | Promise<void>;
|
||||||
|
|
||||||
|
type SDKMessageHandlers = {
|
||||||
|
[type in SDKMessageType]: SDKMessageHandler<type>;
|
||||||
|
};
|
||||||
|
|
||||||
|
// Track bash tool IDs to identify when bash tool results come back
|
||||||
|
const bashToolIds = new Set<string>();
|
||||||
|
|
||||||
|
const messageHandlers: SDKMessageHandlers = {
|
||||||
|
assistant: (data) => {
|
||||||
|
if (data.message?.content) {
|
||||||
|
for (const content of data.message.content) {
|
||||||
|
if (content.type === "text" && content.text?.trim()) {
|
||||||
|
log.box(content.text.trim(), { title: "Claude" });
|
||||||
|
} else if (content.type === "tool_use") {
|
||||||
|
log.info(`→ ${content.name}`);
|
||||||
|
|
||||||
|
// Track bash tool IDs
|
||||||
|
if (content.name === "bash" && content.id) {
|
||||||
|
bashToolIds.add(content.id);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (content.input) {
|
||||||
|
const input = content.input as any;
|
||||||
|
if (input.description) log.info(` └─ ${input.description}`);
|
||||||
|
if (input.command) log.info(` └─ command: ${input.command}`);
|
||||||
|
if (input.file_path) log.info(` └─ file: ${input.file_path}`);
|
||||||
|
if (input.content) {
|
||||||
|
const preview =
|
||||||
|
input.content.length > 100
|
||||||
|
? `${input.content.substring(0, 100)}...`
|
||||||
|
: input.content;
|
||||||
|
log.info(` └─ content: ${preview}`);
|
||||||
|
}
|
||||||
|
if (input.query) log.info(` └─ query: ${input.query}`);
|
||||||
|
if (input.pattern) log.info(` └─ pattern: ${input.pattern}`);
|
||||||
|
if (input.url) log.info(` └─ url: ${input.url}`);
|
||||||
|
if (input.edits && Array.isArray(input.edits)) {
|
||||||
|
log.info(` └─ edits: ${input.edits.length} changes`);
|
||||||
|
input.edits.forEach((edit: any, index: number) => {
|
||||||
|
if (edit.file_path) log.info(` ${index + 1}. ${edit.file_path}`);
|
||||||
|
});
|
||||||
|
}
|
||||||
|
if (input.task) log.info(` └─ task: ${input.task}`);
|
||||||
|
if (input.bash_command) log.info(` └─ bash_command: ${input.bash_command}`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
user: (data) => {
|
||||||
|
if (data.message?.content) {
|
||||||
|
for (const content of data.message.content) {
|
||||||
|
if (content.type === "tool_result") {
|
||||||
|
const toolUseId = (content as any).tool_use_id;
|
||||||
|
const isBashTool = toolUseId && bashToolIds.has(toolUseId);
|
||||||
|
|
||||||
|
if (isBashTool) {
|
||||||
|
// Log bash output in a collapsed group
|
||||||
|
const outputContent =
|
||||||
|
typeof content.content === "string"
|
||||||
|
? content.content
|
||||||
|
: Array.isArray(content.content)
|
||||||
|
? content.content
|
||||||
|
.map((c: any) => (typeof c === "string" ? c : c.text || JSON.stringify(c)))
|
||||||
|
.join("\n")
|
||||||
|
: String(content.content);
|
||||||
|
|
||||||
|
log.startGroup(`bash output`);
|
||||||
|
if (content.is_error) {
|
||||||
|
log.warning(outputContent);
|
||||||
|
} else {
|
||||||
|
log.info(outputContent);
|
||||||
|
}
|
||||||
|
log.endGroup();
|
||||||
|
// Clean up the tracked ID
|
||||||
|
bashToolIds.delete(toolUseId);
|
||||||
|
} else if (content.is_error) {
|
||||||
|
const errorContent =
|
||||||
|
typeof content.content === "string" ? content.content : String(content.content);
|
||||||
|
log.warning(`Tool error: ${errorContent}`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
result: async (data) => {
|
||||||
|
if (data.subtype === "success") {
|
||||||
|
await log.summaryTable([
|
||||||
|
[
|
||||||
|
{ data: "Cost", header: true },
|
||||||
|
{ data: "Input Tokens", header: true },
|
||||||
|
{ data: "Output Tokens", header: true },
|
||||||
],
|
],
|
||||||
env: { ANTHROPIC_API_KEY: this.apiKey },
|
[
|
||||||
timeout: 120000, // 2 minute timeout
|
`$${data.total_cost_usd?.toFixed(4) || "0.0000"}`,
|
||||||
onStdout: (chunk) => {
|
String(data.usage?.input_tokens || 0),
|
||||||
// no logs
|
String(data.usage?.output_tokens || 0),
|
||||||
// process.stdout.write(chunk)
|
],
|
||||||
},
|
]);
|
||||||
onStderr: (chunk) => process.stderr.write(chunk),
|
} else if (data.subtype === "error_max_turns") {
|
||||||
});
|
log.error(`Max turns reached: ${JSON.stringify(data)}`);
|
||||||
|
} else if (data.subtype === "error_during_execution") {
|
||||||
if (result.exitCode !== 0) {
|
log.error(`Execution error: ${JSON.stringify(data)}`);
|
||||||
throw new Error(
|
} else {
|
||||||
`Installation failed with exit code ${result.exitCode}: ${result.stderr}`,
|
log.error(`Failed: ${JSON.stringify(data)}`);
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
core.info("Claude Code installed successfully");
|
|
||||||
} catch (error) {
|
|
||||||
throw new Error(`Failed to install Claude Code: ${error}`);
|
|
||||||
}
|
}
|
||||||
}
|
},
|
||||||
|
system: () => {},
|
||||||
/**
|
stream_event: () => {},
|
||||||
* Execute Claude Code with the given prompt
|
};
|
||||||
*/
|
|
||||||
async execute(prompt: string): Promise<AgentResult> {
|
|
||||||
core.info("Running Claude Code...");
|
|
||||||
// printTable([[prompt]]);
|
|
||||||
|
|
||||||
try {
|
|
||||||
// Execute Claude Code with the prompt directly using proper headless mode
|
|
||||||
// core.info(`Executing Claude Code with prompt: ${prompt.substring(0, 100)}...`);
|
|
||||||
|
|
||||||
const claudePath = `${process.env.HOME}/.local/bin/claude`;
|
|
||||||
// console.log("Using Claude Code from:", claudePath);
|
|
||||||
console.log(boxString(prompt, { title: "Prompt" }));
|
|
||||||
const args = [
|
|
||||||
"--print",
|
|
||||||
"--output-format",
|
|
||||||
"stream-json",
|
|
||||||
"--verbose",
|
|
||||||
"--permission-mode",
|
|
||||||
"acceptEdits",
|
|
||||||
];
|
|
||||||
const env = {
|
|
||||||
ANTHROPIC_API_KEY: this.apiKey,
|
|
||||||
};
|
|
||||||
|
|
||||||
// Start a collapsible log group for streaming output
|
|
||||||
core.startGroup("🔄 Run details");
|
|
||||||
|
|
||||||
// Initialize run statistics
|
|
||||||
this.runStats = {
|
|
||||||
toolsUsed: 0,
|
|
||||||
turns: 0,
|
|
||||||
startTime: Date.now(),
|
|
||||||
};
|
|
||||||
|
|
||||||
const finalResult = "";
|
|
||||||
const totalCost = 0;
|
|
||||||
|
|
||||||
// run Claude Code with the prompt
|
|
||||||
const result = await spawn({
|
|
||||||
cmd: claudePath,
|
|
||||||
args,
|
|
||||||
env,
|
|
||||||
input: prompt,
|
|
||||||
timeout: 10 * 60 * 1000, // 10 minutes
|
|
||||||
onStdout: (_chunk) => {
|
|
||||||
// console.log(chunk);
|
|
||||||
processJSONChunk(_chunk, this);
|
|
||||||
},
|
|
||||||
onStderr: (_chunk) => {
|
|
||||||
if (_chunk.trim()) {
|
|
||||||
// core.warning(`[warn] ${chunk}`);
|
|
||||||
processJSONChunk(_chunk, this);
|
|
||||||
}
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
// throw on non-zero exit code
|
|
||||||
if (result.exitCode !== 0) {
|
|
||||||
throw new Error(
|
|
||||||
`Command failed with exit code ${result.exitCode}\n\nStdout: ${result.stdout}\n\nStderr: ${result.stderr}`,
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Process the complete buffered stdout to extract final results
|
|
||||||
// if (result.stdout.trim()) {
|
|
||||||
// const lines = result.stdout.trim().split("\n");
|
|
||||||
// for (const line of lines) {
|
|
||||||
// if (line.trim()) {
|
|
||||||
// const chunkResult = processJsonChunk(line);
|
|
||||||
// if (chunkResult.finalResult) finalResult = chunkResult.finalResult;
|
|
||||||
// if (chunkResult.totalCost) totalCost = chunkResult.totalCost;
|
|
||||||
// }
|
|
||||||
// }
|
|
||||||
// }
|
|
||||||
|
|
||||||
// Log run summary
|
|
||||||
const duration = Date.now() - this.runStats.startTime;
|
|
||||||
core.info(
|
|
||||||
`📊 Run Summary: ${this.runStats.toolsUsed} tools used, ${this.runStats.turns} turns, ${duration}ms duration`,
|
|
||||||
);
|
|
||||||
|
|
||||||
core.info("✅ Task complete.");
|
|
||||||
core.endGroup(); // End the collapsible log group
|
|
||||||
|
|
||||||
return {
|
|
||||||
success: true,
|
|
||||||
output: finalResult,
|
|
||||||
metadata: {
|
|
||||||
promptLength: prompt.length,
|
|
||||||
exitCode: result.exitCode,
|
|
||||||
durationMs: result.durationMs,
|
|
||||||
totalCost,
|
|
||||||
},
|
|
||||||
};
|
|
||||||
} catch (error: any) {
|
|
||||||
// Ensure group is closed even if error occurs before group is started
|
|
||||||
try {
|
|
||||||
core.endGroup();
|
|
||||||
} catch {
|
|
||||||
// Group might not have been started, ignore
|
|
||||||
}
|
|
||||||
const errorMessage =
|
|
||||||
error instanceof Error ? error.message : "Unknown error";
|
|
||||||
return {
|
|
||||||
success: false,
|
|
||||||
error: `Failed to execute Claude Code: ${errorMessage}`,
|
|
||||||
};
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Process a JSON chunk line and extract result data
|
|
||||||
*/
|
|
||||||
// function processJsonChunk(line: string): { finalResult?: string; totalCost?: number } {
|
|
||||||
// try {
|
|
||||||
// const chunk = JSON.parse(line.trim());
|
|
||||||
// processJSONChunk(chunk);
|
|
||||||
|
|
||||||
// // Collect final result and cost data
|
|
||||||
// if (chunk.type === "result" && chunk.result) {
|
|
||||||
// return {
|
|
||||||
// finalResult: chunk.result,
|
|
||||||
// totalCost: chunk.total_cost_usd || 0,
|
|
||||||
// };
|
|
||||||
// }
|
|
||||||
// return {};
|
|
||||||
// } catch {
|
|
||||||
// core.debug(`Failed to parse JSON line: ${line}`);
|
|
||||||
// return {};
|
|
||||||
// }
|
|
||||||
// }
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Pretty print a JSON chunk based on its type
|
|
||||||
*/
|
|
||||||
function processJSONChunk(chunk: string, agent?: ClaudeAgent): void {
|
|
||||||
try {
|
|
||||||
// Parse the JSON string first
|
|
||||||
console.log(chunk);
|
|
||||||
const parsedChunk = JSON.parse(chunk.trim());
|
|
||||||
|
|
||||||
switch (parsedChunk.type) {
|
|
||||||
case "system":
|
|
||||||
if (parsedChunk.subtype === "init") {
|
|
||||||
core.info(`🚀 Starting Claude Code session...`);
|
|
||||||
// core.info(`📁 Working directory: ${parsedChunk.cwd}`);
|
|
||||||
// core.info(`🔑 Permission mode: ${parsedChunk.permissionMode}`);
|
|
||||||
core.info(
|
|
||||||
tableString([
|
|
||||||
["model", parsedChunk.model],
|
|
||||||
["cwd", parsedChunk.cwd],
|
|
||||||
["permission_mode", parsedChunk.permissionMode],
|
|
||||||
[
|
|
||||||
"tools",
|
|
||||||
parsedChunk.tools?.length
|
|
||||||
? `${parsedChunk.tools.length} tools`
|
|
||||||
: "none",
|
|
||||||
],
|
|
||||||
[
|
|
||||||
"mcp_servers",
|
|
||||||
parsedChunk.mcp_servers?.length
|
|
||||||
? `${parsedChunk.mcp_servers.length} servers`
|
|
||||||
: "none",
|
|
||||||
],
|
|
||||||
[
|
|
||||||
"slash_commands",
|
|
||||||
parsedChunk.slash_commands?.length
|
|
||||||
? `${parsedChunk.slash_commands.length} commands`
|
|
||||||
: "none",
|
|
||||||
],
|
|
||||||
]),
|
|
||||||
);
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
|
|
||||||
case "assistant":
|
|
||||||
if (parsedChunk.message?.content) {
|
|
||||||
// Track turns
|
|
||||||
if (agent) {
|
|
||||||
agent.runStats.turns++;
|
|
||||||
}
|
|
||||||
|
|
||||||
for (const content of parsedChunk.message.content) {
|
|
||||||
if (content.type === "text") {
|
|
||||||
// Skip empty text content
|
|
||||||
if (content.text.trim()) {
|
|
||||||
core.info(
|
|
||||||
boxString(content.text.trim(), { title: "Claude Code" }),
|
|
||||||
);
|
|
||||||
}
|
|
||||||
} else if (content.type === "tool_use") {
|
|
||||||
// Track tools used
|
|
||||||
if (agent) {
|
|
||||||
agent.runStats.toolsUsed++;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Enhanced tool usage logging
|
|
||||||
const toolName = content.name;
|
|
||||||
// const toolId = content.id;
|
|
||||||
|
|
||||||
core.info(`→ ${toolName}`);
|
|
||||||
|
|
||||||
// Log tool-specific details based on tool type
|
|
||||||
if (content.input) {
|
|
||||||
const input = content.input;
|
|
||||||
|
|
||||||
// Common tool input fields
|
|
||||||
if (input.description) {
|
|
||||||
core.info(` └─ ${input.description}`);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Tool-specific input fields
|
|
||||||
if (input.command) {
|
|
||||||
core.info(` └─ command: ${input.command}`);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (input.file_path) {
|
|
||||||
core.info(` └─ file: ${input.file_path}`);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (input.content) {
|
|
||||||
const contentPreview =
|
|
||||||
input.content.length > 100
|
|
||||||
? `${input.content.substring(0, 100)}...`
|
|
||||||
: input.content;
|
|
||||||
core.info(` └─ content: ${contentPreview}`);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (input.query) {
|
|
||||||
core.info(` └─ query: ${input.query}`);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (input.pattern) {
|
|
||||||
core.info(` └─ pattern: ${input.pattern}`);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (input.url) {
|
|
||||||
core.info(` └─ url: ${input.url}`);
|
|
||||||
}
|
|
||||||
|
|
||||||
// For multi-edit or complex operations
|
|
||||||
if (input.edits && Array.isArray(input.edits)) {
|
|
||||||
core.info(` └─ edits: ${input.edits.length} changes`);
|
|
||||||
input.edits.forEach((edit: any, index: number) => {
|
|
||||||
if (edit.file_path) {
|
|
||||||
core.info(` ${index + 1}. ${edit.file_path}`);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
// For task operations
|
|
||||||
if (input.task) {
|
|
||||||
core.info(` └─ task: ${input.task}`);
|
|
||||||
}
|
|
||||||
|
|
||||||
// For bash operations with specific details
|
|
||||||
if (input.bash_command) {
|
|
||||||
core.info(` └─ bash_command: ${input.bash_command}`);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Log tool ID for debugging
|
|
||||||
// core.debug(` 🔗 Tool ID: ${toolId}`);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
|
|
||||||
case "user":
|
|
||||||
if (parsedChunk.message?.content) {
|
|
||||||
for (const content of parsedChunk.message.content) {
|
|
||||||
if (content.type === "tool_result") {
|
|
||||||
if (content.is_error) {
|
|
||||||
core.warning(`❌ Tool error: ${content.content}`);
|
|
||||||
} else {
|
|
||||||
// Enhanced tool result logging
|
|
||||||
const _resultContent = content.content.trim();
|
|
||||||
// do nothing for now. usually useless in headless more.
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
|
|
||||||
case "result":
|
|
||||||
if (parsedChunk.subtype === "success") {
|
|
||||||
// Claude already prints something almost identical to this, so skip for now
|
|
||||||
// if (parsedChunk.result) {
|
|
||||||
// core.info(
|
|
||||||
// boxString(parsedChunk.result.trim(), {
|
|
||||||
// title: "🤖 Claude Code",
|
|
||||||
// maxWidth: 70,
|
|
||||||
// }),
|
|
||||||
// );
|
|
||||||
// }
|
|
||||||
|
|
||||||
core.info(
|
|
||||||
tableString([
|
|
||||||
[
|
|
||||||
"Cost",
|
|
||||||
`$${parsedChunk.total_cost_usd?.toFixed(4) || "0.0000"}`,
|
|
||||||
],
|
|
||||||
["Input Tokens", parsedChunk.usage?.input_tokens || 0],
|
|
||||||
["Output Tokens", parsedChunk.usage?.output_tokens || 0],
|
|
||||||
["Duration", `${parsedChunk.duration_ms}ms`],
|
|
||||||
["Turns", parsedChunk.num_turns || 1],
|
|
||||||
]),
|
|
||||||
);
|
|
||||||
} else {
|
|
||||||
core.error(`❌ Failed: ${parsedChunk.error || "Unknown error"}`);
|
|
||||||
}
|
|
||||||
break;
|
|
||||||
|
|
||||||
default:
|
|
||||||
// Log unknown chunk types for debugging
|
|
||||||
core.debug(`📦 Unknown chunk type: ${parsedChunk.type}`);
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
} catch (error) {
|
|
||||||
core.debug(`Failed to parse chunk: ${error}`);
|
|
||||||
core.debug(`Raw chunk: ${chunk.substring(0, 200)}...`);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|||||||
@@ -1,2 +0,0 @@
|
|||||||
export * from "./claude";
|
|
||||||
export * from "./types";
|
|
||||||
@@ -0,0 +1,63 @@
|
|||||||
|
import type { McpServerConfig } from "@anthropic-ai/claude-agent-sdk";
|
||||||
|
import { ghPullfrogMcpName } from "../mcp/config.ts";
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Result returned by agent execution
|
||||||
|
*/
|
||||||
|
export interface AgentResult {
|
||||||
|
success: boolean;
|
||||||
|
output?: string;
|
||||||
|
error?: string;
|
||||||
|
metadata?: Record<string, unknown>;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Configuration for agent creation
|
||||||
|
*/
|
||||||
|
export interface AgentConfig {
|
||||||
|
apiKey: string;
|
||||||
|
githubInstallationToken: string;
|
||||||
|
prompt: string;
|
||||||
|
mcpServers: Record<string, McpServerConfig>;
|
||||||
|
}
|
||||||
|
|
||||||
|
export type Agent = {
|
||||||
|
run: (config: AgentConfig) => Promise<AgentResult>;
|
||||||
|
};
|
||||||
|
|
||||||
|
export const instructions = `
|
||||||
|
You are a highly intelligent, no-nonsense senior-level software engineering agent. You are careful, to-the-point, and kind. You only say things you know to be true. Your code is focused, minimal, and production-ready. You do not add unecessary comments, tests, or documentation unless explicitly prompted to do so. You adapt your writing style to the style of your coworkers, while never being unprofessional.
|
||||||
|
|
||||||
|
- eagerly inspect your MCP servers to determine what tools are available to you, especially ${ghPullfrogMcpName}
|
||||||
|
- do not under any circumstances use the github cli (\`gh\`). find the corresponding tool from ${ghPullfrogMcpName} instead.
|
||||||
|
- mode selection: choose the appropriate mode based on the prompt payload:
|
||||||
|
- choose "plan mode" if the prompt asks to:
|
||||||
|
- create a plan, break down tasks, outline steps, or analyze requirements
|
||||||
|
- understand the scope of work before implementation
|
||||||
|
- provide a todo list or task breakdown
|
||||||
|
- choose "implement" if the prompt asks to:
|
||||||
|
- implement, build, create, or develop code changes
|
||||||
|
- make specific changes to files or features
|
||||||
|
- execute a plan that was previously created
|
||||||
|
- the prompt includes specific implementation details or requirements
|
||||||
|
- choose "review" if the prompt asks to:
|
||||||
|
- review code, PR, or implementation
|
||||||
|
- provide feedback, suggestions, or identify issues
|
||||||
|
- check code quality, style, or correctness
|
||||||
|
- once you've chosen a mode, follow its associated prompts carefully
|
||||||
|
- when prompted directly (e.g., via issue comment or PR comment):
|
||||||
|
(1) start by creating a single response comment using mcp__${ghPullfrogMcpName}__create_issue_comment
|
||||||
|
- the initial comment should say something like "I'll do {summary of request}" where you summarize what was requested
|
||||||
|
- save the commentId returned from this initial comment creation
|
||||||
|
(2) use mcp__${ghPullfrogMcpName}__edit_issue_comment to progressively update that same comment as you make progress
|
||||||
|
- update the comment with current status, completed tasks, and any relevant information
|
||||||
|
- continue updating the same comment throughout the planning/implementation process
|
||||||
|
(3) create_issue_comment should only be used once initially - all subsequent updates must use edit_issue_comment with the saved commentId
|
||||||
|
- if prompted to review a PR:
|
||||||
|
(1) get PR info with mcp__${ghPullfrogMcpName}__get_pull_request (this automatically prepares the repository by fetching and checking out the PR branch)
|
||||||
|
(2) view diff: git diff origin/<base>...origin/<head> (use line numbers from this for inline comments)
|
||||||
|
(3) read files from the checked-out PR branch to understand the implementation
|
||||||
|
(4) when submitting review: use the 'comments' array for ALL specific code issues - include the file path and line position from the diff
|
||||||
|
(5) only use the 'body' field for a brief summary (1-2 sentences) or for feedback that doesn't apply to a specific code location
|
||||||
|
replace <base> and <head> with 'base' and 'head' from the PR info
|
||||||
|
`;
|
||||||
@@ -1,34 +0,0 @@
|
|||||||
/**
|
|
||||||
* Standard interface for all Pullfrog agents
|
|
||||||
*/
|
|
||||||
export interface Agent {
|
|
||||||
/**
|
|
||||||
* Install the agent and any required dependencies
|
|
||||||
*/
|
|
||||||
install(): Promise<void>;
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Execute the agent with the given prompt
|
|
||||||
* @param prompt The prompt to send to the agent
|
|
||||||
* @param options Additional options specific to the agent
|
|
||||||
*/
|
|
||||||
execute(prompt: string, options?: Record<string, any>): Promise<AgentResult>;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Result returned by agent execution
|
|
||||||
*/
|
|
||||||
export interface AgentResult {
|
|
||||||
success: boolean;
|
|
||||||
output?: string;
|
|
||||||
error?: string;
|
|
||||||
metadata?: Record<string, any>;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Configuration for agent creation
|
|
||||||
*/
|
|
||||||
export interface AgentConfig {
|
|
||||||
apiKey?: string;
|
|
||||||
[key: string]: any;
|
|
||||||
}
|
|
||||||
@@ -0,0 +1,39 @@
|
|||||||
|
#!/usr/bin/env node
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Build entry point - wraps entry.ts to avoid top-level await in CJS output
|
||||||
|
*/
|
||||||
|
|
||||||
|
import * as core from "@actions/core";
|
||||||
|
import { type } from "arktype";
|
||||||
|
import { Inputs, main } from "./main.ts";
|
||||||
|
import packageJson from "./package.json" with { type: "json" };
|
||||||
|
import { log } from "./utils/cli.ts";
|
||||||
|
|
||||||
|
async function run(): Promise<void> {
|
||||||
|
try {
|
||||||
|
log.info(`🐸 Running pullfrog/action@${packageJson.version}...`);
|
||||||
|
|
||||||
|
const inputsJson = process.env.INPUTS_JSON;
|
||||||
|
if (!inputsJson) {
|
||||||
|
throw new Error("INPUTS_JSON environment variable not found");
|
||||||
|
}
|
||||||
|
|
||||||
|
const parsed = type("string.json.parse").assert(inputsJson);
|
||||||
|
const inputs = Inputs.assert(parsed);
|
||||||
|
|
||||||
|
const result = await main(inputs);
|
||||||
|
|
||||||
|
if (!result.success) {
|
||||||
|
throw new Error(result.error || "Agent execution failed");
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
const errorMessage = error instanceof Error ? error.message : "Unknown error occurred";
|
||||||
|
core.setFailed(`Action failed: ${errorMessage}`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Wrap in IIFE to avoid top-level await in CJS
|
||||||
|
(async () => {
|
||||||
|
await run();
|
||||||
|
})();
|
||||||
@@ -2,54 +2,27 @@
|
|||||||
|
|
||||||
/**
|
/**
|
||||||
* Entry point for GitHub Action
|
* Entry point for GitHub Action
|
||||||
* This file is bundled to entry.cjs and called directly by GitHub Actions
|
|
||||||
*/
|
*/
|
||||||
|
|
||||||
import * as core from "@actions/core";
|
import * as core from "@actions/core";
|
||||||
import { main } from "./main";
|
import { type } from "arktype";
|
||||||
import { setupGitHubInstallationToken } from "./utils";
|
import { Inputs, main } from "./main.ts";
|
||||||
|
import packageJson from "./package.json" with { type: "json" };
|
||||||
|
import { log } from "./utils/cli.ts";
|
||||||
|
|
||||||
async function run(): Promise<void> {
|
async function run(): Promise<void> {
|
||||||
try {
|
try {
|
||||||
// Get inputs from GitHub Actions
|
log.info(`🐸 Running pullfrog/action@${packageJson.version}...`);
|
||||||
const prompt = core.getInput("prompt", { required: true });
|
|
||||||
const anthropicApiKey = core.getInput("anthropic_api_key");
|
|
||||||
|
|
||||||
if (!prompt) {
|
const inputsJson = process.env.INPUTS_JSON;
|
||||||
throw new Error("prompt is required");
|
if (!inputsJson) {
|
||||||
|
throw new Error("INPUTS_JSON environment variable not found");
|
||||||
}
|
}
|
||||||
|
|
||||||
// Create params object with new structure
|
const parsed = type("string.json.parse").assert(inputsJson);
|
||||||
const inputs: any = {
|
const inputs = Inputs.assert(parsed);
|
||||||
prompt,
|
|
||||||
anthropic_api_key: anthropicApiKey,
|
|
||||||
};
|
|
||||||
|
|
||||||
// Add optional properties only if they exist
|
const result = await main(inputs);
|
||||||
const githubToken = core.getInput("github_token") || process.env.GITHUB_TOKEN;
|
|
||||||
if (githubToken) {
|
|
||||||
inputs.github_token = githubToken;
|
|
||||||
}
|
|
||||||
|
|
||||||
const githubInstallationToken =
|
|
||||||
core.getInput("github_installation_token") || process.env.GITHUB_INSTALLATION_TOKEN;
|
|
||||||
if (githubInstallationToken) {
|
|
||||||
inputs.github_installation_token = githubInstallationToken;
|
|
||||||
} else {
|
|
||||||
// Setup GitHub installation token
|
|
||||||
await setupGitHubInstallationToken();
|
|
||||||
}
|
|
||||||
|
|
||||||
const params = {
|
|
||||||
inputs,
|
|
||||||
env: {} as Record<string, string>,
|
|
||||||
cwd: process.cwd(),
|
|
||||||
};
|
|
||||||
|
|
||||||
// Run the main function
|
|
||||||
const result = await main(params);
|
|
||||||
|
|
||||||
// TODO: Set outputs
|
|
||||||
|
|
||||||
if (!result.success) {
|
if (!result.success) {
|
||||||
throw new Error(result.error || "Agent execution failed");
|
throw new Error(result.error || "Agent execution failed");
|
||||||
@@ -60,8 +33,4 @@ async function run(): Promise<void> {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Run the action
|
await run();
|
||||||
run().catch((error) => {
|
|
||||||
console.error("Action failed:", error);
|
|
||||||
process.exit(1);
|
|
||||||
});
|
|
||||||
|
|||||||
+10
-2
@@ -3,14 +3,22 @@ import { build } from "esbuild";
|
|||||||
// Build the GitHub Action bundle only
|
// Build the GitHub Action bundle only
|
||||||
// For npm package builds, use zshy (pnpm build:npm)
|
// For npm package builds, use zshy (pnpm build:npm)
|
||||||
await build({
|
await build({
|
||||||
entryPoints: ["./entry.ts"],
|
entryPoints: ["./entry.build.ts"],
|
||||||
bundle: true,
|
bundle: true,
|
||||||
outfile: "./entry.cjs",
|
outfile: "./entry.cjs",
|
||||||
format: "cjs",
|
format: "cjs",
|
||||||
platform: "node",
|
platform: "node",
|
||||||
target: "node20",
|
target: "node20",
|
||||||
minify: false,
|
minify: true,
|
||||||
sourcemap: false,
|
sourcemap: false,
|
||||||
|
// @actions/core is provided by GitHub Actions runtime, but we still need it bundled
|
||||||
|
// for local testing. However, we can mark it to reduce duplication if needed.
|
||||||
|
external: [],
|
||||||
|
// Enable tree-shaking to remove unused code
|
||||||
|
treeShaking: true,
|
||||||
|
// Drop console statements in production (but keep for debugging)
|
||||||
|
drop: [],
|
||||||
});
|
});
|
||||||
|
|
||||||
console.log("✅ Build completed successfully!");
|
console.log("✅ Build completed successfully!");
|
||||||
|
|
||||||
|
|||||||
+5
-9
@@ -1,13 +1,9 @@
|
|||||||
import type { MainParams } from "../main";
|
import type { Inputs } from "../main.ts";
|
||||||
|
|
||||||
const testParams = {
|
const testParams = {
|
||||||
inputs: {
|
prompt:
|
||||||
prompt:
|
"List all files in the current directory, then create a file called dynamic-test.txt with the content 'This was loaded from a TypeScript file!', then delete it.",
|
||||||
"List all files in the current directory, then create a file called dynamic-test.txt with the content 'This was loaded from a TypeScript file!', then delete it.",
|
anthropic_api_key: "sk-test-key",
|
||||||
anthropic_api_key: "sk-test-key",
|
} satisfies Inputs;
|
||||||
},
|
|
||||||
env: {},
|
|
||||||
cwd: process.cwd(),
|
|
||||||
} satisfies MainParams;
|
|
||||||
|
|
||||||
export default testParams;
|
export default testParams;
|
||||||
|
|||||||
+1
-1
@@ -1 +1 @@
|
|||||||
Print the list of tools available. Then create a new file called test.txt with the content "Hello from Pullfrog!".
|
run npx cowsay "don't eat me"
|
||||||
@@ -3,11 +3,9 @@
|
|||||||
* This exports the main function for programmatic usage
|
* This exports the main function for programmatic usage
|
||||||
*/
|
*/
|
||||||
|
|
||||||
export { ClaudeAgent } from "./agents";
|
export type { Agent, AgentConfig, AgentResult } from "./agents/shared.ts";
|
||||||
export type { Agent, AgentConfig, AgentResult } from "./agents/types";
|
|
||||||
export {
|
export {
|
||||||
type ExecutionInputs,
|
type Inputs as ExecutionInputs,
|
||||||
type MainParams,
|
|
||||||
type MainResult,
|
type MainResult,
|
||||||
main,
|
main,
|
||||||
} from "./main";
|
} from "./main.ts";
|
||||||
|
|||||||
@@ -1,25 +1,16 @@
|
|||||||
import * as core from "@actions/core";
|
import { type } from "arktype";
|
||||||
import { ClaudeAgent } from "./agents";
|
import { claude } from "./agents/claude.ts";
|
||||||
|
import { createMcpConfigs } from "./mcp/config.ts";
|
||||||
|
import { log } from "./utils/cli.ts";
|
||||||
|
import { parseRepoContext, setupGitHubInstallationToken } from "./utils/github.ts";
|
||||||
|
import { setupGitAuth, setupGitConfig } from "./utils/setup.ts";
|
||||||
|
|
||||||
// Expected environment variables that should be passed as inputs
|
export const Inputs = type({
|
||||||
export const EXPECTED_INPUTS: string[] = [
|
prompt: "string",
|
||||||
"ANTHROPIC_API_KEY",
|
"anthropic_api_key?": "string | undefined",
|
||||||
"GITHUB_TOKEN",
|
});
|
||||||
"GITHUB_INSTALLATION_TOKEN"
|
|
||||||
];
|
|
||||||
|
|
||||||
export interface ExecutionInputs {
|
export type Inputs = typeof Inputs.infer;
|
||||||
prompt: string;
|
|
||||||
anthropic_api_key: string;
|
|
||||||
github_token?: string;
|
|
||||||
github_installation_token?: string;
|
|
||||||
}
|
|
||||||
|
|
||||||
export interface MainParams {
|
|
||||||
inputs: ExecutionInputs;
|
|
||||||
env: Record<string, string>;
|
|
||||||
cwd: string;
|
|
||||||
}
|
|
||||||
|
|
||||||
export interface MainResult {
|
export interface MainResult {
|
||||||
success: boolean;
|
success: boolean;
|
||||||
@@ -27,28 +18,36 @@ export interface MainResult {
|
|||||||
error?: string | undefined;
|
error?: string | undefined;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
export type PromptJSON = {};
|
||||||
|
|
||||||
export async function main(params: MainParams): Promise<MainResult> {
|
export async function main(inputs: Inputs): Promise<MainResult> {
|
||||||
try {
|
try {
|
||||||
// Extract inputs from params
|
log.info("Starting agent run...");
|
||||||
const { inputs, env, cwd } = params;
|
|
||||||
|
|
||||||
// Set working directory if different from current
|
setupGitConfig();
|
||||||
if (cwd !== process.cwd()) {
|
|
||||||
process.chdir(cwd);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Set environment variables
|
const githubInstallationToken = await setupGitHubInstallationToken();
|
||||||
Object.assign(process.env, env);
|
const repoContext = parseRepoContext();
|
||||||
|
|
||||||
core.info(`→ Starting agent run with Claude Code`);
|
setupGitAuth(githubInstallationToken, repoContext);
|
||||||
|
|
||||||
// Create and install the Claude agent
|
const mcpServers = createMcpConfigs(githubInstallationToken);
|
||||||
const agent = new ClaudeAgent({ apiKey: inputs.anthropic_api_key });
|
|
||||||
await agent.install();
|
|
||||||
|
|
||||||
// Execute the agent with the prompt
|
log.debug(`📋 MCP Config: ${JSON.stringify(mcpServers, null, 2)}`);
|
||||||
const result = await agent.execute(inputs.prompt);
|
|
||||||
|
log.info("Running Claude Agent SDK...");
|
||||||
|
log.box(inputs.prompt, { title: "Prompt" });
|
||||||
|
|
||||||
|
// TODO: check if `inputs.prompts` is JSON
|
||||||
|
// if yes, check if it's a webhook payload or toJSON(github.event)
|
||||||
|
// for webhook payloads, check the specified `agent` field
|
||||||
|
|
||||||
|
const result = await claude.run({
|
||||||
|
prompt: inputs.prompt,
|
||||||
|
mcpServers,
|
||||||
|
githubInstallationToken,
|
||||||
|
apiKey: inputs.anthropic_api_key!,
|
||||||
|
});
|
||||||
|
|
||||||
if (!result.success) {
|
if (!result.success) {
|
||||||
return {
|
return {
|
||||||
@@ -58,12 +57,17 @@ export async function main(params: MainParams): Promise<MainResult> {
|
|||||||
};
|
};
|
||||||
}
|
}
|
||||||
|
|
||||||
|
log.success("Task complete.");
|
||||||
|
await log.writeSummary();
|
||||||
|
|
||||||
return {
|
return {
|
||||||
success: true,
|
success: true,
|
||||||
output: result.output || "",
|
output: result.output || "",
|
||||||
};
|
};
|
||||||
} catch (error) {
|
} catch (error) {
|
||||||
const errorMessage = error instanceof Error ? error.message : "Unknown error occurred";
|
const errorMessage = error instanceof Error ? error.message : "Unknown error occurred";
|
||||||
|
log.error(errorMessage);
|
||||||
|
await log.writeSummary();
|
||||||
return {
|
return {
|
||||||
success: false,
|
success: false,
|
||||||
error: errorMessage,
|
error: errorMessage,
|
||||||
|
|||||||
@@ -0,0 +1,55 @@
|
|||||||
|
import { type } from "arktype";
|
||||||
|
import { contextualize, tool } from "./shared.ts";
|
||||||
|
|
||||||
|
export const Comment = type({
|
||||||
|
issueNumber: type.number.describe("the issue number to comment on"),
|
||||||
|
body: type.string.describe("the comment body content"),
|
||||||
|
});
|
||||||
|
|
||||||
|
export const CreateCommentTool = tool({
|
||||||
|
name: "create_issue_comment",
|
||||||
|
description: "Create a comment on a GitHub issue",
|
||||||
|
parameters: Comment,
|
||||||
|
execute: contextualize(async ({ issueNumber, body }, ctx) => {
|
||||||
|
const result = await ctx.octokit.rest.issues.createComment({
|
||||||
|
owner: ctx.owner,
|
||||||
|
repo: ctx.name,
|
||||||
|
issue_number: issueNumber,
|
||||||
|
body: body,
|
||||||
|
});
|
||||||
|
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
commentId: result.data.id,
|
||||||
|
url: result.data.html_url,
|
||||||
|
body: result.data.body,
|
||||||
|
};
|
||||||
|
}),
|
||||||
|
});
|
||||||
|
|
||||||
|
export const EditComment = type({
|
||||||
|
commentId: type.number.describe("the ID of the comment to edit"),
|
||||||
|
body: type.string.describe("the new comment body content"),
|
||||||
|
});
|
||||||
|
|
||||||
|
export const EditCommentTool = tool({
|
||||||
|
name: "edit_issue_comment",
|
||||||
|
description: "Edit a GitHub issue comment by its ID",
|
||||||
|
parameters: EditComment,
|
||||||
|
execute: contextualize(async ({ commentId, body }, ctx) => {
|
||||||
|
const result = await ctx.octokit.rest.issues.updateComment({
|
||||||
|
owner: ctx.owner,
|
||||||
|
repo: ctx.name,
|
||||||
|
comment_id: commentId,
|
||||||
|
body: body,
|
||||||
|
});
|
||||||
|
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
commentId: result.data.id,
|
||||||
|
url: result.data.html_url,
|
||||||
|
body: result.data.body,
|
||||||
|
updatedAt: result.data.updated_at,
|
||||||
|
};
|
||||||
|
}),
|
||||||
|
});
|
||||||
@@ -0,0 +1,31 @@
|
|||||||
|
/**
|
||||||
|
* Simple MCP configuration helper for adding our minimal GitHub comment server
|
||||||
|
*/
|
||||||
|
|
||||||
|
import type { McpServerConfig } from "@anthropic-ai/claude-agent-sdk";
|
||||||
|
import { fromHere } from "@ark/fs";
|
||||||
|
import { parseRepoContext } from "../utils/github.ts";
|
||||||
|
|
||||||
|
const actionPath = fromHere("..");
|
||||||
|
|
||||||
|
export const ghPullfrogMcpName = "gh-pullfrog";
|
||||||
|
|
||||||
|
export type McpName = typeof ghPullfrogMcpName;
|
||||||
|
|
||||||
|
export type McpConfigs = Record<McpName, McpServerConfig>;
|
||||||
|
|
||||||
|
export function createMcpConfigs(githubInstallationToken: string): McpConfigs {
|
||||||
|
const repoContext = parseRepoContext();
|
||||||
|
const githubRepository = `${repoContext.owner}/${repoContext.name}`;
|
||||||
|
|
||||||
|
return {
|
||||||
|
[ghPullfrogMcpName]: {
|
||||||
|
command: "node",
|
||||||
|
args: [`${actionPath}/mcp/server.ts`],
|
||||||
|
env: {
|
||||||
|
GITHUB_INSTALLATION_TOKEN: githubInstallationToken,
|
||||||
|
GITHUB_REPOSITORY: githubRepository,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
};
|
||||||
|
}
|
||||||
@@ -0,0 +1,42 @@
|
|||||||
|
import { type } from "arktype";
|
||||||
|
import { contextualize, tool } from "./shared.ts";
|
||||||
|
|
||||||
|
export const Issue = type({
|
||||||
|
title: type.string.describe("the title of the issue"),
|
||||||
|
body: type.string.describe("the body content of the issue"),
|
||||||
|
labels: type.string
|
||||||
|
.array()
|
||||||
|
.describe("optional array of label names to apply to the issue")
|
||||||
|
.optional(),
|
||||||
|
assignees: type.string
|
||||||
|
.array()
|
||||||
|
.describe("optional array of usernames to assign to the issue")
|
||||||
|
.optional(),
|
||||||
|
});
|
||||||
|
|
||||||
|
export const IssueTool = tool({
|
||||||
|
name: "create_issue",
|
||||||
|
description: "Create a new GitHub issue",
|
||||||
|
parameters: Issue,
|
||||||
|
execute: contextualize(async ({ title, body, labels, assignees }, ctx) => {
|
||||||
|
const result = await ctx.octokit.rest.issues.create({
|
||||||
|
owner: ctx.owner,
|
||||||
|
repo: ctx.name,
|
||||||
|
title: title,
|
||||||
|
body: body,
|
||||||
|
labels: labels ?? [],
|
||||||
|
assignees: assignees ?? [],
|
||||||
|
});
|
||||||
|
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
issueId: result.data.id,
|
||||||
|
number: result.data.number,
|
||||||
|
url: result.data.html_url,
|
||||||
|
title: result.data.title,
|
||||||
|
state: result.data.state,
|
||||||
|
labels: result.data.labels?.map((label) => (typeof label === "string" ? label : label.name)),
|
||||||
|
assignees: result.data.assignees?.map((assignee) => assignee.login),
|
||||||
|
};
|
||||||
|
}),
|
||||||
|
});
|
||||||
@@ -0,0 +1,43 @@
|
|||||||
|
import { execSync } from "node:child_process";
|
||||||
|
import { type } from "arktype";
|
||||||
|
import { log } from "../utils/cli.ts";
|
||||||
|
import { contextualize, tool } from "./shared.ts";
|
||||||
|
|
||||||
|
export const PullRequest = type({
|
||||||
|
title: type.string.describe("the title of the pull request"),
|
||||||
|
body: type.string.describe("the body content of the pull request"),
|
||||||
|
base: type.string.describe("the base branch to merge into (e.g., 'main')"),
|
||||||
|
});
|
||||||
|
|
||||||
|
export const PullRequestTool = tool({
|
||||||
|
name: "create_pull_request",
|
||||||
|
description: "Create a pull request from the current branch",
|
||||||
|
parameters: PullRequest,
|
||||||
|
execute: contextualize(async ({ title, body, base }, ctx) => {
|
||||||
|
// Get the current branch name
|
||||||
|
const currentBranch = execSync("git rev-parse --abbrev-ref HEAD", {
|
||||||
|
encoding: "utf8",
|
||||||
|
}).trim();
|
||||||
|
|
||||||
|
log.info(`Current branch: ${currentBranch}`);
|
||||||
|
|
||||||
|
const result = await ctx.octokit.rest.pulls.create({
|
||||||
|
owner: ctx.owner,
|
||||||
|
repo: ctx.name,
|
||||||
|
title: title,
|
||||||
|
body: body,
|
||||||
|
head: currentBranch,
|
||||||
|
base: base,
|
||||||
|
});
|
||||||
|
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
pullRequestId: result.data.id,
|
||||||
|
number: result.data.number,
|
||||||
|
url: result.data.html_url,
|
||||||
|
title: result.data.title,
|
||||||
|
head: result.data.head.ref,
|
||||||
|
base: result.data.base.ref,
|
||||||
|
};
|
||||||
|
}),
|
||||||
|
});
|
||||||
@@ -0,0 +1,52 @@
|
|||||||
|
import { execSync } from "node:child_process";
|
||||||
|
import { type } from "arktype";
|
||||||
|
import { log } from "../utils/cli.ts";
|
||||||
|
import { contextualize, tool } from "./shared.ts";
|
||||||
|
|
||||||
|
export const PullRequestInfo = type({
|
||||||
|
pull_number: type.number.describe("The pull request number to fetch"),
|
||||||
|
});
|
||||||
|
|
||||||
|
export const PullRequestInfoTool = tool({
|
||||||
|
name: "get_pull_request",
|
||||||
|
description:
|
||||||
|
"Retrieve PR information and automatically prepare the repository for review by fetching and checking out the PR branch.",
|
||||||
|
parameters: PullRequestInfo,
|
||||||
|
execute: contextualize(async ({ pull_number }, ctx) => {
|
||||||
|
const pr = await ctx.octokit.rest.pulls.get({
|
||||||
|
owner: ctx.owner,
|
||||||
|
repo: ctx.name,
|
||||||
|
pull_number,
|
||||||
|
});
|
||||||
|
|
||||||
|
const data = pr.data;
|
||||||
|
|
||||||
|
const baseBranch = data.base.ref;
|
||||||
|
const headBranch = data.head.ref;
|
||||||
|
|
||||||
|
if (!baseBranch) {
|
||||||
|
throw new Error(`Base branch not found for PR #${pull_number}`);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Automatically fetch and checkout branches for review
|
||||||
|
log.info(`Fetching base branch: origin/${baseBranch}`);
|
||||||
|
execSync(`git fetch origin ${baseBranch} --depth=20`, { stdio: "inherit" });
|
||||||
|
|
||||||
|
log.info(`Fetching PR branch: origin/${headBranch}`);
|
||||||
|
execSync(`git fetch origin ${headBranch}`, { stdio: "inherit" });
|
||||||
|
|
||||||
|
log.info(`Checking out PR branch: origin/${headBranch}`);
|
||||||
|
execSync(`git checkout origin/${headBranch}`, { stdio: "inherit" });
|
||||||
|
|
||||||
|
return {
|
||||||
|
number: data.number,
|
||||||
|
url: data.html_url,
|
||||||
|
title: data.title,
|
||||||
|
state: data.state,
|
||||||
|
draft: data.draft,
|
||||||
|
merged: data.merged,
|
||||||
|
base: baseBranch,
|
||||||
|
head: headBranch,
|
||||||
|
};
|
||||||
|
}),
|
||||||
|
});
|
||||||
@@ -0,0 +1,94 @@
|
|||||||
|
import type { RestEndpointMethodTypes } from "@octokit/rest";
|
||||||
|
import { type } from "arktype";
|
||||||
|
import { contextualize, tool } from "./shared.ts";
|
||||||
|
|
||||||
|
export const Review = type({
|
||||||
|
pull_number: type.number.describe("The pull request number to review"),
|
||||||
|
event: type
|
||||||
|
.enumerated("APPROVE", "REQUEST_CHANGES", "COMMENT")
|
||||||
|
.describe("'APPROVE', 'REQUEST_CHANGES', or 'COMMENT' (the review action)"),
|
||||||
|
body: type.string
|
||||||
|
.describe(
|
||||||
|
"Brief summary or general feedback that doesn't apply to specific code locations. Keep it concise - most feedback should be in the 'comments' array."
|
||||||
|
)
|
||||||
|
.optional(),
|
||||||
|
commit_id: type.string
|
||||||
|
.describe("Optional SHA of the commit being reviewed. Defaults to latest.")
|
||||||
|
.optional(),
|
||||||
|
comments: type({
|
||||||
|
path: type.string.describe("The file path to comment on (relative to repo root)"),
|
||||||
|
line: type.number.describe(
|
||||||
|
"The line number in the file (use line numbers from the diff - usually the RIGHT side/new code)"
|
||||||
|
),
|
||||||
|
side: type
|
||||||
|
.enumerated("LEFT", "RIGHT")
|
||||||
|
.describe(
|
||||||
|
"Side of the diff: LEFT (old code) or RIGHT (new code). Defaults to RIGHT if not provided."
|
||||||
|
)
|
||||||
|
.optional(),
|
||||||
|
body: type.string.describe("The comment text for this specific line"),
|
||||||
|
start_line: type.number
|
||||||
|
.describe("Start line for multi-line comments (optional, for commenting on ranges)")
|
||||||
|
.optional(),
|
||||||
|
})
|
||||||
|
.array()
|
||||||
|
.describe(
|
||||||
|
"REQUIRED: Array of inline comments for specific code issues. Use this for all location-specific feedback. Use 'git diff origin/<base>...origin/<head>' to find the correct line numbers (typically use the line numbers shown on the RIGHT side for new code, LEFT side for old code)."
|
||||||
|
)
|
||||||
|
.optional(),
|
||||||
|
});
|
||||||
|
|
||||||
|
export const ReviewTool = tool({
|
||||||
|
name: "submit_pull_request_review",
|
||||||
|
description:
|
||||||
|
"Submit a review (approve, request changes, or comment) for an existing pull request. " +
|
||||||
|
"IMPORTANT: Use 'comments' array for ALL specific code issues at the line-level. " +
|
||||||
|
"Only use 'body' for a brief summary or feedback that doesn't apply to a specific location.",
|
||||||
|
parameters: Review,
|
||||||
|
execute: contextualize(async ({ pull_number, event, body, commit_id, comments = [] }, ctx) => {
|
||||||
|
// Get the PR to determine the head commit if commit_id not provided
|
||||||
|
const pr = await ctx.octokit.rest.pulls.get({
|
||||||
|
owner: ctx.owner,
|
||||||
|
repo: ctx.name,
|
||||||
|
pull_number,
|
||||||
|
});
|
||||||
|
|
||||||
|
// Compose the request
|
||||||
|
const params: RestEndpointMethodTypes["pulls"]["createReview"]["parameters"] = {
|
||||||
|
owner: ctx.owner,
|
||||||
|
repo: ctx.name,
|
||||||
|
pull_number,
|
||||||
|
event,
|
||||||
|
};
|
||||||
|
if (body) params.body = body;
|
||||||
|
if (commit_id) {
|
||||||
|
params.commit_id = commit_id;
|
||||||
|
} else {
|
||||||
|
params.commit_id = pr.data.head.sha;
|
||||||
|
}
|
||||||
|
if (comments.length > 0) {
|
||||||
|
type ReviewComment = (typeof params.comments & {})[number];
|
||||||
|
// Convert comments to the format expected by GitHub API
|
||||||
|
params.comments = comments.map((comment) => {
|
||||||
|
const reviewComment: ReviewComment = {
|
||||||
|
...comment,
|
||||||
|
};
|
||||||
|
reviewComment.side = comment.side || "RIGHT";
|
||||||
|
if (comment.start_line) {
|
||||||
|
reviewComment.start_line = comment.start_line;
|
||||||
|
reviewComment.start_side = comment.side || "RIGHT";
|
||||||
|
}
|
||||||
|
return reviewComment;
|
||||||
|
});
|
||||||
|
}
|
||||||
|
const result = await ctx.octokit.rest.pulls.createReview(params);
|
||||||
|
return {
|
||||||
|
success: true,
|
||||||
|
reviewId: result.data.id,
|
||||||
|
html_url: result.data.html_url,
|
||||||
|
state: result.data.state,
|
||||||
|
user: result.data.user?.login,
|
||||||
|
submitted_at: result.data.submitted_at,
|
||||||
|
};
|
||||||
|
}),
|
||||||
|
});
|
||||||
@@ -0,0 +1,25 @@
|
|||||||
|
#!/usr/bin/env node
|
||||||
|
// Minimal GitHub Issue Comment MCP Server
|
||||||
|
import { FastMCP } from "fastmcp";
|
||||||
|
import { CreateCommentTool, EditCommentTool } from "./comment.ts";
|
||||||
|
import { IssueTool } from "./issue.ts";
|
||||||
|
import { PullRequestTool } from "./pr.ts";
|
||||||
|
import { PullRequestInfoTool } from "./prInfo.ts";
|
||||||
|
import { ReviewTool } from "./review.ts";
|
||||||
|
import { addTools } from "./shared.ts";
|
||||||
|
|
||||||
|
const server = new FastMCP({
|
||||||
|
name: "gh-pullfrog",
|
||||||
|
version: "0.0.1",
|
||||||
|
});
|
||||||
|
|
||||||
|
addTools(server, [
|
||||||
|
CreateCommentTool,
|
||||||
|
EditCommentTool,
|
||||||
|
IssueTool,
|
||||||
|
PullRequestTool,
|
||||||
|
ReviewTool,
|
||||||
|
PullRequestInfoTool,
|
||||||
|
]);
|
||||||
|
|
||||||
|
server.start();
|
||||||
@@ -0,0 +1,76 @@
|
|||||||
|
import { cached } from "@ark/util";
|
||||||
|
import { Octokit } from "@octokit/rest";
|
||||||
|
import type { StandardSchemaV1 } from "@standard-schema/spec";
|
||||||
|
import type { FastMCP, Tool } from "fastmcp";
|
||||||
|
import { parseRepoContext, type RepoContext } from "../utils/github.ts";
|
||||||
|
|
||||||
|
export interface ToolResult {
|
||||||
|
content: {
|
||||||
|
type: "text";
|
||||||
|
text: string;
|
||||||
|
}[];
|
||||||
|
isError?: boolean;
|
||||||
|
}
|
||||||
|
|
||||||
|
export const getMcpContext = cached((): McpContext => {
|
||||||
|
const githubInstallationToken = process.env.GITHUB_INSTALLATION_TOKEN;
|
||||||
|
if (!githubInstallationToken) {
|
||||||
|
throw new Error("GITHUB_INSTALLATION_TOKEN environment variable is required");
|
||||||
|
}
|
||||||
|
|
||||||
|
return {
|
||||||
|
...parseRepoContext(),
|
||||||
|
octokit: new Octokit({
|
||||||
|
auth: githubInstallationToken,
|
||||||
|
}),
|
||||||
|
};
|
||||||
|
});
|
||||||
|
|
||||||
|
export interface McpContext extends RepoContext {
|
||||||
|
octokit: Octokit;
|
||||||
|
}
|
||||||
|
|
||||||
|
export const tool = <const params>(tool: Tool<any, StandardSchemaV1<params>>) => tool;
|
||||||
|
|
||||||
|
export const addTools = (server: FastMCP, tools: Tool<any, any>[]) => {
|
||||||
|
for (const tool of tools) {
|
||||||
|
server.addTool(tool);
|
||||||
|
}
|
||||||
|
return server;
|
||||||
|
};
|
||||||
|
|
||||||
|
export const contextualize =
|
||||||
|
<T>(executor: (params: T, ctx: McpContext) => Promise<Record<string, any>>) =>
|
||||||
|
async (params: T): Promise<ToolResult> => {
|
||||||
|
try {
|
||||||
|
const ctx = getMcpContext();
|
||||||
|
const result = await executor(params, ctx);
|
||||||
|
return handleToolSuccess(result);
|
||||||
|
} catch (error) {
|
||||||
|
return handleToolError(error);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
const handleToolSuccess = (data: Record<string, any>): ToolResult => {
|
||||||
|
return {
|
||||||
|
content: [
|
||||||
|
{
|
||||||
|
type: "text",
|
||||||
|
text: JSON.stringify(data, null, 2),
|
||||||
|
},
|
||||||
|
],
|
||||||
|
};
|
||||||
|
};
|
||||||
|
|
||||||
|
const handleToolError = (error: unknown): ToolResult => {
|
||||||
|
const errorMessage = error instanceof Error ? error.message : String(error);
|
||||||
|
return {
|
||||||
|
content: [
|
||||||
|
{
|
||||||
|
type: "text",
|
||||||
|
text: `Error: ${errorMessage}`,
|
||||||
|
},
|
||||||
|
],
|
||||||
|
isError: true,
|
||||||
|
};
|
||||||
|
};
|
||||||
+18
-14
@@ -1,6 +1,6 @@
|
|||||||
{
|
{
|
||||||
"name": "@pullfrog/action",
|
"name": "@pullfrog/action",
|
||||||
"version": "0.0.8",
|
"version": "0.0.74",
|
||||||
"type": "module",
|
"type": "module",
|
||||||
"files": [
|
"files": [
|
||||||
"index.js",
|
"index.js",
|
||||||
@@ -12,31 +12,35 @@
|
|||||||
"main.js",
|
"main.js",
|
||||||
"main.d.ts"
|
"main.d.ts"
|
||||||
],
|
],
|
||||||
"directories": {
|
|
||||||
"example": "examples"
|
|
||||||
},
|
|
||||||
"scripts": {
|
"scripts": {
|
||||||
"test": "echo \"Error: no test specified\" && exit 1",
|
"test": "echo \"Error: no test specified\" && exit 1",
|
||||||
"typecheck": "tsc --noEmit",
|
"typecheck": "tsc --noEmit",
|
||||||
"build": "node esbuild.config.js",
|
"build": "node esbuild.config.js",
|
||||||
"build:npm": "zshy",
|
"play": "node play.ts",
|
||||||
"build:dev": "node esbuild.config.js",
|
"upDeps": "pnpm up --latest",
|
||||||
"prepare": "husky",
|
"lock": "pnpm --ignore-workspace install",
|
||||||
"play": "tsx play.ts"
|
"prepare": "husky"
|
||||||
},
|
},
|
||||||
"dependencies": {
|
"dependencies": {
|
||||||
"@actions/core": "^1.11.1",
|
"@actions/core": "^1.11.1",
|
||||||
"dotenv": "^17.2.2",
|
"@anthropic-ai/claude-agent-sdk": "^0.1.26",
|
||||||
|
"@ark/fs": "0.53.0",
|
||||||
|
"@ark/util": "0.53.0",
|
||||||
|
"@octokit/rest": "^22.0.0",
|
||||||
|
"@octokit/webhooks-types": "^7.6.1",
|
||||||
|
"@standard-schema/spec": "1.0.0",
|
||||||
|
"arktype": "2.1.25",
|
||||||
|
"dotenv": "^17.2.3",
|
||||||
"execa": "^9.6.0",
|
"execa": "^9.6.0",
|
||||||
|
"fastmcp": "^3.20.0",
|
||||||
"table": "^6.9.0"
|
"table": "^6.9.0"
|
||||||
},
|
},
|
||||||
"devDependencies": {
|
"devDependencies": {
|
||||||
"@types/node": "^20.10.0",
|
"@types/node": "^24.7.2",
|
||||||
"commander": "^14.0.0",
|
"arg": "^5.0.2",
|
||||||
"esbuild": "^0.25.9",
|
"esbuild": "^0.25.9",
|
||||||
"husky": "^9.0.0",
|
"husky": "^9.0.0",
|
||||||
"typescript": "^5.3.0",
|
"typescript": "^5.9.3"
|
||||||
"zshy": "^0.4.1"
|
|
||||||
},
|
},
|
||||||
"repository": {
|
"repository": {
|
||||||
"type": "git",
|
"type": "git",
|
||||||
@@ -44,7 +48,7 @@
|
|||||||
},
|
},
|
||||||
"keywords": [],
|
"keywords": [],
|
||||||
"author": "",
|
"author": "",
|
||||||
"license": "ISC",
|
"license": "MIT",
|
||||||
"bugs": {
|
"bugs": {
|
||||||
"url": "https://github.com/pullfrog/action/issues"
|
"url": "https://github.com/pullfrog/action/issues"
|
||||||
},
|
},
|
||||||
|
|||||||
@@ -1,160 +1,148 @@
|
|||||||
import { existsSync, readFileSync } from "node:fs";
|
import { existsSync, readFileSync } from "node:fs";
|
||||||
import { dirname, extname, join, resolve } from "node:path";
|
import { extname, join, resolve } from "node:path";
|
||||||
import { fileURLToPath, pathToFileURL } from "node:url";
|
import { pathToFileURL } from "node:url";
|
||||||
import { Command } from "commander";
|
import { fromHere } from "@ark/fs";
|
||||||
|
import arg from "arg";
|
||||||
import { config } from "dotenv";
|
import { config } from "dotenv";
|
||||||
import { main } from "./main";
|
import { type Inputs, main } from "./main.ts";
|
||||||
import { runAct } from "./utils/act";
|
import packageJson from "./package.json" with { type: "json" };
|
||||||
import { setupTestRepo } from "./utils/setup";
|
import { log } from "./utils/cli.ts";
|
||||||
|
import { setupTestRepo } from "./utils/setup.ts";
|
||||||
|
|
||||||
// Load environment variables from .env file
|
|
||||||
config();
|
config();
|
||||||
|
|
||||||
|
export async function run(
|
||||||
|
prompt: string
|
||||||
|
): Promise<{ success: boolean; output?: string | undefined; error?: string | undefined }> {
|
||||||
|
try {
|
||||||
|
log.info(`🐸 Running pullfrog/action@${packageJson.version}...`);
|
||||||
|
|
||||||
const __filename = fileURLToPath(import.meta.url);
|
const tempDir = join(process.cwd(), ".temp");
|
||||||
const __dirname = dirname(__filename);
|
setupTestRepo({ tempDir, forceClean: true });
|
||||||
|
|
||||||
async function loadPrompt(filePath: string): Promise<string> {
|
const originalCwd = process.cwd();
|
||||||
const ext = extname(filePath).toLowerCase();
|
process.chdir(tempDir);
|
||||||
|
|
||||||
// Try to resolve the file path
|
log.info("🚀 Running action with prompt...");
|
||||||
let resolvedPath: string;
|
log.separator();
|
||||||
|
log.box(prompt, { title: "Prompt" });
|
||||||
|
log.separator();
|
||||||
|
|
||||||
// First try as fixtures path
|
const inputs: Inputs = {
|
||||||
const fixturesPath = join(__dirname, "fixtures", filePath);
|
prompt,
|
||||||
if (existsSync(fixturesPath)) {
|
anthropic_api_key: process.env.ANTHROPIC_API_KEY,
|
||||||
resolvedPath = fixturesPath;
|
};
|
||||||
} else if (existsSync(filePath)) {
|
|
||||||
resolvedPath = resolve(filePath);
|
|
||||||
} else {
|
|
||||||
throw new Error(`File not found: ${filePath}`);
|
|
||||||
}
|
|
||||||
|
|
||||||
switch (ext) {
|
const result = await main(inputs);
|
||||||
case ".txt": {
|
|
||||||
// Plain text - pass directly as prompt
|
process.chdir(originalCwd);
|
||||||
return readFileSync(resolvedPath, "utf8").trim();
|
|
||||||
|
if (result.success) {
|
||||||
|
log.success("Action completed successfully");
|
||||||
|
return { success: true, output: result.output || undefined, error: undefined };
|
||||||
|
} else {
|
||||||
|
log.error(`Action failed: ${result.error || "Unknown error"}`);
|
||||||
|
return { success: false, error: result.error || undefined, output: undefined };
|
||||||
}
|
}
|
||||||
|
} catch (err) {
|
||||||
case ".json": {
|
const errorMessage = (err as Error).message;
|
||||||
// JSON - stringify and pass as prompt
|
log.error(`Error: ${errorMessage}`);
|
||||||
const content = readFileSync(resolvedPath, "utf8");
|
return { success: false, error: errorMessage, output: undefined };
|
||||||
const parsed = JSON.parse(content);
|
|
||||||
return JSON.stringify(parsed, null, 2);
|
|
||||||
}
|
|
||||||
|
|
||||||
case ".ts": {
|
|
||||||
// TypeScript - dynamic import and stringify default export
|
|
||||||
const fileUrl = pathToFileURL(resolvedPath).href;
|
|
||||||
const module = await import(fileUrl);
|
|
||||||
|
|
||||||
if (!module.default) {
|
|
||||||
throw new Error(`TypeScript file ${filePath} must have a default export`);
|
|
||||||
}
|
|
||||||
|
|
||||||
// If it's a string, use it directly
|
|
||||||
if (typeof module.default === "string") {
|
|
||||||
return module.default;
|
|
||||||
}
|
|
||||||
|
|
||||||
// If it's a MainParams object with a prompt field, extract the prompt
|
|
||||||
if (typeof module.default === "object" && module.default.prompt) {
|
|
||||||
return module.default.prompt;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Otherwise stringify it
|
|
||||||
return JSON.stringify(module.default, null, 2);
|
|
||||||
}
|
|
||||||
|
|
||||||
default:
|
|
||||||
throw new Error(`Unsupported file type: ${ext}. Supported types: .txt, .json, .ts`);
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
async function runPlay(filePath: string, options: { act?: boolean }): Promise<void> {
|
if (import.meta.url === `file://${process.argv[1]}`) {
|
||||||
try {
|
const args = arg({
|
||||||
// Load the prompt from the specified file
|
"--help": Boolean,
|
||||||
const prompt = await loadPrompt(filePath);
|
"--raw": String,
|
||||||
|
"-h": "--help",
|
||||||
|
});
|
||||||
|
|
||||||
if (options.act) {
|
if (args["--help"]) {
|
||||||
// Use Docker/act to run the action
|
log.info(`
|
||||||
console.log("🐳 Running with Docker/act...");
|
Usage: tsx play.ts [file] [options]
|
||||||
runAct(prompt);
|
|
||||||
|
Test the Pullfrog action with various prompts.
|
||||||
|
|
||||||
|
Arguments:
|
||||||
|
file Prompt file to use (.txt, .json, or .ts) [default: fixtures/basic.txt]
|
||||||
|
|
||||||
|
Options:
|
||||||
|
--raw [prompt] Use raw string as prompt instead of loading from file
|
||||||
|
-h, --help Show this help message
|
||||||
|
|
||||||
|
Examples:
|
||||||
|
tsx play.ts # Use default fixture
|
||||||
|
tsx play.ts fixtures/basic.txt # Use specific text file
|
||||||
|
tsx play.ts custom.json # Use JSON file
|
||||||
|
tsx play.ts fixtures/test.ts # Use TypeScript file
|
||||||
|
tsx play.ts --raw "Hello world" # Use raw string as prompt
|
||||||
|
`);
|
||||||
|
process.exit(0);
|
||||||
|
}
|
||||||
|
|
||||||
|
let prompt: string;
|
||||||
|
|
||||||
|
if (args["--raw"]) {
|
||||||
|
prompt = args["--raw"];
|
||||||
|
} else {
|
||||||
|
const filePath = args._[0] || "fixtures/basic.txt";
|
||||||
|
const ext = extname(filePath).toLowerCase();
|
||||||
|
let resolvedPath: string;
|
||||||
|
|
||||||
|
const fixturesPath = fromHere("fixtures", filePath);
|
||||||
|
if (existsSync(fixturesPath)) {
|
||||||
|
resolvedPath = fixturesPath;
|
||||||
|
} else if (existsSync(filePath)) {
|
||||||
|
resolvedPath = resolve(filePath);
|
||||||
} else {
|
} else {
|
||||||
// Setup test repository and run directly
|
throw new Error(`File not found: ${filePath}`);
|
||||||
const tempDir = join(process.cwd(), ".temp");
|
|
||||||
setupTestRepo({ tempDir, forceClean: true });
|
|
||||||
|
|
||||||
// Change to the temp directory
|
|
||||||
process.chdir(tempDir);
|
|
||||||
|
|
||||||
console.log("🚀 Running test in .temp directory...");
|
|
||||||
console.log("─".repeat(50));
|
|
||||||
console.log(`Prompt from ${filePath}:`);
|
|
||||||
console.log(prompt);
|
|
||||||
console.log("─".repeat(50));
|
|
||||||
|
|
||||||
// Set environment variables from our .env for the action to use
|
|
||||||
const { EXPECTED_INPUTS } = await import("./main");
|
|
||||||
EXPECTED_INPUTS.forEach((inputName) => {
|
|
||||||
const value = process.env[inputName];
|
|
||||||
if (value) {
|
|
||||||
process.env[`INPUT_${inputName.toLowerCase()}`] = value;
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
// Run main with the new params structure
|
|
||||||
const inputs: any = {
|
|
||||||
prompt,
|
|
||||||
anthropic_api_key: process.env.ANTHROPIC_API_KEY || "",
|
|
||||||
};
|
|
||||||
|
|
||||||
// Add optional properties only if they exist
|
|
||||||
if (process.env.GITHUB_TOKEN) {
|
|
||||||
inputs.github_token = process.env.GITHUB_TOKEN;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (process.env.GITHUB_INSTALLATION_TOKEN) {
|
|
||||||
inputs.github_installation_token = process.env.GITHUB_INSTALLATION_TOKEN;
|
|
||||||
}
|
|
||||||
|
|
||||||
const result = await main({
|
|
||||||
inputs,
|
|
||||||
env: process.env as Record<string, string>,
|
|
||||||
cwd: process.cwd(),
|
|
||||||
});
|
|
||||||
|
|
||||||
if (result.success) {
|
|
||||||
console.log("✅ Test completed successfully");
|
|
||||||
if (result.output) {
|
|
||||||
console.log("Output:", result.output);
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
console.error("❌ Test failed:", result.error);
|
|
||||||
process.exit(1);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
} catch (error) {
|
|
||||||
console.error("❌ Error:", (error as Error).message);
|
switch (ext) {
|
||||||
|
case ".txt": {
|
||||||
|
prompt = readFileSync(resolvedPath, "utf8").trim();
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
case ".json": {
|
||||||
|
const content = readFileSync(resolvedPath, "utf8");
|
||||||
|
const parsed = JSON.parse(content);
|
||||||
|
prompt = JSON.stringify(parsed, null, 2);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
case ".ts": {
|
||||||
|
const fileUrl = pathToFileURL(resolvedPath).href;
|
||||||
|
const module = await import(fileUrl);
|
||||||
|
|
||||||
|
if (!module.default) {
|
||||||
|
throw new Error(`TypeScript file ${filePath} must have a default export`);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (typeof module.default === "string") {
|
||||||
|
prompt = module.default;
|
||||||
|
} else if (typeof module.default === "object" && module.default.prompt) {
|
||||||
|
prompt = module.default.prompt;
|
||||||
|
} else {
|
||||||
|
prompt = JSON.stringify(module.default, null, 2);
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
default:
|
||||||
|
throw new Error(`Unsupported file type: ${ext}. Supported types: .txt, .json, .ts`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
const result = await run(prompt);
|
||||||
|
|
||||||
|
if (!result.success) {
|
||||||
|
process.exit(1);
|
||||||
|
}
|
||||||
|
} catch (err) {
|
||||||
|
log.error((err as Error).message);
|
||||||
process.exit(1);
|
process.exit(1);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Set up CLI
|
|
||||||
const program = new Command();
|
|
||||||
|
|
||||||
program
|
|
||||||
.name("play")
|
|
||||||
.description("Test the Pullfrog action with various prompts")
|
|
||||||
.version("1.0.0")
|
|
||||||
.argument("[file]", "Prompt file to use (.txt, .json, or .ts)", "fixtures/basic.txt")
|
|
||||||
.option("--act", "Use Docker/act to run the action instead of running directly")
|
|
||||||
.action(async (file: string, options: { act?: boolean }) => {
|
|
||||||
await runPlay(file, options);
|
|
||||||
});
|
|
||||||
|
|
||||||
// Parse arguments and run
|
|
||||||
program.parseAsync(process.argv).catch((error) => {
|
|
||||||
console.error("Fatal error:", error);
|
|
||||||
process.exit(1);
|
|
||||||
});
|
|
||||||
|
|||||||
Generated
+1329
-276
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,8 @@
|
|||||||
|
[x] add modes to prompt
|
||||||
|
[x] progressively update comment
|
||||||
|
[] don't allow rejecting prs
|
||||||
|
[] fix pnpm caching
|
||||||
|
[] try to avoid claude narrating the initial comment
|
||||||
|
[] fix prompt to avoid narration like "I just read all tools from MCP server"
|
||||||
|
[] investigate including terminal output from bash commands as collapsed groups
|
||||||
|
[] avoid exposing env
|
||||||
+17
-39
@@ -1,45 +1,23 @@
|
|||||||
{
|
{
|
||||||
// Visit https://aka.ms/tsconfig to read more about this file
|
|
||||||
"compilerOptions": {
|
"compilerOptions": {
|
||||||
// File Layout
|
|
||||||
// "rootDir": "./src",
|
|
||||||
"outDir": "./dist",
|
"outDir": "./dist",
|
||||||
|
"module": "NodeNext",
|
||||||
// Environment Settings
|
"target": "ESNext",
|
||||||
// See also https://aka.ms/tsconfig/module
|
"moduleResolution": "NodeNext",
|
||||||
"module": "esnext",
|
"lib": ["ESNext"],
|
||||||
"moduleResolution": "bundler",
|
"allowImportingTsExtensions": true,
|
||||||
"target": "esnext",
|
"rewriteRelativeImportExtensions": true,
|
||||||
"types": [],
|
"skipLibCheck": true,
|
||||||
// For nodejs:
|
"strict": true,
|
||||||
// "lib": ["esnext"],
|
"noUncheckedSideEffectImports": true,
|
||||||
// "types": ["node"],
|
"declaration": true,
|
||||||
// and npm install -D @types/node
|
"verbatimModuleSyntax": true,
|
||||||
|
"esModuleInterop": true,
|
||||||
// Other Outputs
|
"resolveJsonModule": true,
|
||||||
"sourceMap": true,
|
"exactOptionalPropertyTypes": true,
|
||||||
"declaration": true,
|
"forceConsistentCasingInFileNames": true,
|
||||||
"declarationMap": true,
|
"stripInternal": true,
|
||||||
|
|
||||||
// Stricter Typechecking Options
|
|
||||||
"noUncheckedIndexedAccess": true,
|
|
||||||
"exactOptionalPropertyTypes": true,
|
|
||||||
|
|
||||||
// Style Options
|
|
||||||
// "noImplicitReturns": true,
|
|
||||||
// "noImplicitOverride": true,
|
|
||||||
// "noUnusedLocals": true,
|
|
||||||
// "noUnusedParameters": true,
|
|
||||||
// "noFallthroughCasesInSwitch": true,
|
|
||||||
// "noPropertyAccessFromIndexSignature": true,
|
|
||||||
|
|
||||||
// Recommended Options
|
|
||||||
"strict": true,
|
|
||||||
"jsx": "react-jsx",
|
|
||||||
"verbatimModuleSyntax": true,
|
|
||||||
"isolatedModules": true,
|
|
||||||
"noUncheckedSideEffectImports": true,
|
|
||||||
"moduleDetection": "force",
|
"moduleDetection": "force",
|
||||||
"skipLibCheck": true
|
"useUnknownInCatchVariables": true
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,89 +0,0 @@
|
|||||||
import { execSync } from "node:child_process";
|
|
||||||
import { existsSync } from "node:fs";
|
|
||||||
import { dirname, join } from "node:path";
|
|
||||||
import { fileURLToPath } from "node:url";
|
|
||||||
import { config } from "dotenv";
|
|
||||||
import { buildAction, setupTestRepo } from "./setup";
|
|
||||||
|
|
||||||
const __filename = fileURLToPath(import.meta.url);
|
|
||||||
const __dirname = dirname(__filename);
|
|
||||||
|
|
||||||
// Environment variables that should be passed as secrets to the workflow
|
|
||||||
const ENV_VARS = ["ANTHROPIC_API_KEY", "GITHUB_INSTALLATION_TOKEN"];
|
|
||||||
|
|
||||||
export function runAct(prompt: string): void {
|
|
||||||
const tempDir = join(__dirname, "..", ".temp");
|
|
||||||
const actionPath = join(__dirname, "..");
|
|
||||||
const envPath = join(__dirname, "..", "..", ".env");
|
|
||||||
|
|
||||||
// Setup test repository
|
|
||||||
setupTestRepo({ tempDir });
|
|
||||||
|
|
||||||
// Load environment variables
|
|
||||||
config({ path: envPath });
|
|
||||||
|
|
||||||
// Build action bundles
|
|
||||||
buildAction(actionPath);
|
|
||||||
|
|
||||||
const workflowPath = join(tempDir, ".github", "workflows", "pullfrog.yml");
|
|
||||||
|
|
||||||
// Create minimal dist for act (avoids pnpm symlink issues)
|
|
||||||
const distPath = join(actionPath, ".act-dist");
|
|
||||||
console.log("📦 Creating minimal distribution for act...");
|
|
||||||
execSync(`rm -rf "${distPath}" && mkdir -p "${distPath}"`, { shell: "/bin/bash" });
|
|
||||||
|
|
||||||
// Copy only necessary files (bundled, no node_modules needed)
|
|
||||||
["action.yml", "entry.cjs", "index.cjs", "package.json"].forEach((file) => {
|
|
||||||
const src = join(actionPath, file);
|
|
||||||
if (existsSync(src)) {
|
|
||||||
execSync(`cp "${src}" "${distPath}"`);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
try {
|
|
||||||
// Build the act command with input directly
|
|
||||||
// Properly escape the prompt for shell
|
|
||||||
const escapedPrompt = prompt.replace(/'/g, "'\\''");
|
|
||||||
|
|
||||||
const actCommandParts = [
|
|
||||||
"act",
|
|
||||||
"workflow_dispatch",
|
|
||||||
"-W",
|
|
||||||
workflowPath,
|
|
||||||
"--input",
|
|
||||||
`prompt='${escapedPrompt}'`,
|
|
||||||
"--local-repository",
|
|
||||||
`pullfrog/action@v0=${distPath}`, // Use minimal dist without symlinks
|
|
||||||
];
|
|
||||||
|
|
||||||
// Add environment variables as secrets that will be available to the workflow
|
|
||||||
ENV_VARS.forEach((key) => {
|
|
||||||
if (process.env[key]) {
|
|
||||||
actCommandParts.push("-s", key);
|
|
||||||
}
|
|
||||||
});
|
|
||||||
|
|
||||||
// We only need the specific ENV_VARS, no need to add other variables
|
|
||||||
|
|
||||||
const actCommand = actCommandParts.join(" ");
|
|
||||||
|
|
||||||
console.log("🚀 Running act with prompt:");
|
|
||||||
console.log("─".repeat(50));
|
|
||||||
console.log(prompt);
|
|
||||||
console.log("─".repeat(50));
|
|
||||||
console.log("");
|
|
||||||
|
|
||||||
// Execute act
|
|
||||||
execSync(actCommand, {
|
|
||||||
stdio: "inherit",
|
|
||||||
cwd: join(__dirname, "..", ".."),
|
|
||||||
});
|
|
||||||
// Clean up
|
|
||||||
execSync(`rm -rf "${distPath}"`);
|
|
||||||
} catch (error) {
|
|
||||||
// Clean up on error
|
|
||||||
execSync(`rm -rf "${distPath}"`);
|
|
||||||
console.error("❌ Act execution failed:", (error as Error).message);
|
|
||||||
process.exit(1);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
+264
@@ -0,0 +1,264 @@
|
|||||||
|
/**
|
||||||
|
* CLI output utilities that work well in both local and GitHub Actions environments
|
||||||
|
*/
|
||||||
|
|
||||||
|
import * as core from "@actions/core";
|
||||||
|
|
||||||
|
const isGitHubActions = !!process.env.GITHUB_ACTIONS;
|
||||||
|
const isDebugEnabled = process.env.LOG_LEVEL === "debug";
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Start a collapsed group (GitHub Actions) or regular group (local)
|
||||||
|
*/
|
||||||
|
function startGroup(name: string): void {
|
||||||
|
if (isGitHubActions) {
|
||||||
|
core.startGroup(name);
|
||||||
|
} else {
|
||||||
|
console.group(name);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* End a collapsed group
|
||||||
|
*/
|
||||||
|
function endGroup(): void {
|
||||||
|
if (isGitHubActions) {
|
||||||
|
core.endGroup();
|
||||||
|
} else {
|
||||||
|
console.groupEnd();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Print a formatted box with text (for console output)
|
||||||
|
*/
|
||||||
|
function boxString(
|
||||||
|
text: string,
|
||||||
|
options?: {
|
||||||
|
title?: string;
|
||||||
|
maxWidth?: number;
|
||||||
|
indent?: string;
|
||||||
|
padding?: number;
|
||||||
|
}
|
||||||
|
): string {
|
||||||
|
const { title, maxWidth = 80, indent = "", padding = 1 } = options || {};
|
||||||
|
|
||||||
|
const lines = text.trim().split("\n");
|
||||||
|
const wrappedLines: string[] = [];
|
||||||
|
|
||||||
|
for (const line of lines) {
|
||||||
|
if (line.length <= maxWidth - padding * 2) {
|
||||||
|
wrappedLines.push(line);
|
||||||
|
} else {
|
||||||
|
const words = line.split(" ");
|
||||||
|
let currentLine = "";
|
||||||
|
|
||||||
|
for (const word of words) {
|
||||||
|
const testLine = currentLine ? `${currentLine} ${word}` : word;
|
||||||
|
if (testLine.length <= maxWidth - padding * 2) {
|
||||||
|
currentLine = testLine;
|
||||||
|
} else {
|
||||||
|
if (currentLine) {
|
||||||
|
wrappedLines.push(currentLine);
|
||||||
|
currentLine = word;
|
||||||
|
} else {
|
||||||
|
wrappedLines.push(word.substring(0, maxWidth - padding * 2));
|
||||||
|
currentLine = word.substring(maxWidth - padding * 2);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (currentLine) {
|
||||||
|
wrappedLines.push(currentLine);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const maxLineLength = Math.max(...wrappedLines.map((line) => line.length));
|
||||||
|
const boxWidth = maxLineLength + padding * 2;
|
||||||
|
|
||||||
|
let result = "";
|
||||||
|
|
||||||
|
if (title) {
|
||||||
|
const titleLine = ` ${title} `;
|
||||||
|
const titlePadding = Math.max(0, boxWidth - titleLine.length);
|
||||||
|
result += `${indent}┌${titleLine}${"─".repeat(titlePadding)}┐\n`;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!title) {
|
||||||
|
result += `${indent}┌${"─".repeat(boxWidth)}┐\n`;
|
||||||
|
}
|
||||||
|
|
||||||
|
for (const line of wrappedLines) {
|
||||||
|
const paddedLine = line.padEnd(maxLineLength);
|
||||||
|
result += `${indent}│${" ".repeat(padding)}${paddedLine}${" ".repeat(padding)}│\n`;
|
||||||
|
}
|
||||||
|
|
||||||
|
result += `${indent}└${"─".repeat(boxWidth)}┘`;
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Print a formatted box with text
|
||||||
|
* Works well in both local and GitHub Actions environments
|
||||||
|
*/
|
||||||
|
function box(
|
||||||
|
text: string,
|
||||||
|
options?: {
|
||||||
|
title?: string;
|
||||||
|
maxWidth?: number;
|
||||||
|
}
|
||||||
|
): void {
|
||||||
|
const boxContent = boxString(text, options);
|
||||||
|
core.info(boxContent);
|
||||||
|
if (isGitHubActions) {
|
||||||
|
// Add as markdown code block for summary (no headers)
|
||||||
|
core.summary.addRaw(`\`\`\`\n${text}\n\`\`\`\n`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Add a table to GitHub Actions job summary (rich formatting)
|
||||||
|
* Also logs to console. Only use this once at the end of execution.
|
||||||
|
*/
|
||||||
|
async function summaryTable(
|
||||||
|
rows: Array<Array<{ data: string; header?: boolean } | string>>,
|
||||||
|
options?: {
|
||||||
|
title?: string;
|
||||||
|
}
|
||||||
|
): Promise<void> {
|
||||||
|
const { title } = options || {};
|
||||||
|
|
||||||
|
// Convert rows to format expected by Job Summaries API
|
||||||
|
const formattedRows = rows.map((row) =>
|
||||||
|
row.map((cell) => {
|
||||||
|
if (typeof cell === "string") {
|
||||||
|
return { data: cell };
|
||||||
|
}
|
||||||
|
return cell;
|
||||||
|
})
|
||||||
|
);
|
||||||
|
|
||||||
|
if (isGitHubActions) {
|
||||||
|
const summary = core.summary;
|
||||||
|
if (title) {
|
||||||
|
summary.addRaw(`**${title}**\n\n`);
|
||||||
|
}
|
||||||
|
summary.addTable(formattedRows);
|
||||||
|
// Note: Don't write immediately, let it accumulate with other summary content
|
||||||
|
}
|
||||||
|
|
||||||
|
// Also log to console for visibility
|
||||||
|
if (title) {
|
||||||
|
core.info(`\n${title}`);
|
||||||
|
}
|
||||||
|
const tableText = formattedRows.map((row) => row.map((cell) => cell.data).join(" | ")).join("\n");
|
||||||
|
core.info(`\n${tableText}\n`);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Print a separator line
|
||||||
|
*/
|
||||||
|
function separator(length: number = 50): void {
|
||||||
|
const separatorText = "─".repeat(length);
|
||||||
|
core.info(separatorText);
|
||||||
|
if (isGitHubActions) {
|
||||||
|
core.summary.addRaw(`---\n`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Main logging utility object - import this once and access all utilities
|
||||||
|
*/
|
||||||
|
export const log = {
|
||||||
|
/**
|
||||||
|
* Print info message
|
||||||
|
*/
|
||||||
|
info: (message: string): void => {
|
||||||
|
core.info(message);
|
||||||
|
if (isGitHubActions) {
|
||||||
|
core.summary.addRaw(`${message}\n`);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Print warning message
|
||||||
|
*/
|
||||||
|
warning: (message: string): void => {
|
||||||
|
core.warning(message);
|
||||||
|
if (isGitHubActions) {
|
||||||
|
core.summary.addRaw(`⚠️ ${message}\n`);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Print error message
|
||||||
|
*/
|
||||||
|
error: (message: string): void => {
|
||||||
|
core.error(message);
|
||||||
|
if (isGitHubActions) {
|
||||||
|
core.summary.addRaw(`❌ ${message}\n`);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Print success message
|
||||||
|
*/
|
||||||
|
success: (message: string): void => {
|
||||||
|
const successMessage = `✅ ${message}`;
|
||||||
|
core.info(successMessage);
|
||||||
|
if (isGitHubActions) {
|
||||||
|
core.summary.addRaw(`${successMessage}\n`);
|
||||||
|
}
|
||||||
|
},
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Print debug message (only if LOG_LEVEL=debug)
|
||||||
|
*/
|
||||||
|
debug: (message: string): void => {
|
||||||
|
if (isDebugEnabled) {
|
||||||
|
if (isGitHubActions) {
|
||||||
|
core.debug(message);
|
||||||
|
} else {
|
||||||
|
core.info(`[DEBUG] ${message}`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
},
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Print a formatted box with text
|
||||||
|
*/
|
||||||
|
box,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Add a table to GitHub Actions job summary (rich formatting)
|
||||||
|
* Only use this once at the end of execution
|
||||||
|
*/
|
||||||
|
summaryTable,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Print a separator line
|
||||||
|
*/
|
||||||
|
separator,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Write all accumulated summary content to the job summary
|
||||||
|
* Call this at the end of execution to finalize the summary
|
||||||
|
*/
|
||||||
|
writeSummary: async (): Promise<void> => {
|
||||||
|
if (isGitHubActions) {
|
||||||
|
await core.summary.write();
|
||||||
|
}
|
||||||
|
},
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Start a collapsed group (GitHub Actions) or regular group (local)
|
||||||
|
*/
|
||||||
|
startGroup,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* End a collapsed group
|
||||||
|
*/
|
||||||
|
endGroup,
|
||||||
|
};
|
||||||
@@ -1,13 +0,0 @@
|
|||||||
import { mkdtempSync, writeFileSync } from "node:fs";
|
|
||||||
import { tmpdir } from "node:os";
|
|
||||||
import { join } from "node:path";
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Create a temporary file with the given content
|
|
||||||
*/
|
|
||||||
export function createTempFile(content: string, filename = "temp.txt"): string {
|
|
||||||
const tempDir = mkdtempSync(join(tmpdir(), "pullfrog-"));
|
|
||||||
const filePath = join(tempDir, filename);
|
|
||||||
writeFileSync(filePath, content);
|
|
||||||
return filePath;
|
|
||||||
}
|
|
||||||
+261
-43
@@ -1,56 +1,274 @@
|
|||||||
|
import { createSign } from "node:crypto";
|
||||||
import * as core from "@actions/core";
|
import * as core from "@actions/core";
|
||||||
|
import { log } from "./cli.ts";
|
||||||
|
|
||||||
|
export interface InstallationToken {
|
||||||
|
token: string;
|
||||||
|
expires_at: string;
|
||||||
|
installation_id: number;
|
||||||
|
repository: string;
|
||||||
|
ref: string;
|
||||||
|
runner_environment: string;
|
||||||
|
owner?: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
interface GitHubAppConfig {
|
||||||
|
appId: string;
|
||||||
|
privateKey: string;
|
||||||
|
repoOwner: string;
|
||||||
|
repoName: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
interface Installation {
|
||||||
|
id: number;
|
||||||
|
account: {
|
||||||
|
login: string;
|
||||||
|
type: string;
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
interface Repository {
|
||||||
|
owner: {
|
||||||
|
login: string;
|
||||||
|
};
|
||||||
|
name: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
interface InstallationTokenResponse {
|
||||||
|
token: string;
|
||||||
|
expires_at: string;
|
||||||
|
}
|
||||||
|
|
||||||
|
interface RepositoriesResponse {
|
||||||
|
repositories: Repository[];
|
||||||
|
}
|
||||||
|
|
||||||
|
function checkExistingToken(): string | null {
|
||||||
|
const inputToken = core.getInput("github_installation_token");
|
||||||
|
const envToken = process.env.GITHUB_INSTALLATION_TOKEN;
|
||||||
|
return inputToken || envToken || null;
|
||||||
|
}
|
||||||
|
|
||||||
|
function isGitHubActionsEnvironment(): boolean {
|
||||||
|
return Boolean(process.env.GITHUB_ACTIONS);
|
||||||
|
}
|
||||||
|
|
||||||
|
async function acquireTokenViaOIDC(): Promise<string> {
|
||||||
|
log.info("Generating OIDC token...");
|
||||||
|
|
||||||
|
const oidcToken = await core.getIDToken("pullfrog-api");
|
||||||
|
log.info("OIDC token generated successfully");
|
||||||
|
|
||||||
|
const apiUrl = process.env.API_URL || "https://pullfrog.ai";
|
||||||
|
|
||||||
|
log.info("Exchanging OIDC token for installation token...");
|
||||||
|
const tokenResponse = await fetch(`${apiUrl}/api/github/installation-token`, {
|
||||||
|
method: "POST",
|
||||||
|
headers: {
|
||||||
|
Authorization: `Bearer ${oidcToken}`,
|
||||||
|
"Content-Type": "application/json",
|
||||||
|
},
|
||||||
|
});
|
||||||
|
|
||||||
|
if (!tokenResponse.ok) {
|
||||||
|
const errorText = await tokenResponse.text();
|
||||||
|
throw new Error(
|
||||||
|
`Token exchange failed: ${tokenResponse.status} ${tokenResponse.statusText} - ${errorText}`
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
const tokenData = (await tokenResponse.json()) as InstallationToken;
|
||||||
|
log.info(`Installation token obtained for ${tokenData.repository || "all repositories"}`);
|
||||||
|
|
||||||
|
return tokenData.token;
|
||||||
|
}
|
||||||
|
|
||||||
|
const base64UrlEncode = (str: string): string => {
|
||||||
|
return Buffer.from(str)
|
||||||
|
.toString("base64")
|
||||||
|
.replace(/\+/g, "-")
|
||||||
|
.replace(/\//g, "_")
|
||||||
|
.replace(/=/g, "");
|
||||||
|
};
|
||||||
|
|
||||||
|
const generateJWT = (appId: string, privateKey: string): string => {
|
||||||
|
const now = Math.floor(Date.now() / 1000);
|
||||||
|
const payload = {
|
||||||
|
iat: now - 60,
|
||||||
|
exp: now + 5 * 60,
|
||||||
|
iss: appId,
|
||||||
|
};
|
||||||
|
|
||||||
|
const header = {
|
||||||
|
alg: "RS256",
|
||||||
|
typ: "JWT",
|
||||||
|
};
|
||||||
|
|
||||||
|
const encodedHeader = base64UrlEncode(JSON.stringify(header));
|
||||||
|
const encodedPayload = base64UrlEncode(JSON.stringify(payload));
|
||||||
|
const signaturePart = `${encodedHeader}.${encodedPayload}`;
|
||||||
|
|
||||||
|
const signature = createSign("RSA-SHA256")
|
||||||
|
.update(signaturePart)
|
||||||
|
.sign(privateKey, "base64")
|
||||||
|
.replace(/\+/g, "-")
|
||||||
|
.replace(/\//g, "_")
|
||||||
|
.replace(/=/g, "");
|
||||||
|
|
||||||
|
return `${signaturePart}.${signature}`;
|
||||||
|
};
|
||||||
|
|
||||||
|
const githubRequest = async <T>(
|
||||||
|
path: string,
|
||||||
|
options: {
|
||||||
|
method?: string;
|
||||||
|
headers?: Record<string, string>;
|
||||||
|
body?: string;
|
||||||
|
} = {}
|
||||||
|
): Promise<T> => {
|
||||||
|
const { method = "GET", headers = {}, body } = options;
|
||||||
|
|
||||||
|
const url = `https://api.github.com${path}`;
|
||||||
|
const requestHeaders = {
|
||||||
|
Accept: "application/vnd.github.v3+json",
|
||||||
|
"User-Agent": "Pullfrog-Installation-Token-Generator/1.0",
|
||||||
|
...headers,
|
||||||
|
};
|
||||||
|
|
||||||
|
const response = await fetch(url, {
|
||||||
|
method,
|
||||||
|
headers: requestHeaders,
|
||||||
|
...(body && { body }),
|
||||||
|
});
|
||||||
|
|
||||||
|
if (!response.ok) {
|
||||||
|
const errorText = await response.text();
|
||||||
|
throw new Error(
|
||||||
|
`GitHub API request failed: ${response.status} ${response.statusText}\n${errorText}`
|
||||||
|
);
|
||||||
|
}
|
||||||
|
|
||||||
|
return response.json() as T;
|
||||||
|
};
|
||||||
|
|
||||||
|
const checkRepositoryAccess = async (
|
||||||
|
token: string,
|
||||||
|
repoOwner: string,
|
||||||
|
repoName: string
|
||||||
|
): Promise<boolean> => {
|
||||||
|
try {
|
||||||
|
const response = await githubRequest<RepositoriesResponse>("/installation/repositories", {
|
||||||
|
headers: { Authorization: `token ${token}` },
|
||||||
|
});
|
||||||
|
|
||||||
|
return response.repositories.some(
|
||||||
|
(repo) => repo.owner.login === repoOwner && repo.name === repoName
|
||||||
|
);
|
||||||
|
} catch {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
const createInstallationToken = async (jwt: string, installationId: number): Promise<string> => {
|
||||||
|
const response = await githubRequest<InstallationTokenResponse>(
|
||||||
|
`/app/installations/${installationId}/access_tokens`,
|
||||||
|
{
|
||||||
|
method: "POST",
|
||||||
|
headers: { Authorization: `Bearer ${jwt}` },
|
||||||
|
}
|
||||||
|
);
|
||||||
|
|
||||||
|
return response.token;
|
||||||
|
};
|
||||||
|
|
||||||
|
const findInstallationId = async (
|
||||||
|
jwt: string,
|
||||||
|
repoOwner: string,
|
||||||
|
repoName: string
|
||||||
|
): Promise<number> => {
|
||||||
|
const installations = await githubRequest<Installation[]>("/app/installations", {
|
||||||
|
headers: { Authorization: `Bearer ${jwt}` },
|
||||||
|
});
|
||||||
|
|
||||||
|
for (const installation of installations) {
|
||||||
|
try {
|
||||||
|
const tempToken = await createInstallationToken(jwt, installation.id);
|
||||||
|
const hasAccess = await checkRepositoryAccess(tempToken, repoOwner, repoName);
|
||||||
|
|
||||||
|
if (hasAccess) {
|
||||||
|
return installation.id;
|
||||||
|
}
|
||||||
|
} catch {}
|
||||||
|
}
|
||||||
|
|
||||||
|
throw new Error(
|
||||||
|
`No installation found with access to ${repoOwner}/${repoName}. ` +
|
||||||
|
"Ensure the GitHub App is installed on the target repository."
|
||||||
|
);
|
||||||
|
};
|
||||||
|
|
||||||
|
async function acquireTokenViaGitHubApp(): Promise<string> {
|
||||||
|
const repoContext = parseRepoContext();
|
||||||
|
|
||||||
|
const config: GitHubAppConfig = {
|
||||||
|
appId: process.env.GITHUB_APP_ID!,
|
||||||
|
privateKey: process.env.GITHUB_PRIVATE_KEY?.replace(/\\n/g, "\n")!,
|
||||||
|
repoOwner: repoContext.owner,
|
||||||
|
repoName: repoContext.name,
|
||||||
|
};
|
||||||
|
|
||||||
|
const jwt = generateJWT(config.appId, config.privateKey);
|
||||||
|
const installationId = await findInstallationId(jwt, config.repoOwner, config.repoName);
|
||||||
|
const token = await createInstallationToken(jwt, installationId);
|
||||||
|
|
||||||
|
return token;
|
||||||
|
}
|
||||||
|
|
||||||
|
async function acquireNewToken(): Promise<string> {
|
||||||
|
if (isGitHubActionsEnvironment()) {
|
||||||
|
return await acquireTokenViaOIDC();
|
||||||
|
} else {
|
||||||
|
return await acquireTokenViaGitHubApp();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Setup GitHub installation token for the action
|
* Setup GitHub installation token for the action
|
||||||
*/
|
*/
|
||||||
export async function setupGitHubInstallationToken(): Promise<string> {
|
export async function setupGitHubInstallationToken(): Promise<string> {
|
||||||
// Check if we have an installation token from inputs or environment
|
const existingToken = checkExistingToken();
|
||||||
const inputToken = core.getInput("github_installation_token");
|
|
||||||
const envToken = process.env.GITHUB_INSTALLATION_TOKEN;
|
|
||||||
|
|
||||||
const existingToken = inputToken || envToken;
|
|
||||||
if (existingToken) {
|
if (existingToken) {
|
||||||
core.info("Using provided GitHub installation token");
|
core.setSecret(existingToken);
|
||||||
|
log.info("Using provided GitHub installation token");
|
||||||
return existingToken;
|
return existingToken;
|
||||||
}
|
}
|
||||||
|
|
||||||
core.info("No cached installation token found, generating OIDC token...");
|
const token = await acquireNewToken();
|
||||||
|
|
||||||
try {
|
core.setSecret(token);
|
||||||
// Generate OIDC token for our API
|
process.env.GITHUB_INSTALLATION_TOKEN = token;
|
||||||
const oidcToken = await core.getIDToken("pullfrog-api");
|
|
||||||
core.info("OIDC token generated successfully");
|
|
||||||
|
|
||||||
// Exchange OIDC token for installation token
|
return token;
|
||||||
const apiUrl = process.env.API_URL || "https://pullfrog.ai";
|
}
|
||||||
|
|
||||||
const tokenResponse = await fetch(`${apiUrl}/api/github/installation-token`, {
|
export interface RepoContext {
|
||||||
method: "POST",
|
owner: string;
|
||||||
headers: {
|
name: string;
|
||||||
Authorization: `Bearer ${oidcToken}`,
|
}
|
||||||
"Content-Type": "application/json",
|
|
||||||
},
|
/**
|
||||||
});
|
* Parse repository context from GITHUB_REPOSITORY environment variable.
|
||||||
|
*/
|
||||||
if (!tokenResponse.ok) {
|
export function parseRepoContext(): RepoContext {
|
||||||
const errorText = await tokenResponse.text();
|
const githubRepo = process.env.GITHUB_REPOSITORY;
|
||||||
throw new Error(
|
if (!githubRepo) {
|
||||||
`Token exchange failed: ${tokenResponse.status} ${tokenResponse.statusText} - ${errorText}`
|
throw new Error("GITHUB_REPOSITORY environment variable is required");
|
||||||
);
|
}
|
||||||
}
|
|
||||||
|
const [owner, name] = githubRepo.split("/");
|
||||||
const tokenData = await tokenResponse.json();
|
if (!owner || !name) {
|
||||||
core.info(
|
throw new Error(`Invalid GITHUB_REPOSITORY format: ${githubRepo}. Expected 'owner/repo'`);
|
||||||
`Installation token obtained for ${tokenData.owner}/${tokenData.repository || "all repositories"}`
|
}
|
||||||
);
|
|
||||||
|
return { owner, name };
|
||||||
// Set the token as an environment variable for this run
|
|
||||||
process.env.GITHUB_INSTALLATION_TOKEN = tokenData.token;
|
|
||||||
|
|
||||||
return tokenData.token;
|
|
||||||
} catch (error) {
|
|
||||||
throw new Error(
|
|
||||||
`Failed to setup GitHub installation token: ${error instanceof Error ? error.message : "Unknown error"}`
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,5 +0,0 @@
|
|||||||
export * from "./files";
|
|
||||||
export * from "./github";
|
|
||||||
export * from "./setup";
|
|
||||||
export * from "./subprocess";
|
|
||||||
export * from "./table";
|
|
||||||
+31
-13
@@ -1,5 +1,7 @@
|
|||||||
import { execSync } from "node:child_process";
|
import { execSync } from "node:child_process";
|
||||||
import { existsSync, rmSync } from "node:fs";
|
import { existsSync, rmSync } from "node:fs";
|
||||||
|
import { log } from "./cli.ts";
|
||||||
|
import type { RepoContext } from "./github.ts";
|
||||||
|
|
||||||
export interface SetupOptions {
|
export interface SetupOptions {
|
||||||
tempDir: string;
|
tempDir: string;
|
||||||
@@ -17,35 +19,51 @@ export function setupTestRepo(options: SetupOptions): void {
|
|||||||
forceClean = false,
|
forceClean = false,
|
||||||
} = options;
|
} = options;
|
||||||
|
|
||||||
// Handle existing temp directory
|
|
||||||
if (existsSync(tempDir)) {
|
if (existsSync(tempDir)) {
|
||||||
if (forceClean) {
|
if (forceClean) {
|
||||||
console.log("🗑️ Removing existing .temp directory...");
|
log.info("🗑️ Removing existing .temp directory...");
|
||||||
rmSync(tempDir, { recursive: true, force: true });
|
rmSync(tempDir, { recursive: true, force: true });
|
||||||
|
|
||||||
// Clone the repository
|
log.info("📦 Cloning pullfrogai/scratch into .temp...");
|
||||||
console.log("📦 Cloning pullfrogai/scratch into .temp...");
|
|
||||||
execSync(`git clone ${repoUrl} ${tempDir}`, { stdio: "inherit" });
|
execSync(`git clone ${repoUrl} ${tempDir}`, { stdio: "inherit" });
|
||||||
} else {
|
} else {
|
||||||
console.log("📦 Resetting existing .temp repository...");
|
log.info("📦 Resetting existing .temp repository...");
|
||||||
execSync("git reset --hard HEAD && git clean -fd", {
|
execSync("git reset --hard HEAD && git clean -fd", {
|
||||||
cwd: tempDir,
|
cwd: tempDir,
|
||||||
stdio: "inherit",
|
stdio: "inherit",
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
console.log("📦 Cloning pullfrogai/scratch into .temp...");
|
log.info("📦 Cloning pullfrogai/scratch into .temp...");
|
||||||
execSync(`git clone ${repoUrl} ${tempDir}`, { stdio: "inherit" });
|
execSync(`git clone ${repoUrl} ${tempDir}`, { stdio: "inherit" });
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Build the action bundles
|
* Setup git configuration to avoid identity errors
|
||||||
*/
|
*/
|
||||||
export function buildAction(actionPath: string): void {
|
export function setupGitConfig(): void {
|
||||||
console.log("🔨 Building fresh bundles with esbuild...");
|
log.info("🔧 Setting up git configuration...");
|
||||||
execSync("node esbuild.config.js", {
|
execSync('git config user.email "action@pullfrog.ai"', { stdio: "inherit" });
|
||||||
cwd: actionPath,
|
execSync('git config user.name "Pullfrog Action"', { stdio: "inherit" });
|
||||||
stdio: "inherit",
|
}
|
||||||
});
|
|
||||||
|
/**
|
||||||
|
* Setup git authentication using GitHub token
|
||||||
|
*/
|
||||||
|
export function setupGitAuth(githubToken: string, repoContext: RepoContext): void {
|
||||||
|
log.info("🔐 Setting up git authentication...");
|
||||||
|
|
||||||
|
// Remove existing git auth headers that actions/checkout might have set
|
||||||
|
try {
|
||||||
|
execSync("git config --unset-all http.https://github.com/.extraheader", { stdio: "inherit" });
|
||||||
|
log.info("✓ Removed existing authentication headers");
|
||||||
|
} catch {
|
||||||
|
log.info("No existing authentication headers to remove");
|
||||||
|
}
|
||||||
|
|
||||||
|
// Update remote URL to embed the token
|
||||||
|
const remoteUrl = `https://x-access-token:${githubToken}@github.com/${repoContext.owner}/${repoContext.name}.git`;
|
||||||
|
execSync(`git remote set-url origin "${remoteUrl}"`, { stdio: "inherit" });
|
||||||
|
log.info("✓ Updated remote URL with authentication token");
|
||||||
}
|
}
|
||||||
|
|||||||
+4
-11
@@ -6,6 +6,7 @@ export interface SpawnOptions {
|
|||||||
env?: Record<string, string>;
|
env?: Record<string, string>;
|
||||||
input?: string;
|
input?: string;
|
||||||
timeout?: number;
|
timeout?: number;
|
||||||
|
cwd?: string;
|
||||||
onStdout?: (chunk: string) => void;
|
onStdout?: (chunk: string) => void;
|
||||||
onStderr?: (chunk: string) => void;
|
onStderr?: (chunk: string) => void;
|
||||||
}
|
}
|
||||||
@@ -21,20 +22,19 @@ export interface SpawnResult {
|
|||||||
* Spawn a subprocess with streaming callbacks and buffered results
|
* Spawn a subprocess with streaming callbacks and buffered results
|
||||||
*/
|
*/
|
||||||
export async function spawn(options: SpawnOptions): Promise<SpawnResult> {
|
export async function spawn(options: SpawnOptions): Promise<SpawnResult> {
|
||||||
const { cmd, args, env, input, timeout, onStdout, onStderr } = options;
|
const { cmd, args, env, input, timeout, cwd, onStdout, onStderr } = options;
|
||||||
|
|
||||||
const startTime = Date.now();
|
const startTime = Date.now();
|
||||||
let stdoutBuffer = "";
|
let stdoutBuffer = "";
|
||||||
let stderrBuffer = "";
|
let stderrBuffer = "";
|
||||||
|
|
||||||
return new Promise((resolve, reject) => {
|
return new Promise((resolve, reject) => {
|
||||||
// Spawn the child process
|
|
||||||
const child = nodeSpawn(cmd, args, {
|
const child = nodeSpawn(cmd, args, {
|
||||||
env: env ? { ...process.env, ...env } : process.env,
|
env: env ? { ...process.env, ...env } : process.env,
|
||||||
stdio: ["pipe", "pipe", "pipe"],
|
stdio: ["pipe", "pipe", "pipe"],
|
||||||
|
cwd: cwd || process.cwd(),
|
||||||
});
|
});
|
||||||
|
|
||||||
// Set up timeout if specified
|
|
||||||
let timeoutId: NodeJS.Timeout | undefined;
|
let timeoutId: NodeJS.Timeout | undefined;
|
||||||
let isTimedOut = false;
|
let isTimedOut = false;
|
||||||
|
|
||||||
@@ -43,7 +43,6 @@ export async function spawn(options: SpawnOptions): Promise<SpawnResult> {
|
|||||||
isTimedOut = true;
|
isTimedOut = true;
|
||||||
child.kill("SIGTERM");
|
child.kill("SIGTERM");
|
||||||
|
|
||||||
// If SIGTERM doesn't work, use SIGKILL after 5 seconds
|
|
||||||
setTimeout(() => {
|
setTimeout(() => {
|
||||||
if (!child.killed) {
|
if (!child.killed) {
|
||||||
child.kill("SIGKILL");
|
child.kill("SIGKILL");
|
||||||
@@ -52,7 +51,6 @@ export async function spawn(options: SpawnOptions): Promise<SpawnResult> {
|
|||||||
}, timeout);
|
}, timeout);
|
||||||
}
|
}
|
||||||
|
|
||||||
// Handle stdout streaming
|
|
||||||
if (child.stdout) {
|
if (child.stdout) {
|
||||||
child.stdout.on("data", (data: Buffer) => {
|
child.stdout.on("data", (data: Buffer) => {
|
||||||
const chunk = data.toString();
|
const chunk = data.toString();
|
||||||
@@ -61,7 +59,6 @@ export async function spawn(options: SpawnOptions): Promise<SpawnResult> {
|
|||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
// Handle stderr streaming
|
|
||||||
if (child.stderr) {
|
if (child.stderr) {
|
||||||
child.stderr.on("data", (data: Buffer) => {
|
child.stderr.on("data", (data: Buffer) => {
|
||||||
const chunk = data.toString();
|
const chunk = data.toString();
|
||||||
@@ -70,7 +67,6 @@ export async function spawn(options: SpawnOptions): Promise<SpawnResult> {
|
|||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
// Handle process completion
|
|
||||||
child.on("close", (exitCode) => {
|
child.on("close", (exitCode) => {
|
||||||
const durationMs = Date.now() - startTime;
|
const durationMs = Date.now() - startTime;
|
||||||
|
|
||||||
@@ -91,15 +87,13 @@ export async function spawn(options: SpawnOptions): Promise<SpawnResult> {
|
|||||||
});
|
});
|
||||||
});
|
});
|
||||||
|
|
||||||
// Handle process errors
|
child.on("error", (_error) => {
|
||||||
child.on("error", (error) => {
|
|
||||||
const durationMs = Date.now() - startTime;
|
const durationMs = Date.now() - startTime;
|
||||||
|
|
||||||
if (timeoutId) {
|
if (timeoutId) {
|
||||||
clearTimeout(timeoutId);
|
clearTimeout(timeoutId);
|
||||||
}
|
}
|
||||||
|
|
||||||
// Still return buffered output even on error
|
|
||||||
resolve({
|
resolve({
|
||||||
stdout: stdoutBuffer,
|
stdout: stdoutBuffer,
|
||||||
stderr: stderrBuffer,
|
stderr: stderrBuffer,
|
||||||
@@ -108,7 +102,6 @@ export async function spawn(options: SpawnOptions): Promise<SpawnResult> {
|
|||||||
});
|
});
|
||||||
});
|
});
|
||||||
|
|
||||||
// Send input if provided
|
|
||||||
if (input && child.stdin) {
|
if (input && child.stdin) {
|
||||||
child.stdin.write(input);
|
child.stdin.write(input);
|
||||||
child.stdin.end();
|
child.stdin.end();
|
||||||
|
|||||||
-172
@@ -1,172 +0,0 @@
|
|||||||
import { table } from "table";
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Print a formatted table with consistent styling
|
|
||||||
* @param rows - Array of string arrays representing table rows
|
|
||||||
* @param options - Optional table configuration
|
|
||||||
*/
|
|
||||||
|
|
||||||
export function tableString(
|
|
||||||
rows: string[][],
|
|
||||||
options?: {
|
|
||||||
title?: string;
|
|
||||||
indent?: string;
|
|
||||||
drawHorizontalLine?: (lineIndex: number, rowCount: number) => boolean;
|
|
||||||
}
|
|
||||||
): string {
|
|
||||||
const {
|
|
||||||
title,
|
|
||||||
indent = "",
|
|
||||||
drawHorizontalLine = (lineIndex: number, rowCount: number) => {
|
|
||||||
return lineIndex === 0 || (options?.title && lineIndex === 1) || lineIndex === rowCount;
|
|
||||||
},
|
|
||||||
} = options || {};
|
|
||||||
|
|
||||||
if (options?.title) {
|
|
||||||
rows.unshift([options.title]);
|
|
||||||
}
|
|
||||||
|
|
||||||
const tableOutput = table(rows, {
|
|
||||||
drawHorizontalLine,
|
|
||||||
border: {
|
|
||||||
topBody: `─`,
|
|
||||||
topJoin: `┬`,
|
|
||||||
topLeft: `┌`,
|
|
||||||
topRight: `┐`,
|
|
||||||
|
|
||||||
bottomBody: `─`,
|
|
||||||
bottomJoin: `┴`,
|
|
||||||
bottomLeft: `└`,
|
|
||||||
bottomRight: `┘`,
|
|
||||||
|
|
||||||
bodyLeft: `│`,
|
|
||||||
bodyRight: `│`,
|
|
||||||
bodyJoin: `│`,
|
|
||||||
|
|
||||||
joinBody: `─`,
|
|
||||||
joinLeft: `├`,
|
|
||||||
joinRight: `┤`,
|
|
||||||
joinJoin: `┼`,
|
|
||||||
},
|
|
||||||
});
|
|
||||||
|
|
||||||
const indentedOutput = tableOutput.split("\n").join(`\n${indent}`).trim();
|
|
||||||
|
|
||||||
return `${indent}${indentedOutput}`;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Print a multi-line string in a formatted box with text wrapping
|
|
||||||
* @param text - The text to display
|
|
||||||
* @param options - Optional configuration for the box
|
|
||||||
*/
|
|
||||||
export function boxString(
|
|
||||||
text: string,
|
|
||||||
options?: {
|
|
||||||
title?: string;
|
|
||||||
maxWidth?: number;
|
|
||||||
indent?: string;
|
|
||||||
padding?: number;
|
|
||||||
}
|
|
||||||
): string {
|
|
||||||
const { title, maxWidth = 80, indent = "", padding = 1 } = options || {};
|
|
||||||
|
|
||||||
// Clean up the text and split into lines
|
|
||||||
const lines = text.trim().split("\n");
|
|
||||||
|
|
||||||
// Word wrap each line to fit within maxWidth
|
|
||||||
const wrappedLines: string[] = [];
|
|
||||||
for (const line of lines) {
|
|
||||||
if (line.length <= maxWidth - padding * 2) {
|
|
||||||
wrappedLines.push(line);
|
|
||||||
} else {
|
|
||||||
// Word wrap the line
|
|
||||||
const words = line.split(" ");
|
|
||||||
let currentLine = "";
|
|
||||||
|
|
||||||
for (const word of words) {
|
|
||||||
const testLine = currentLine ? `${currentLine} ${word}` : word;
|
|
||||||
if (testLine.length <= maxWidth - padding * 2) {
|
|
||||||
currentLine = testLine;
|
|
||||||
} else {
|
|
||||||
if (currentLine) {
|
|
||||||
wrappedLines.push(currentLine);
|
|
||||||
currentLine = word;
|
|
||||||
} else {
|
|
||||||
// Word is too long, break it
|
|
||||||
wrappedLines.push(word.substring(0, maxWidth - padding * 2));
|
|
||||||
currentLine = word.substring(maxWidth - padding * 2);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (currentLine) {
|
|
||||||
wrappedLines.push(currentLine);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Find the maximum line length for box sizing
|
|
||||||
const maxLineLength = Math.max(...wrappedLines.map((line) => line.length));
|
|
||||||
const boxWidth = maxLineLength + padding * 2;
|
|
||||||
|
|
||||||
// Create the box
|
|
||||||
const topBorder = "┌" + "─".repeat(boxWidth) + "┐";
|
|
||||||
const bottomBorder = "└" + "─".repeat(boxWidth) + "┘";
|
|
||||||
const sideBorder = "│";
|
|
||||||
|
|
||||||
let result = "";
|
|
||||||
|
|
||||||
// Add title if provided
|
|
||||||
if (title) {
|
|
||||||
const titleLine = ` ${title} `;
|
|
||||||
const titlePadding = Math.max(0, boxWidth - titleLine.length);
|
|
||||||
result += `${indent}┌${titleLine}${"─".repeat(titlePadding)}┐\n`;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Add top border (or title border)
|
|
||||||
if (!title) {
|
|
||||||
result += `${indent}${topBorder}\n`;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Add content lines
|
|
||||||
for (const line of wrappedLines) {
|
|
||||||
const paddedLine = line.padEnd(maxLineLength);
|
|
||||||
result += `${indent}${sideBorder}${" ".repeat(padding)}${paddedLine}${" ".repeat(padding)}${sideBorder}\n`;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Add bottom border
|
|
||||||
result += `${indent}${bottomBorder}`;
|
|
||||||
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
|
|
||||||
// /**
|
|
||||||
// * Create a simple two-column table for displaying key-value pairs
|
|
||||||
// * @param data - Array of [key, value] pairs
|
|
||||||
// * @param title - Optional table title
|
|
||||||
// * @param indent - Optional indentation string
|
|
||||||
// */
|
|
||||||
// export function printKeyValueTable(
|
|
||||||
// data: [string, string][],
|
|
||||||
// title?: string,
|
|
||||||
// indent?: string
|
|
||||||
// ): void {
|
|
||||||
// const rows: string[][] = [["Key", "Value"], ...data];
|
|
||||||
// const options: Parameters<typeof printTable>[1] = {};
|
|
||||||
// if (title !== undefined) options.title = title;
|
|
||||||
// if (indent !== undefined) options.indent = indent;
|
|
||||||
// printTable(rows, options);
|
|
||||||
// }
|
|
||||||
|
|
||||||
// /**
|
|
||||||
// * Create a path resolution table (specific use case)
|
|
||||||
// * @param pathData - Array of [location, resolvedPath] pairs
|
|
||||||
// * @param indent - Optional indentation string
|
|
||||||
// */
|
|
||||||
// export function printPathTable(pathData: [string, string][], indent?: string): void {
|
|
||||||
// const rows: string[][] = [["Location", "Resolved path"], ...pathData];
|
|
||||||
// const options: Parameters<typeof printTable>[1] = {};
|
|
||||||
// if (indent !== undefined) options.indent = indent;
|
|
||||||
// printTable(rows, options);
|
|
||||||
// }
|
|
||||||
Reference in New Issue
Block a user