forked from bots-garden/ori
| ✨ Workspace panel, selectors, previews, desktop app, sandbox template, resizable file tree, light/dark theme | 1 | package files |
| 2 | ||
| 3 | import ( | |
| 4 | "mime" | |
| 5 | "net/http" | |
| 6 | "os" | |
| 7 | "path/filepath" | |
| 8 | "strings" | |
| 9 | ) | |
| 10 | ||
| 11 | // imageTypes pins the Content-Type of the image formats the preview pane | |
| 12 | // renders, independently of the host's mime database (which may lack | |
| 13 | // entries such as .webp or .avif, or spell .ico differently). | |
| 14 | var imageTypes = map[string]string{ | |
| 15 | ".avif": "image/avif", | |
| 16 | ".bmp": "image/bmp", | |
| 17 | ".gif": "image/gif", | |
| 18 | ".ico": "image/x-icon", | |
| 19 | ".jpeg": "image/jpeg", | |
| 20 | ".jpg": "image/jpeg", | |
| 21 | ".png": "image/png", | |
| 22 | ".svg": "image/svg+xml", | |
| 23 | ".webp": "image/webp", | |
| 24 | } | |
| 25 | ||
| ✨ Switch the ACP adapter to @agentclientprotocol/claude-agent-acp (template 0.0.2) | 26 | // textTypes pins the text formats the preview pane renders itself, for the |
| 27 | // same reason as imageTypes: Go seeds its mime table from the host (macOS's | |
| 28 | // /etc/apache2/mime.types leaves text/markdown commented out), so without a | |
| 29 | // pin a .md file is served as text/plain on one machine and text/markdown on | |
| 30 | // another. | |
| 31 | var textTypes = map[string]string{ | |
| 32 | ".markdown": "text/markdown; charset=utf-8", | |
| 33 | ".md": "text/markdown; charset=utf-8", | |
| 34 | } | |
| 35 | ||
| ✨ Workspace panel, selectors, previews, desktop app, sandbox template, resizable file tree, light/dark theme | 36 | // ContentTypeFor returns the Content-Type the raw endpoint serves a file |
| ✨ Switch the ACP adapter to @agentclientprotocol/claude-agent-acp (template 0.0.2) | 37 | // under: pinned image and text types first, then the host mime database, and |
| 38 | // a content sniff of the first bytes as a last resort. | |
| ✨ Workspace panel, selectors, previews, desktop app, sandbox template, resizable file tree, light/dark theme | 39 | // |
| 40 | // Example: | |
| 41 | // | |
| 42 | // files.ContentTypeFor("logo.webp", nil) // "image/webp" | |
| ✨ Switch the ACP adapter to @agentclientprotocol/claude-agent-acp (template 0.0.2) | 43 | // files.ContentTypeFor("README.md", nil) // "text/markdown; charset=utf-8" |
| ✨ Workspace panel, selectors, previews, desktop app, sandbox template, resizable file tree, light/dark theme | 44 | func ContentTypeFor(path string, head []byte) string { |
| 45 | ext := strings.ToLower(filepath.Ext(path)) | |
| 46 | if typ, ok := imageTypes[ext]; ok { | |
| 47 | return typ | |
| 48 | } | |
| ✨ Switch the ACP adapter to @agentclientprotocol/claude-agent-acp (template 0.0.2) | 49 | if typ, ok := textTypes[ext]; ok { |
| 50 | return typ | |
| 51 | } | |
| ✨ Workspace panel, selectors, previews, desktop app, sandbox template, resizable file tree, light/dark theme | 52 | if typ := mime.TypeByExtension(ext); typ != "" { |
| 53 | return typ | |
| 54 | } | |
| 55 | return http.DetectContentType(head) | |
| 56 | } | |
| 57 | ||
| 58 | // rawFile streams a file's bytes as-is, for the browser to render (images) | |
| 59 | // or download. Unlike readFile it has no size cap and no text requirement: | |
| 60 | // http.ServeContent streams and honours Range requests. | |
| 61 | func (s *Service) rawFile(w http.ResponseWriter, r *http.Request) { | |
| 62 | if r.URL.Query().Get("path") == "" { | |
| 63 | writeError(w, http.StatusBadRequest, "path query parameter is required") | |
| 64 | return | |
| 65 | } | |
| 66 | path := s.resolve(r.URL.Query().Get("path")) | |
| 67 | info, err := os.Stat(path) | |
| 68 | if err != nil { | |
| 69 | writeFSError(w, err) | |
| 70 | return | |
| 71 | } | |
| 72 | if info.IsDir() { | |
| 73 | writeError(w, http.StatusBadRequest, "path is a directory") | |
| 74 | return | |
| 75 | } | |
| 76 | file, err := os.Open(path) | |
| 77 | if err != nil { | |
| 78 | writeFSError(w, err) | |
| 79 | return | |
| 80 | } | |
| 81 | // Read-only handle: a Close error carries no information the client | |
| 82 | // could act on, and the response headers are already committed. | |
| 83 | defer func() { _ = file.Close() }() | |
| 84 | ||
| 85 | head := make([]byte, 512) | |
| 86 | n, _ := file.Read(head) | |
| 87 | if _, err := file.Seek(0, 0); err != nil { | |
| 88 | writeFSError(w, err) | |
| 89 | return | |
| 90 | } | |
| 91 | w.Header().Set("Content-Type", ContentTypeFor(path, head[:n])) | |
| 92 | w.Header().Set("X-Content-Type-Options", "nosniff") | |
| 93 | http.ServeContent(w, r, info.Name(), info.ModTime(), file) | |
| 94 | } |