Initial commit
This commit is contained in:
commit
fbd5c87dbf
|
|
@ -0,0 +1,5 @@
|
||||||
|
node_modules/
|
||||||
|
.env
|
||||||
|
*.log
|
||||||
|
.DS_Store
|
||||||
|
|
||||||
|
|
@ -0,0 +1,53 @@
|
||||||
|
# Screenshot Server
|
||||||
|
|
||||||
|
Tiny HTTP server that serves the latest screenshot from ~/Desktop, then deletes it.
|
||||||
|
|
||||||
|
## Build (on Mac)
|
||||||
|
|
||||||
|
```bash
|
||||||
|
cd /path/to/screenshot-server
|
||||||
|
go build -o screenshot-server
|
||||||
|
```
|
||||||
|
|
||||||
|
## Install as service
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# Copy binary
|
||||||
|
sudo cp screenshot-server /usr/local/bin/
|
||||||
|
|
||||||
|
# Install launchd service
|
||||||
|
cp com.inou.screenshot-server.plist ~/Library/LaunchAgents/
|
||||||
|
|
||||||
|
# Start it
|
||||||
|
launchctl load ~/Library/LaunchAgents/com.inou.screenshot-server.plist
|
||||||
|
|
||||||
|
# Check status
|
||||||
|
launchctl list | grep screenshot
|
||||||
|
```
|
||||||
|
|
||||||
|
## Usage
|
||||||
|
|
||||||
|
Take a screenshot (Cmd+Shift+4), then:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
curl http://192.168.1.14:9123/screenshot -o screenshot.png
|
||||||
|
```
|
||||||
|
|
||||||
|
## Endpoints
|
||||||
|
|
||||||
|
- `GET /screenshot` - Returns latest screenshot PNG, then deletes it
|
||||||
|
- `GET /health` - Returns "ok"
|
||||||
|
|
||||||
|
## Uninstall
|
||||||
|
|
||||||
|
```bash
|
||||||
|
launchctl unload ~/Library/LaunchAgents/com.inou.screenshot-server.plist
|
||||||
|
rm ~/Library/LaunchAgents/com.inou.screenshot-server.plist
|
||||||
|
sudo rm /usr/local/bin/screenshot-server
|
||||||
|
```
|
||||||
|
|
||||||
|
## Logs
|
||||||
|
|
||||||
|
```bash
|
||||||
|
tail -f /tmp/screenshot-server.log
|
||||||
|
```
|
||||||
|
|
@ -0,0 +1,20 @@
|
||||||
|
<?xml version="1.0" encoding="UTF-8"?>
|
||||||
|
<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
|
||||||
|
<plist version="1.0">
|
||||||
|
<dict>
|
||||||
|
<key>Label</key>
|
||||||
|
<string>com.inou.screenshot-server</string>
|
||||||
|
<key>ProgramArguments</key>
|
||||||
|
<array>
|
||||||
|
<string>/usr/local/bin/screenshot-server</string>
|
||||||
|
</array>
|
||||||
|
<key>RunAtLoad</key>
|
||||||
|
<true/>
|
||||||
|
<key>KeepAlive</key>
|
||||||
|
<true/>
|
||||||
|
<key>StandardOutPath</key>
|
||||||
|
<string>/tmp/screenshot-server.log</string>
|
||||||
|
<key>StandardErrorPath</key>
|
||||||
|
<string>/tmp/screenshot-server.log</string>
|
||||||
|
</dict>
|
||||||
|
</plist>
|
||||||
|
|
@ -0,0 +1,117 @@
|
||||||
|
// screenshot-server: Serves the latest screenshot from ~/Desktop, then deletes it
|
||||||
|
// Run: go build -o screenshot-server && ./screenshot-server
|
||||||
|
// Fetch: curl http://mac-ip:9123/screenshot -o screenshot.png
|
||||||
|
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"log"
|
||||||
|
"net/http"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"sort"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
port = ":9123"
|
||||||
|
desktopDir = "" // Will be set to ~/Desktop
|
||||||
|
)
|
||||||
|
|
||||||
|
func getDesktopPath() string {
|
||||||
|
home, err := os.UserHomeDir()
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal("Cannot find home directory:", err)
|
||||||
|
}
|
||||||
|
return filepath.Join(home, "Desktop")
|
||||||
|
}
|
||||||
|
|
||||||
|
func findLatestScreenshot(desktop string) (string, error) {
|
||||||
|
entries, err := os.ReadDir(desktop)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
var screenshots []os.DirEntry
|
||||||
|
for _, e := range entries {
|
||||||
|
name := e.Name()
|
||||||
|
if strings.HasPrefix(name, "Screenshot") && strings.HasSuffix(name, ".png") {
|
||||||
|
screenshots = append(screenshots, e)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(screenshots) == 0 {
|
||||||
|
return "", fmt.Errorf("no screenshots found")
|
||||||
|
}
|
||||||
|
|
||||||
|
// Sort by modification time (newest first)
|
||||||
|
sort.Slice(screenshots, func(i, j int) bool {
|
||||||
|
infoI, _ := screenshots[i].Info()
|
||||||
|
infoJ, _ := screenshots[j].Info()
|
||||||
|
return infoI.ModTime().After(infoJ.ModTime())
|
||||||
|
})
|
||||||
|
|
||||||
|
return filepath.Join(desktop, screenshots[0].Name()), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func screenshotHandler(w http.ResponseWriter, r *http.Request) {
|
||||||
|
if r.Method != http.MethodGet {
|
||||||
|
http.Error(w, "GET only", http.StatusMethodNotAllowed)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
desktop := getDesktopPath()
|
||||||
|
path, err := findLatestScreenshot(desktop)
|
||||||
|
if err != nil {
|
||||||
|
http.Error(w, err.Error(), http.StatusNotFound)
|
||||||
|
log.Println("No screenshot:", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Open and serve
|
||||||
|
f, err := os.Open(path)
|
||||||
|
if err != nil {
|
||||||
|
http.Error(w, "Cannot open file", http.StatusInternalServerError)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer f.Close()
|
||||||
|
|
||||||
|
filename := filepath.Base(path)
|
||||||
|
w.Header().Set("Content-Type", "image/png")
|
||||||
|
w.Header().Set("X-Screenshot-Name", filename)
|
||||||
|
|
||||||
|
_, err = io.Copy(w, f)
|
||||||
|
if err != nil {
|
||||||
|
log.Println("Error sending:", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// Delete after successful send
|
||||||
|
f.Close()
|
||||||
|
if err := os.Remove(path); err != nil {
|
||||||
|
log.Println("Warning: could not delete:", err)
|
||||||
|
} else {
|
||||||
|
log.Println("Served and deleted:", filename)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func healthHandler(w http.ResponseWriter, r *http.Request) {
|
||||||
|
w.Write([]byte("ok"))
|
||||||
|
}
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
http.HandleFunc("/screenshot", screenshotHandler)
|
||||||
|
http.HandleFunc("/health", healthHandler)
|
||||||
|
|
||||||
|
log.Printf("Screenshot server starting on %s", port)
|
||||||
|
log.Printf("Desktop: %s", getDesktopPath())
|
||||||
|
log.Printf("Endpoints:")
|
||||||
|
log.Printf(" GET /screenshot - fetch & delete latest screenshot")
|
||||||
|
log.Printf(" GET /health - health check")
|
||||||
|
|
||||||
|
if err := http.ListenAndServe(port, nil); err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue