@brainstack/plantuml-mcp

A Model Context Protocol (MCP) server for generating UML diagrams using PlantUML. This package enables AI assistants to create professional diagrams from text descriptions through the MCP standard.

Features

Easy Diagram Generation: Create UML diagrams from PlantUML code
Multiple Output Formats: Generate diagrams in PNG, SVG, or text formats
Automatic Branding: Adds a customized footer to all diagrams
MCP Standard: Compatible with the Model Context Protocol for AI tools
Dark Theme Support: Works well with dark-themed diagrams

Installation

Using npm

npm install @brainstack/plantuml-mcp

Using the pre-configured scripts

Clone the repository:

git clone https://github.com/mouimet-infinisoft/augment-code-mcp.git
cd augment-code-mcp/plantuml-mcp

Make the run script executable:
```
chmod +x config/run-plantuml-mcp.sh
```
Run the MCP server:
```
./config/run-plantuml-mcp.sh
```

Configuration

Environment Variables

Create a .env file in your project root with these options:

# Optional: Custom PlantUML server URL
PLANTUML_SERVER_URL=https://www.plantuml.com/plantuml

Configuration Files

The package includes pre-configured files in the config directory:

mcp-config-plantuml.json: MCP server configuration
run-plantuml-mcp.sh: Script to run the server
mcp-config-plantuml-wsl.json: Configuration for WSL environments

Custom Server Configuration

If you're using the package programmatically, you can configure the server:

const { createServer, startServer } = require('@brainstack/plantuml-mcp');

// Create a server with custom PlantUML URL
const server = createServer({
  plantUmlServerUrl: 'https://your-custom-plantuml-server.com'
});

// Start the server
startServer(server);

Usage with Augment Code

Install the Augment Code extension for VSCode
Run the PlantUML MCP server using the provided script
Configure Augment Code to use this MCP server:
- Open VSCode settings
- Search for "Augment Code MCP"
- Add the PlantUML MCP server configuration
Generate diagrams using natural language in VSCode

Usage with WSL 2 (Windows)

For Windows users with WSL 2:

Use the WSL-specific configuration file:

cp config/mcp-config-plantuml-wsl.json config/mcp-config-plantuml.json

Run the server from your WSL environment:
```
./config/run-plantuml-mcp.sh
```
Configure Augment Code to connect to the WSL endpoint

Diagram Examples

Class Diagram

@startuml
!theme spacelab
skinparam backgroundColor #333333
skinparam classFontColor #FFFFFF

class Car {
  +String make
  +String model
  +int year
  +start()
  +stop()
}
class Driver {
  +String name
  +String licenseNumber
  +drive(Car car)
}
Driver -- Car : drives >
@enduml

Sequence Diagram

@startuml
!theme spacelab
skinparam backgroundColor #333333
skinparam sequenceParticipantFontColor #FFFFFF

participant User
participant System
participant Database

User -> System: Login request
System -> Database: Validate credentials
Database --> System: Authentication result
System --> User: Login response
@enduml

Entity Relationship Diagram

@startuml
!theme spacelab
skinparam backgroundColor #333333
skinparam entityFontColor #FFFFFF

entity "Customer" as customer {
  * id : number <<PK>>
  --
  * name : text
  * email : text
  created_at : datetime
}

entity "Order" as order {
  * id : number <<PK>>
  --
  * customer_id : number <<FK>>
  * amount : decimal
  status : text
  created_at : datetime
}

customer ||--o{ order : places
@enduml

API Reference

MCP Tool: `generateDiagram`

This MCP server provides a single tool for generating diagrams:

Parameter	Type	Description	Default
`plantUmlCode`	string	The PlantUML code to generate a diagram from	(required)
`format`	string	The format of the diagram: 'png', 'svg', or 'txt'	'png'

Response:

{
  "url": "https://www.plantuml.com/plantuml/png/...",
  "format": "png"
}

Styling Recommendations

For best results with dark-themed diagrams:

Add the spacelab theme:
```
!theme spacelab
```
Set a dark background:
```
skinparam backgroundColor #333333
```

Use light text colors:

skinparam classFontColor #FFFFFF
skinparam entityFontColor #FFFFFF
skinparam sequenceParticipantFontColor #FFFFFF

Add accent colors for arrows and borders:

skinparam arrowColor #4285F4
skinparam classBorderColor #666666

Footer Annotation

All diagrams generated with this MCP server automatically include a footer annotation:

Generated by @brainstack/plantuml-mcp | [current date and time]

This footer is styled with a blue highlight for the package name and a lighter gray for the timestamp.

Resources

PlantUML Official Website - For syntax reference and examples
Model Context Protocol - For MCP standard details
Augment Code - For integration with VSCode