12000e523c
All checks were successful
Beta Release / beta (push) Successful in 1m1s
- Fix token count reset on app restart: persist realTokens in conversation.json - Fix token/context window values: Studio 150K (summarize at 120K), Terminal 100K - Fix table rendering in terminal tab: correct thead/tbody display model - Fix copy button always top-right in Studio code blocks - Add markdown horizontal rule (---) support in Studio and Terminal - Fix bullet list double dot: remove CSS ::before duplicate bullet point - Add image attachments support (VLM description, file mentions @file.ext) - Add sudo detection with cache (sync.Once) - Fix message content serialization (TextContent wrapper) - Guide AI to use read_file instead of cat in studio prompt 💘 Generated with Crush Assisted-by: GLM-5.1 via Crush <crush@charm.land>
105 lines
2.2 KiB
Go
105 lines
2.2 KiB
Go
package api
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"fmt"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"sync/atomic"
|
|
"time"
|
|
|
|
"github.com/muyue/muyue/internal/config"
|
|
)
|
|
|
|
var imageDir string
|
|
|
|
func init() {
|
|
dir, err := config.ConfigDir()
|
|
if err != nil {
|
|
dir = "/tmp/muyue"
|
|
}
|
|
imageDir = filepath.Join(dir, "images")
|
|
os.MkdirAll(imageDir, 0755)
|
|
}
|
|
|
|
var imageCounter uint64
|
|
|
|
func saveImage(dataURI, filename, mimeType string) (string, error) {
|
|
parts := strings.SplitN(dataURI, ",", 2)
|
|
if len(parts) != 2 {
|
|
return "", fmt.Errorf("invalid data URI")
|
|
}
|
|
encoded := parts[1]
|
|
|
|
decoded, err := base64.StdEncoding.DecodeString(encoded)
|
|
if err != nil {
|
|
return "", fmt.Errorf("base64 decode: %w", err)
|
|
}
|
|
|
|
id := fmt.Sprintf("%d-%d", time.Now().UnixMilli(), atomic.AddUint64(&imageCounter, 1))
|
|
ext := ".png"
|
|
switch mimeType {
|
|
case "image/jpeg":
|
|
ext = ".jpg"
|
|
case "image/webp":
|
|
ext = ".webp"
|
|
}
|
|
|
|
filePath := filepath.Join(imageDir, id+ext)
|
|
if err := os.WriteFile(filePath, decoded, 0600); err != nil {
|
|
return "", fmt.Errorf("write image: %w", err)
|
|
}
|
|
|
|
return id + ext, nil
|
|
}
|
|
|
|
func imagePath(id string) string {
|
|
return filepath.Join(imageDir, filepath.Base(id))
|
|
}
|
|
|
|
func cleanupImages(ids []string) {
|
|
for _, id := range ids {
|
|
p := imagePath(id)
|
|
if err := os.Remove(p); err != nil && !os.IsNotExist(err) {
|
|
log.Printf("[images] failed to delete %s: %v", id, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (s *Server) handleServeImage(w http.ResponseWriter, r *http.Request) {
|
|
if r.Method != "GET" {
|
|
writeError(w, "GET only", http.StatusMethodNotAllowed)
|
|
return
|
|
}
|
|
|
|
id := strings.TrimPrefix(r.URL.Path, "/api/images/")
|
|
if id == "" {
|
|
writeError(w, "image id required", http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
filePath := imagePath(id)
|
|
if _, err := os.Stat(filePath); err != nil {
|
|
writeError(w, "image not found", http.StatusNotFound)
|
|
return
|
|
}
|
|
|
|
ext := strings.ToLower(filepath.Ext(id))
|
|
switch ext {
|
|
case ".jpg", ".jpeg":
|
|
w.Header().Set("Content-Type", "image/jpeg")
|
|
case ".png":
|
|
w.Header().Set("Content-Type", "image/png")
|
|
case ".webp":
|
|
w.Header().Set("Content-Type", "image/webp")
|
|
default:
|
|
w.Header().Set("Content-Type", "application/octet-stream")
|
|
}
|
|
w.Header().Set("Cache-Control", "public, max-age=86400")
|
|
|
|
http.ServeFile(w, r, filePath)
|
|
}
|