Add missing cmd/iso-bot files
This commit is contained in:
Hoid
parent
6a9562c406
commit
4e4acaa78d
2 changed files with 358 additions and 0 deletions
214
cmd/iso-bot/gen_testdata.go
Normal file
214
cmd/iso-bot/gen_testdata.go
Normal file
|
|
@ -0,0 +1,214 @@
|
|||
//go:build ignore
|
||||
|
||||
// Generate test data for development.
|
||||
// Run with: go run gen_testdata.go
|
||||
package main
|
||||
|
||||
import (
|
||||
"image"
|
||||
"image/color"
|
||||
"image/draw"
|
||||
"image/png"
|
||||
"log"
|
||||
"os"
|
||||
"path/filepath"
|
||||
)
|
||||
|
||||
func main() {
|
||||
// Create testdata directory
|
||||
if err := os.MkdirAll("testdata", 0755); err != nil {
|
||||
log.Fatalf("Failed to create testdata directory: %v", err)
|
||||
}
|
||||
|
||||
// Generate synthetic D2R screenshot
|
||||
img := generateD2RScreenshot()
|
||||
|
||||
// Save as PNG
|
||||
filename := filepath.Join("testdata", "d2r_1080p.png")
|
||||
file, err := os.Create(filename)
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to create test image file: %v", err)
|
||||
}
|
||||
defer file.Close()
|
||||
|
||||
if err := png.Encode(file, img); err != nil {
|
||||
log.Fatalf("Failed to encode test image: %v", err)
|
||||
}
|
||||
|
||||
log.Printf("Generated test image: %s", filename)
|
||||
}
|
||||
|
||||
func generateD2RScreenshot() image.Image {
|
||||
// Create 1920x1080 black image
|
||||
img := image.NewRGBA(image.Rect(0, 0, 1920, 1080))
|
||||
|
||||
// Fill with dark background (simulating game background)
|
||||
darkGray := color.RGBA{20, 20, 25, 255}
|
||||
draw.Draw(img, img.Bounds(), &image.Uniform{darkGray}, image.Point{}, draw.Src)
|
||||
|
||||
// Add some noise/texture to make it more realistic
|
||||
for y := 0; y < 1080; y += 5 {
|
||||
for x := 0; x < 1920; x += 5 {
|
||||
noise := color.RGBA{
|
||||
uint8(15 + (x+y)%20),
|
||||
uint8(15 + (x+y)%25),
|
||||
uint8(20 + (x+y)%30),
|
||||
255,
|
||||
}
|
||||
fillRect(img, image.Rect(x, y, x+3, y+3), noise)
|
||||
}
|
||||
}
|
||||
|
||||
// Health orb (red circle) - Region: (28, 545, 198, 715)
|
||||
healthRegion := image.Rect(28, 545, 198, 715)
|
||||
healthColor := color.RGBA{220, 20, 20, 255} // Bright red for health
|
||||
drawOrb(img, healthRegion, healthColor, 0.75) // 75% filled
|
||||
|
||||
// Mana orb (blue circle) - Region: (1722, 545, 1892, 715)
|
||||
manaRegion := image.Rect(1722, 545, 1892, 715)
|
||||
manaColor := color.RGBA{20, 20, 220, 255} // Bright blue for mana
|
||||
drawOrb(img, manaRegion, manaColor, 0.60) // 60% filled
|
||||
|
||||
// XP bar - Region: (0, 1058, 1920, 1080)
|
||||
xpRegion := image.Rect(0, 1058, 1920, 1080)
|
||||
xpColor := color.RGBA{255, 215, 0, 255} // Gold color
|
||||
fillRect(img, xpRegion, color.RGBA{40, 40, 40, 255}) // Dark background
|
||||
// Partial XP fill
|
||||
xpFillWidth := int(float64(xpRegion.Dx()) * 0.3) // 30% XP
|
||||
xpFillRegion := image.Rect(xpRegion.Min.X, xpRegion.Min.Y, xpRegion.Min.X+xpFillWidth, xpRegion.Max.Y)
|
||||
fillRect(img, xpFillRegion, xpColor)
|
||||
|
||||
// Belt - Region: (838, 1010, 1082, 1058)
|
||||
beltRegion := image.Rect(838, 1010, 1082, 1058)
|
||||
beltColor := color.RGBA{80, 60, 40, 255} // Brown belt background
|
||||
fillRect(img, beltRegion, beltColor)
|
||||
|
||||
// Add some belt slots
|
||||
slotWidth := beltRegion.Dx() / 4
|
||||
for i := 0; i < 4; i++ {
|
||||
slotX := beltRegion.Min.X + i*slotWidth + 5
|
||||
slotRegion := image.Rect(slotX, beltRegion.Min.Y+5, slotX+slotWidth-10, beltRegion.Max.Y-5)
|
||||
slotColor := color.RGBA{60, 45, 30, 255}
|
||||
fillRect(img, slotRegion, slotColor)
|
||||
|
||||
// Add a potion in first slot
|
||||
if i == 0 {
|
||||
potionColor := color.RGBA{200, 50, 50, 255} // Red potion
|
||||
potionRegion := image.Rect(slotX+5, beltRegion.Min.Y+10, slotX+slotWidth-15, beltRegion.Max.Y-10)
|
||||
fillRect(img, potionRegion, potionColor)
|
||||
}
|
||||
}
|
||||
|
||||
// Minimap - Region: (1600, 0, 1920, 320)
|
||||
minimapRegion := image.Rect(1600, 0, 1920, 320)
|
||||
minimapColor := color.RGBA{40, 40, 50, 255} // Dark blue-gray
|
||||
fillRect(img, minimapRegion, minimapColor)
|
||||
|
||||
// Add some minimap elements
|
||||
// Player dot (center)
|
||||
centerX := minimapRegion.Min.X + minimapRegion.Dx()/2
|
||||
centerY := minimapRegion.Min.Y + minimapRegion.Dy()/2
|
||||
playerDot := image.Rect(centerX-3, centerY-3, centerX+3, centerY+3)
|
||||
fillRect(img, playerDot, color.RGBA{255, 255, 255, 255}) // White player dot
|
||||
|
||||
// Some terrain
|
||||
for i := 0; i < 50; i++ {
|
||||
x := minimapRegion.Min.X + (i*7)%minimapRegion.Dx()
|
||||
y := minimapRegion.Min.Y + (i*11)%minimapRegion.Dy()
|
||||
terrainColor := color.RGBA{60 + uint8(i%30), 80 + uint8(i%20), 40 + uint8(i%25), 255}
|
||||
terrainDot := image.Rect(x, y, x+2, y+2)
|
||||
fillRect(img, terrainDot, terrainColor)
|
||||
}
|
||||
|
||||
// Skills - left and right
|
||||
// Left skill: (194, 1036, 246, 1088)
|
||||
leftSkillRegion := image.Rect(194, 1036, 246, 1088)
|
||||
skillColor := color.RGBA{100, 100, 150, 255}
|
||||
fillRect(img, leftSkillRegion, skillColor)
|
||||
|
||||
// Right skill: (1674, 1036, 1726, 1088)
|
||||
rightSkillRegion := image.Rect(1674, 1036, 1726, 1088)
|
||||
fillRect(img, rightSkillRegion, skillColor)
|
||||
|
||||
// Add some item drops on the ground (colored text rectangles)
|
||||
items := []struct {
|
||||
text string
|
||||
color color.RGBA
|
||||
x, y int
|
||||
}{
|
||||
{"Shako", color.RGBA{255, 165, 0, 255}, 400, 600}, // Orange unique
|
||||
{"Rune", color.RGBA{255, 215, 0, 255}, 500, 650}, // Gold
|
||||
{"Magic Item", color.RGBA{100, 100, 255, 255}, 600, 700}, // Blue magic
|
||||
}
|
||||
|
||||
for _, item := range items {
|
||||
textRegion := image.Rect(item.x, item.y, item.x+len(item.text)*8, item.y+15)
|
||||
fillRect(img, textRegion, item.color)
|
||||
}
|
||||
|
||||
return img
|
||||
}
|
||||
|
||||
// drawOrb draws a circular orb with fill percentage
|
||||
func drawOrb(img *image.RGBA, region image.Rectangle, fillColor color.RGBA, fillPct float64) {
|
||||
centerX := region.Min.X + region.Dx()/2
|
||||
centerY := region.Min.Y + region.Dy()/2
|
||||
radius := region.Dx() / 2
|
||||
|
||||
// Draw orb background (dark)
|
||||
darkColor := color.RGBA{40, 40, 40, 255}
|
||||
fillCircle(img, centerX, centerY, radius, darkColor)
|
||||
|
||||
// Draw border
|
||||
borderColor := color.RGBA{100, 100, 100, 255}
|
||||
drawCircleBorder(img, centerX, centerY, radius, borderColor, 2)
|
||||
|
||||
// Draw filled portion (from bottom up)
|
||||
if fillPct > 0 {
|
||||
fillHeight := int(float64(region.Dy()) * fillPct)
|
||||
fillStartY := region.Max.Y - fillHeight
|
||||
|
||||
// Fill the circle partially
|
||||
for y := fillStartY; y < region.Max.Y; y++ {
|
||||
for x := region.Min.X; x < region.Max.X; x++ {
|
||||
dx := x - centerX
|
||||
dy := y - centerY
|
||||
if dx*dx + dy*dy <= radius*radius {
|
||||
img.Set(x, y, fillColor)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// fillCircle fills a circle with the given color
|
||||
func fillCircle(img *image.RGBA, centerX, centerY, radius int, col color.RGBA) {
|
||||
for y := centerY - radius; y <= centerY + radius; y++ {
|
||||
for x := centerX - radius; x <= centerX + radius; x++ {
|
||||
dx := x - centerX
|
||||
dy := y - centerY
|
||||
if dx*dx + dy*dy <= radius*radius {
|
||||
img.Set(x, y, col)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// drawCircleBorder draws a circle border
|
||||
func drawCircleBorder(img *image.RGBA, centerX, centerY, radius int, col color.RGBA, thickness int) {
|
||||
for y := centerY - radius - thickness; y <= centerY + radius + thickness; y++ {
|
||||
for x := centerX - radius - thickness; x <= centerX + radius + thickness; x++ {
|
||||
dx := x - centerX
|
||||
dy := y - centerY
|
||||
dist := dx*dx + dy*dy
|
||||
if dist <= (radius+thickness)*(radius+thickness) && dist >= (radius-thickness)*(radius-thickness) {
|
||||
img.Set(x, y, col)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// fillRect fills a rectangle with the given color
|
||||
func fillRect(img *image.RGBA, rect image.Rectangle, col color.RGBA) {
|
||||
draw.Draw(img, rect, &image.Uniform{col}, image.Point{}, draw.Src)
|
||||
}
|
||||
144
cmd/iso-bot/main.go
Normal file
144
cmd/iso-bot/main.go
Normal file
|
|
@ -0,0 +1,144 @@
|
|||
// iso-bot — isometric game bot engine.
|
||||
//
|
||||
// Usage:
|
||||
//
|
||||
// iso-bot --game d2r --routine mephisto --api :8080
|
||||
// iso-bot --dev --api :8080 --capture-file testdata/d2r_1080p.png
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"flag"
|
||||
"fmt"
|
||||
"log"
|
||||
"os"
|
||||
"os/signal"
|
||||
"path/filepath"
|
||||
"syscall"
|
||||
|
||||
"git.cloonar.com/openclawd/iso-bot/pkg/api"
|
||||
"git.cloonar.com/openclawd/iso-bot/pkg/engine"
|
||||
"git.cloonar.com/openclawd/iso-bot/pkg/engine/capture/backends"
|
||||
"git.cloonar.com/openclawd/iso-bot/pkg/plugin"
|
||||
"git.cloonar.com/openclawd/iso-bot/plugins/d2r"
|
||||
)
|
||||
|
||||
func main() {
|
||||
// Command line flags
|
||||
game := flag.String("game", "d2r", "game plugin to load")
|
||||
routine := flag.String("routine", "mephisto", "farming routine to run")
|
||||
apiAddr := flag.String("api", ":8080", "API server address")
|
||||
configFile := flag.String("config", "", "path to config file (YAML)")
|
||||
devMode := flag.Bool("dev", false, "enable development mode (no input simulation, serves dev dashboard)")
|
||||
captureFile := flag.String("capture-file", "", "file to capture from (for dev/testing)")
|
||||
captureDir := flag.String("capture-dir", "", "directory to capture from (for dev/testing)")
|
||||
flag.Parse()
|
||||
|
||||
log.Printf("iso-bot starting: game=%s routine=%s api=%s dev=%t", *game, *routine, *apiAddr, *devMode)
|
||||
|
||||
if *configFile != "" {
|
||||
log.Printf("Loading config from %s", *configFile)
|
||||
// TODO: Load YAML config
|
||||
}
|
||||
|
||||
// Set up capture source
|
||||
var captureSource string
|
||||
var captureConfig map[string]interface{}
|
||||
|
||||
if *devMode {
|
||||
if *captureFile != "" {
|
||||
captureSource = *captureFile
|
||||
captureConfig = map[string]interface{}{
|
||||
"path": *captureFile,
|
||||
"type": "image",
|
||||
"loop": true,
|
||||
}
|
||||
} else if *captureDir != "" {
|
||||
captureSource = *captureDir
|
||||
captureConfig = map[string]interface{}{
|
||||
"path": *captureDir,
|
||||
"type": "directory",
|
||||
"frame_rate": 1.0, // 1 FPS for directory scanning
|
||||
"loop": true,
|
||||
}
|
||||
} else {
|
||||
// Default test image
|
||||
testDataPath := filepath.Join("testdata", "d2r_1080p.png")
|
||||
captureSource = testDataPath
|
||||
captureConfig = map[string]interface{}{
|
||||
"path": testDataPath,
|
||||
"type": "image",
|
||||
"loop": true,
|
||||
}
|
||||
log.Printf("No capture source specified, using default: %s", testDataPath)
|
||||
}
|
||||
} else {
|
||||
// Production mode - would use window/monitor capture
|
||||
log.Fatal("Production mode not implemented yet. Use --dev for development.")
|
||||
}
|
||||
|
||||
// Create capture backend
|
||||
registry := backends.GetDefault()
|
||||
source, err := registry.Create(backends.BackendFile, captureConfig)
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to create capture source %s: %v", captureSource, err)
|
||||
}
|
||||
defer source.Close()
|
||||
|
||||
// Load game plugin
|
||||
var gamePlugin plugin.Plugin
|
||||
switch *game {
|
||||
case "d2r":
|
||||
gamePlugin = d2r.New()
|
||||
default:
|
||||
log.Fatalf("Unknown game plugin: %s", *game)
|
||||
}
|
||||
|
||||
// Create engine
|
||||
eng, err := engine.NewEngine(source, gamePlugin, *devMode)
|
||||
if err != nil {
|
||||
log.Fatalf("Failed to create engine: %v", err)
|
||||
}
|
||||
|
||||
// Set up web root for dev dashboard
|
||||
webRoot := "web/dev"
|
||||
if _, err := os.Stat(webRoot); os.IsNotExist(err) {
|
||||
// Try relative to executable
|
||||
execDir, _ := filepath.Abs(filepath.Dir(os.Args[0]))
|
||||
webRoot = filepath.Join(execDir, "web", "dev")
|
||||
}
|
||||
|
||||
// Start API server
|
||||
server := api.NewServer(*apiAddr, eng, webRoot)
|
||||
go func() {
|
||||
log.Printf("Starting API server on %s", *apiAddr)
|
||||
if *devMode {
|
||||
log.Printf("Dev dashboard will be available at http://localhost%s", *apiAddr)
|
||||
}
|
||||
if err := server.Start(); err != nil {
|
||||
log.Fatalf("API server error: %v", err)
|
||||
}
|
||||
}()
|
||||
|
||||
// Start engine
|
||||
ctx, cancel := context.WithCancel(context.Background())
|
||||
defer cancel()
|
||||
|
||||
go func() {
|
||||
if err := eng.Start(ctx); err != nil {
|
||||
log.Printf("Engine error: %v", err)
|
||||
}
|
||||
}()
|
||||
|
||||
fmt.Printf("Game: %s, Routine: %s, Dev Mode: %t\n", *game, *routine, *devMode)
|
||||
log.Printf("Capture source: %s", captureSource)
|
||||
|
||||
// Wait for interrupt
|
||||
sig := make(chan os.Signal, 1)
|
||||
signal.Notify(sig, syscall.SIGINT, syscall.SIGTERM)
|
||||
<-sig
|
||||
|
||||
log.Println("Shutting down...")
|
||||
cancel()
|
||||
eng.Stop()
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue