mirror of
https://gitea.com/gitea/gitea-mcp.git
synced 2026-08-03 15:49:23 +02:00
feat(issue): add attachment read downloads (#218)
- Add a read-only `attachment_read` tool for issue and issue comment attachments.
- Use the existing Gitea SDK attachment metadata APIs for `list` and `get`.
- Add authenticated `/attachments/{uuid}` downloads in the MCP server for `download`, with small image inline results and streamed file downloads for larger content.
- Add `-max-inline-attachment-bytes` / `GITEA_MAX_INLINE_ATTACHMENT_BYTES` to configure inline image limits.
- Update the tool tables in all README variants.
Refs: #209
Reviewed-on: https://gitea.com/gitea/gitea-mcp/pulls/218
Reviewed-by: Lunny Xiao <xiaolunwen@gmail.com>
Co-authored-by: yp05327 <576951401@qq.com>
This commit is contained in:
@@ -151,6 +151,7 @@ Once configured, try `list all my repositories` in the chat box.
|
|||||||
| package_read | Packages | Read | Read package registry: list packages, list versions, or get a version |
|
| package_read | Packages | Read | Read package registry: list packages, list versions, or get a version |
|
||||||
| package_write | Packages | Write | Delete a package version (irreversible) |
|
| package_write | Packages | Write | Delete a package version (irreversible) |
|
||||||
| list_issues | Issue | Read | List repository issues |
|
| list_issues | Issue | Read | List repository issues |
|
||||||
|
| attachment_read | Issue | Read | Read issue/comment attachments: list metadata, get metadata, or download content |
|
||||||
| issue_read | Issue | Read | Read issue: details, comments, or labels |
|
| issue_read | Issue | Read | Read issue: details, comments, or labels |
|
||||||
| issue_write | Issue | Write | Write issues: create, update, manage comments and labels |
|
| issue_write | Issue | Write | Write issues: create, update, manage comments and labels |
|
||||||
| list_pull_requests | Pull Request | Read | List repository pull requests |
|
| list_pull_requests | Pull Request | Read | List repository pull requests |
|
||||||
|
|||||||
@@ -151,6 +151,7 @@ Cursor 等客户端可使用 stdio 命令:
|
|||||||
| package_read | 软件包 | 读取 | 读取软件包注册表:列出软件包、列出版本或获取某个版本 |
|
| package_read | 软件包 | 读取 | 读取软件包注册表:列出软件包、列出版本或获取某个版本 |
|
||||||
| package_write | 软件包 | 写入 | 删除软件包版本(不可恢复) |
|
| package_write | 软件包 | 写入 | 删除软件包版本(不可恢复) |
|
||||||
| list_issues | 问题 | 读取 | 列出仓库问题 |
|
| list_issues | 问题 | 读取 | 列出仓库问题 |
|
||||||
|
| attachment_read | 问题 | 读取 | 读取问题/评论附件:列出元数据、获取元数据或下载内容 |
|
||||||
| issue_read | 问题 | 读取 | 读取问题:详情、评论或标签 |
|
| issue_read | 问题 | 读取 | 读取问题:详情、评论或标签 |
|
||||||
| issue_write | 问题 | 写入 | 写入问题:创建、更新、管理评论和标签 |
|
| issue_write | 问题 | 写入 | 写入问题:创建、更新、管理评论和标签 |
|
||||||
| list_pull_requests | 拉取请求 | 读取 | 列出仓库拉取请求 |
|
| list_pull_requests | 拉取请求 | 读取 | 列出仓库拉取请求 |
|
||||||
|
|||||||
@@ -151,6 +151,7 @@ Cursor 等客戶端可使用 stdio 命令:
|
|||||||
| package_read | 軟體套件 | 讀取 | 讀取軟體套件註冊表:列出套件、列出版本或取得某個版本 |
|
| package_read | 軟體套件 | 讀取 | 讀取軟體套件註冊表:列出套件、列出版本或取得某個版本 |
|
||||||
| package_write | 軟體套件 | 寫入 | 刪除軟體套件版本(不可復原) |
|
| package_write | 軟體套件 | 寫入 | 刪除軟體套件版本(不可復原) |
|
||||||
| list_issues | 問題 | 讀取 | 列出倉庫問題 |
|
| list_issues | 問題 | 讀取 | 列出倉庫問題 |
|
||||||
|
| attachment_read | 問題 | 讀取 | 讀取問題/評論附件:列出中繼資料、取得中繼資料或下載內容 |
|
||||||
| issue_read | 問題 | 讀取 | 讀取問題:詳情、評論或標籤 |
|
| issue_read | 問題 | 讀取 | 讀取問題:詳情、評論或標籤 |
|
||||||
| issue_write | 問題 | 寫入 | 寫入問題:創建、更新、管理評論和標籤 |
|
| issue_write | 問題 | 寫入 | 寫入問題:創建、更新、管理評論和標籤 |
|
||||||
| list_pull_requests | 拉取請求 | 讀取 | 列出倉庫拉取請求 |
|
| list_pull_requests | 拉取請求 | 讀取 | 列出倉庫拉取請求 |
|
||||||
|
|||||||
+77
-47
@@ -4,7 +4,9 @@ import (
|
|||||||
"context"
|
"context"
|
||||||
"flag"
|
"flag"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"io"
|
||||||
"os"
|
"os"
|
||||||
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
"text/tabwriter"
|
"text/tabwriter"
|
||||||
|
|
||||||
@@ -14,39 +16,56 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
host string
|
host string
|
||||||
port int
|
port int
|
||||||
token string
|
token string
|
||||||
tools string
|
tools string
|
||||||
version bool
|
version bool
|
||||||
|
maxInlineAttachmentBytes int
|
||||||
|
maxInlineAttachmentBytesFlagSet bool
|
||||||
|
osExit = os.Exit
|
||||||
)
|
)
|
||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
flag.StringVar(&flagPkg.Mode, "t", "stdio", "")
|
initFlagSet(flag.CommandLine, os.Args[1:], os.Getenv, os.ReadFile, os.Stderr)
|
||||||
flag.StringVar(&flagPkg.Mode, "transport", "stdio", "")
|
}
|
||||||
flag.StringVar(&host, "H", os.Getenv("GITEA_HOST"), "")
|
|
||||||
flag.StringVar(&host, "host", os.Getenv("GITEA_HOST"), "")
|
|
||||||
flag.IntVar(&port, "p", 8080, "")
|
|
||||||
flag.IntVar(&port, "port", 8080, "")
|
|
||||||
flag.StringVar(&token, "T", "", "")
|
|
||||||
flag.StringVar(&token, "token", "", "")
|
|
||||||
flag.BoolVar(&flagPkg.ReadOnly, "r", false, "")
|
|
||||||
flag.BoolVar(&flagPkg.ReadOnly, "read-only", false, "")
|
|
||||||
defaultTools := os.Getenv("GITEA_TOOLS")
|
|
||||||
flag.StringVar(&tools, "O", defaultTools, "")
|
|
||||||
flag.StringVar(&tools, "tools", defaultTools, "")
|
|
||||||
flag.BoolVar(&flagPkg.Debug, "d", false, "")
|
|
||||||
flag.BoolVar(&flagPkg.Debug, "debug", false, "")
|
|
||||||
flag.BoolVar(&flagPkg.Insecure, "k", false, "")
|
|
||||||
flag.BoolVar(&flagPkg.Insecure, "insecure", false, "")
|
|
||||||
flag.BoolVar(&version, "v", false, "")
|
|
||||||
flag.BoolVar(&version, "version", false, "")
|
|
||||||
|
|
||||||
flag.Usage = func() {
|
func initFlagSet(fs *flag.FlagSet, args []string, getenv func(string) string, readFile func(string) ([]byte, error), stderr io.Writer) {
|
||||||
w := tabwriter.NewWriter(os.Stderr, 0, 0, 3, ' ', 0)
|
fs.StringVar(&flagPkg.Mode, "t", "stdio", "")
|
||||||
fmt.Fprintln(os.Stderr, "Usage: gitea-mcp [options]")
|
fs.StringVar(&flagPkg.Mode, "transport", "stdio", "")
|
||||||
fmt.Fprintln(os.Stderr)
|
fs.StringVar(&host, "H", getenv("GITEA_HOST"), "")
|
||||||
fmt.Fprintln(os.Stderr, "Options:")
|
fs.StringVar(&host, "host", getenv("GITEA_HOST"), "")
|
||||||
|
fs.IntVar(&port, "p", 8080, "")
|
||||||
|
fs.IntVar(&port, "port", 8080, "")
|
||||||
|
fs.StringVar(&token, "T", "", "")
|
||||||
|
fs.StringVar(&token, "token", "", "")
|
||||||
|
fs.BoolVar(&flagPkg.ReadOnly, "r", false, "")
|
||||||
|
fs.BoolVar(&flagPkg.ReadOnly, "read-only", false, "")
|
||||||
|
defaultTools := getenv("GITEA_TOOLS")
|
||||||
|
fs.StringVar(&tools, "O", defaultTools, "")
|
||||||
|
fs.StringVar(&tools, "tools", defaultTools, "")
|
||||||
|
fs.BoolVar(&flagPkg.Debug, "d", false, "")
|
||||||
|
fs.BoolVar(&flagPkg.Debug, "debug", false, "")
|
||||||
|
fs.BoolVar(&flagPkg.Insecure, "k", false, "")
|
||||||
|
fs.BoolVar(&flagPkg.Insecure, "insecure", false, "")
|
||||||
|
fs.BoolVar(&version, "v", false, "")
|
||||||
|
fs.BoolVar(&version, "version", false, "")
|
||||||
|
maxInlineAttachmentBytes = 5 * 1024 * 1024
|
||||||
|
fs.Func("max-inline-attachment-bytes", "", func(val string) error {
|
||||||
|
parsed, err := strconv.Atoi(val)
|
||||||
|
if err != nil || parsed < 0 {
|
||||||
|
return fmt.Errorf("invalid value %q", val)
|
||||||
|
}
|
||||||
|
maxInlineAttachmentBytes = parsed
|
||||||
|
maxInlineAttachmentBytesFlagSet = true
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
|
||||||
|
fs.Usage = func() {
|
||||||
|
w := tabwriter.NewWriter(stderr, 0, 0, 3, ' ', 0)
|
||||||
|
fmt.Fprintln(stderr, "Usage: gitea-mcp [options]")
|
||||||
|
fmt.Fprintln(stderr)
|
||||||
|
fmt.Fprintln(stderr, "Options:")
|
||||||
fmt.Fprintf(w, " -t, -transport <type>\tTransport type: stdio or http (default: stdio)\n")
|
fmt.Fprintf(w, " -t, -transport <type>\tTransport type: stdio or http (default: stdio)\n")
|
||||||
fmt.Fprintf(w, " -H, -host <url>\tGitea host URL (default: https://gitea.com)\n")
|
fmt.Fprintf(w, " -H, -host <url>\tGitea host URL (default: https://gitea.com)\n")
|
||||||
fmt.Fprintf(w, " -p, -port <number>\tHTTP server port (default: 8080)\n")
|
fmt.Fprintf(w, " -p, -port <number>\tHTTP server port (default: 8080)\n")
|
||||||
@@ -55,6 +74,7 @@ func init() {
|
|||||||
fmt.Fprintf(w, " -O, -tools <names>\tComma-separated list of tool names to expose\n")
|
fmt.Fprintf(w, " -O, -tools <names>\tComma-separated list of tool names to expose\n")
|
||||||
fmt.Fprintf(w, " -d, -debug\tEnable debug mode\n")
|
fmt.Fprintf(w, " -d, -debug\tEnable debug mode\n")
|
||||||
fmt.Fprintf(w, " -k, -insecure\tIgnore TLS certificate errors\n")
|
fmt.Fprintf(w, " -k, -insecure\tIgnore TLS certificate errors\n")
|
||||||
|
fmt.Fprintf(w, " -max-inline-attachment-bytes <bytes>\tInline image attachments up to this size (default: 5242880)\n")
|
||||||
fmt.Fprintf(w, " -v, -version\tPrint version and exit\n")
|
fmt.Fprintf(w, " -v, -version\tPrint version and exit\n")
|
||||||
fmt.Fprintln(w)
|
fmt.Fprintln(w)
|
||||||
fmt.Fprintln(w, "Environment variables:")
|
fmt.Fprintln(w, "Environment variables:")
|
||||||
@@ -63,13 +83,14 @@ func init() {
|
|||||||
fmt.Fprintf(w, " GITEA_DEBUG\tSet to 'true' for debug mode\n")
|
fmt.Fprintf(w, " GITEA_DEBUG\tSet to 'true' for debug mode\n")
|
||||||
fmt.Fprintf(w, " GITEA_HOST\tOverride Gitea host URL\n")
|
fmt.Fprintf(w, " GITEA_HOST\tOverride Gitea host URL\n")
|
||||||
fmt.Fprintf(w, " GITEA_INSECURE\tSet to 'true' to ignore TLS errors\n")
|
fmt.Fprintf(w, " GITEA_INSECURE\tSet to 'true' to ignore TLS errors\n")
|
||||||
|
fmt.Fprintf(w, " GITEA_MAX_INLINE_ATTACHMENT_BYTES\tOverride inline image attachment size limit in bytes\n")
|
||||||
fmt.Fprintf(w, " GITEA_READONLY\tSet to 'true' for read-only mode\n")
|
fmt.Fprintf(w, " GITEA_READONLY\tSet to 'true' for read-only mode\n")
|
||||||
fmt.Fprintf(w, " GITEA_TOOLS\tComma-separated list of tool names to expose\n")
|
fmt.Fprintf(w, " GITEA_TOOLS\tComma-separated list of tool names to expose\n")
|
||||||
fmt.Fprintf(w, " MCP_MODE\tOverride transport mode\n")
|
fmt.Fprintf(w, " MCP_MODE\tOverride transport mode\n")
|
||||||
w.Flush()
|
_ = w.Flush()
|
||||||
}
|
}
|
||||||
|
|
||||||
flag.Parse()
|
_ = fs.Parse(args)
|
||||||
|
|
||||||
flagPkg.Host = host
|
flagPkg.Host = host
|
||||||
if flagPkg.Host == "" {
|
if flagPkg.Host == "" {
|
||||||
@@ -77,27 +98,27 @@ func init() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
flagPkg.Port = port
|
flagPkg.Port = port
|
||||||
|
flagPkg.MaxInlineAttachmentBytes = maxInlineAttachmentBytes
|
||||||
|
|
||||||
flagPkg.Token = token
|
flagPkg.Token = token
|
||||||
if flagPkg.Token == "" {
|
if flagPkg.Token == "" {
|
||||||
flagPkg.Token = os.Getenv("GITEA_ACCESS_TOKEN")
|
flagPkg.Token = getenv("GITEA_ACCESS_TOKEN")
|
||||||
}
|
}
|
||||||
if flagPkg.Token == "" {
|
if flagPkg.Token == "" {
|
||||||
if tokenFile := os.Getenv("GITEA_ACCESS_TOKEN_FILE"); tokenFile != "" {
|
if tokenFile := getenv("GITEA_ACCESS_TOKEN_FILE"); tokenFile != "" {
|
||||||
data, err := os.ReadFile(tokenFile)
|
data, err := readFile(tokenFile)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
fmt.Fprintf(os.Stderr, "error reading GITEA_ACCESS_TOKEN_FILE: %v\n", err)
|
fmt.Fprintf(stderr, "error reading GITEA_ACCESS_TOKEN_FILE: %v\n", err)
|
||||||
os.Exit(1)
|
osExit(1)
|
||||||
}
|
}
|
||||||
flagPkg.Token = strings.TrimRight(string(data), "\r\n")
|
flagPkg.Token = strings.TrimRight(string(data), "\r\n")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if os.Getenv("MCP_MODE") != "" {
|
if getenv("MCP_MODE") != "" {
|
||||||
flagPkg.Mode = os.Getenv("MCP_MODE")
|
flagPkg.Mode = getenv("MCP_MODE")
|
||||||
}
|
}
|
||||||
|
if getenv("GITEA_READONLY") == "true" {
|
||||||
if os.Getenv("GITEA_READONLY") == "true" {
|
|
||||||
flagPkg.ReadOnly = true
|
flagPkg.ReadOnly = true
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -110,15 +131,22 @@ func init() {
|
|||||||
if len(allowed) > 0 {
|
if len(allowed) > 0 {
|
||||||
flagPkg.AllowedTools = allowed
|
flagPkg.AllowedTools = allowed
|
||||||
}
|
}
|
||||||
|
if getenv("GITEA_DEBUG") == "true" {
|
||||||
if os.Getenv("GITEA_DEBUG") == "true" {
|
|
||||||
flagPkg.Debug = true
|
flagPkg.Debug = true
|
||||||
}
|
}
|
||||||
|
if getenv("GITEA_INSECURE") == "true" {
|
||||||
// Set insecure mode based on environment variable
|
|
||||||
if os.Getenv("GITEA_INSECURE") == "true" {
|
|
||||||
flagPkg.Insecure = true
|
flagPkg.Insecure = true
|
||||||
}
|
}
|
||||||
|
if !maxInlineAttachmentBytesFlagSet {
|
||||||
|
if val := getenv("GITEA_MAX_INLINE_ATTACHMENT_BYTES"); val != "" {
|
||||||
|
parsed, err := strconv.Atoi(val)
|
||||||
|
if err != nil || parsed < 0 {
|
||||||
|
fmt.Fprintf(stderr, "invalid GITEA_MAX_INLINE_ATTACHMENT_BYTES: %q\n", val)
|
||||||
|
osExit(1)
|
||||||
|
}
|
||||||
|
flagPkg.MaxInlineAttachmentBytes = parsed
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func Execute() {
|
func Execute() {
|
||||||
@@ -126,12 +154,14 @@ func Execute() {
|
|||||||
fmt.Fprintln(os.Stdout, flagPkg.Version)
|
fmt.Fprintln(os.Stdout, flagPkg.Version)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
defer log.Default().Sync() //nolint:errcheck // best-effort flush
|
|
||||||
if err := operation.Run(); err != nil {
|
if err := operation.Run(); err != nil {
|
||||||
if err == context.Canceled {
|
if err == context.Canceled {
|
||||||
log.Info("Server shutdown due to context cancellation")
|
log.Info("Server shutdown due to context cancellation")
|
||||||
|
_ = log.Default().Sync() // best-effort flush
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
log.Fatalf("Run Gitea MCP Server Error: %v", err) //nolint:gocritic // intentional exit after defer
|
_ = log.Default().Sync() // best-effort flush
|
||||||
|
log.Fatalf("Run Gitea MCP Server Error: %v", err)
|
||||||
}
|
}
|
||||||
|
_ = log.Default().Sync() // best-effort flush
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -0,0 +1,311 @@
|
|||||||
|
package issue
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"context"
|
||||||
|
"encoding/base64"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"mime"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"gitea.com/gitea/gitea-mcp/pkg/annotation"
|
||||||
|
"gitea.com/gitea/gitea-mcp/pkg/flag"
|
||||||
|
"gitea.com/gitea/gitea-mcp/pkg/gitea"
|
||||||
|
"gitea.com/gitea/gitea-mcp/pkg/params"
|
||||||
|
"gitea.com/gitea/gitea-mcp/pkg/to"
|
||||||
|
|
||||||
|
gitea_sdk "gitea.dev/sdk"
|
||||||
|
"github.com/mark3labs/mcp-go/mcp"
|
||||||
|
"github.com/mark3labs/mcp-go/server"
|
||||||
|
)
|
||||||
|
|
||||||
|
const AttachmentReadToolName = "attachment_read"
|
||||||
|
|
||||||
|
var AttachmentReadTool = mcp.NewTool(
|
||||||
|
AttachmentReadToolName,
|
||||||
|
mcp.WithDescription("Read issue/comment attachments: list metadata, get metadata, or download content."),
|
||||||
|
mcp.WithToolAnnotation(annotation.ReadOnly("Read issue or comment attachments")),
|
||||||
|
mcp.WithString("method", mcp.Required(), mcp.Enum("list", "get", "download")),
|
||||||
|
mcp.WithString("owner", mcp.Required(), mcp.Description(params.OwnerDesc)),
|
||||||
|
mcp.WithString("repo", mcp.Required(), mcp.Description(params.RepoDesc)),
|
||||||
|
mcp.WithNumber("issue_number", mcp.Description("required for issue attachment list/get or issue-scoped metadata lookup")),
|
||||||
|
mcp.WithNumber("comment_id", mcp.Description("required for comment attachment list/get or comment-scoped metadata lookup")),
|
||||||
|
mcp.WithNumber("attachment_id", mcp.Description("required for get and for download when attachment_uuid is not provided")),
|
||||||
|
mcp.WithString("attachment_uuid", mcp.Description("attachment UUID for direct download path lookup")),
|
||||||
|
mcp.WithString("output_path", mcp.Description("write the attachment to this exact path")),
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
Tool.RegisterRead(server.ServerTool{Tool: AttachmentReadTool, Handler: attachmentReadFn})
|
||||||
|
}
|
||||||
|
|
||||||
|
func attachmentReadFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||||
|
method, err := params.GetString(req.GetArguments(), "method")
|
||||||
|
if err != nil {
|
||||||
|
return to.ErrorResult(err)
|
||||||
|
}
|
||||||
|
switch method {
|
||||||
|
case "list":
|
||||||
|
return listAttachmentsFn(ctx, req)
|
||||||
|
case "get":
|
||||||
|
return getAttachmentFn(ctx, req)
|
||||||
|
case "download":
|
||||||
|
return downloadAttachmentFn(ctx, req)
|
||||||
|
default:
|
||||||
|
return to.ErrorResult(fmt.Errorf("unknown method: %s", method))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func listAttachmentsFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||||
|
owner, repo, issueNumber, commentID, err := attachmentScopeArgs(req)
|
||||||
|
if err != nil {
|
||||||
|
return to.ErrorResult(err)
|
||||||
|
}
|
||||||
|
client, err := gitea.ClientFromContext(ctx)
|
||||||
|
if err != nil {
|
||||||
|
return to.ErrorResult(fmt.Errorf("get gitea client err: %v", err))
|
||||||
|
}
|
||||||
|
var attachments []*gitea_sdk.Attachment
|
||||||
|
if issueNumber > 0 {
|
||||||
|
attachments, _, err = client.Issues.ListIssueAttachments(ctx, owner, repo, issueNumber)
|
||||||
|
} else {
|
||||||
|
attachments, _, err = client.Issues.ListIssueCommentAttachments(ctx, owner, repo, commentID)
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
return to.ErrorResult(fmt.Errorf("list attachments err: %v", err))
|
||||||
|
}
|
||||||
|
return to.TextResult(slimAttachments(attachments))
|
||||||
|
}
|
||||||
|
|
||||||
|
func getAttachmentFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||||
|
att, err := lookupAttachment(ctx, req)
|
||||||
|
if err != nil {
|
||||||
|
return to.ErrorResult(err)
|
||||||
|
}
|
||||||
|
return to.TextResult(slimAttachment(att))
|
||||||
|
}
|
||||||
|
|
||||||
|
func downloadAttachmentFn(ctx context.Context, req mcp.CallToolRequest) (*mcp.CallToolResult, error) {
|
||||||
|
owner, err := params.GetString(req.GetArguments(), "owner")
|
||||||
|
if err != nil {
|
||||||
|
return to.ErrorResult(err)
|
||||||
|
}
|
||||||
|
repo, err := params.GetString(req.GetArguments(), "repo")
|
||||||
|
if err != nil {
|
||||||
|
return to.ErrorResult(err)
|
||||||
|
}
|
||||||
|
explicitOutputPath := params.GetOptionalString(req.GetArguments(), "output_path", "")
|
||||||
|
attachmentUUID := strings.TrimSpace(params.GetOptionalString(req.GetArguments(), "attachment_uuid", ""))
|
||||||
|
|
||||||
|
var att *gitea_sdk.Attachment
|
||||||
|
if attachmentUUID == "" {
|
||||||
|
att, err = lookupAttachment(ctx, req)
|
||||||
|
if err != nil {
|
||||||
|
return to.ErrorResult(err)
|
||||||
|
}
|
||||||
|
attachmentUUID = strings.TrimSpace(att.UUID)
|
||||||
|
}
|
||||||
|
if attachmentUUID == "" {
|
||||||
|
return to.ErrorResult(errors.New("attachment_uuid or attachment metadata with uuid is required"))
|
||||||
|
}
|
||||||
|
|
||||||
|
name := attachmentUUID
|
||||||
|
if att != nil && strings.TrimSpace(att.Name) != "" {
|
||||||
|
name = strings.TrimSpace(att.Name)
|
||||||
|
}
|
||||||
|
|
||||||
|
resp, err := gitea.OpenAttachment(ctx, "/attachments/"+attachmentUUID, "*/*")
|
||||||
|
if err != nil {
|
||||||
|
return to.ErrorResult(fmt.Errorf("download attachment err: %v", err))
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
mimeType := normalizeAttachmentContentType(resp.ContentType, name)
|
||||||
|
if explicitOutputPath == "" && shouldInlineAttachment(att, mimeType) {
|
||||||
|
limited, readErr := io.ReadAll(io.LimitReader(resp.Body, int64(flag.MaxInlineAttachmentBytes)+1))
|
||||||
|
if readErr != nil {
|
||||||
|
return to.ErrorResult(fmt.Errorf("read attachment err: %v", readErr))
|
||||||
|
}
|
||||||
|
if len(limited) <= flag.MaxInlineAttachmentBytes {
|
||||||
|
text := fmt.Sprintf("attachment %s (%s, %d bytes, %s)", name, attachmentUUID, len(limited), mimeType)
|
||||||
|
return mcp.NewToolResultImage(text, base64.StdEncoding.EncodeToString(limited), mimeType), nil
|
||||||
|
}
|
||||||
|
outputPath := defaultAttachmentPath(owner, repo, name, attachmentUUID)
|
||||||
|
if err := os.MkdirAll(filepath.Dir(outputPath), 0o700); err != nil {
|
||||||
|
return to.ErrorResult(fmt.Errorf("create output dir err: %v", err))
|
||||||
|
}
|
||||||
|
reader := io.MultiReader(bytes.NewReader(limited), resp.Body)
|
||||||
|
written, err := gitea.WriteAttachment(reader, outputPath)
|
||||||
|
if err != nil {
|
||||||
|
return to.ErrorResult(fmt.Errorf("write attachment file err: %v", err))
|
||||||
|
}
|
||||||
|
return attachmentFileResult(att, outputPath, written, name, attachmentUUID, mimeType)
|
||||||
|
}
|
||||||
|
|
||||||
|
outputPath := explicitOutputPath
|
||||||
|
if outputPath == "" {
|
||||||
|
outputPath = defaultAttachmentPath(owner, repo, name, attachmentUUID)
|
||||||
|
}
|
||||||
|
if err := os.MkdirAll(filepath.Dir(outputPath), 0o700); err != nil {
|
||||||
|
return to.ErrorResult(fmt.Errorf("create output dir err: %v", err))
|
||||||
|
}
|
||||||
|
written, err := gitea.WriteAttachment(resp.Body, outputPath)
|
||||||
|
if err != nil {
|
||||||
|
return to.ErrorResult(fmt.Errorf("write attachment file err: %v", err))
|
||||||
|
}
|
||||||
|
return attachmentFileResult(att, outputPath, written, name, attachmentUUID, mimeType)
|
||||||
|
}
|
||||||
|
|
||||||
|
func shouldInlineAttachment(att *gitea_sdk.Attachment, mimeType string) bool {
|
||||||
|
if !strings.HasPrefix(mimeType, "image/") || flag.MaxInlineAttachmentBytes <= 0 {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
if att == nil || att.Size <= 0 {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
return att.Size <= int64(flag.MaxInlineAttachmentBytes)
|
||||||
|
}
|
||||||
|
|
||||||
|
func attachmentFileResult(att *gitea_sdk.Attachment, outputPath string, written int64, name, attachmentUUID, mimeType string) (*mcp.CallToolResult, error) {
|
||||||
|
res := map[string]any{
|
||||||
|
"path": outputPath,
|
||||||
|
"bytes": written,
|
||||||
|
"name": name,
|
||||||
|
"uuid": attachmentUUID,
|
||||||
|
"mime_type": mimeType,
|
||||||
|
"content_type": mimeType,
|
||||||
|
}
|
||||||
|
if att != nil {
|
||||||
|
res["attachment_id"] = att.ID
|
||||||
|
}
|
||||||
|
return to.TextResult(res)
|
||||||
|
}
|
||||||
|
|
||||||
|
func attachmentScopeArgs(req mcp.CallToolRequest) (owner, repo string, issueNumber, commentID int64, err error) {
|
||||||
|
owner, err = params.GetString(req.GetArguments(), "owner")
|
||||||
|
if err != nil {
|
||||||
|
return "", "", 0, 0, err
|
||||||
|
}
|
||||||
|
repo, err = params.GetString(req.GetArguments(), "repo")
|
||||||
|
if err != nil {
|
||||||
|
return "", "", 0, 0, err
|
||||||
|
}
|
||||||
|
issueNumber = params.GetOptionalInt(req.GetArguments(), "issue_number", 0)
|
||||||
|
commentID = params.GetOptionalInt(req.GetArguments(), "comment_id", 0)
|
||||||
|
if (issueNumber > 0) == (commentID > 0) {
|
||||||
|
return "", "", 0, 0, errors.New("exactly one of issue_number or comment_id is required")
|
||||||
|
}
|
||||||
|
return owner, repo, issueNumber, commentID, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func lookupAttachment(ctx context.Context, req mcp.CallToolRequest) (*gitea_sdk.Attachment, error) {
|
||||||
|
owner, repo, issueNumber, commentID, err := attachmentScopeArgs(req)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
attachmentID := params.GetOptionalInt(req.GetArguments(), "attachment_id", 0)
|
||||||
|
if attachmentID <= 0 {
|
||||||
|
return nil, errors.New("attachment_id is required")
|
||||||
|
}
|
||||||
|
client, err := gitea.ClientFromContext(ctx)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("get gitea client err: %v", err)
|
||||||
|
}
|
||||||
|
if issueNumber > 0 {
|
||||||
|
att, _, err := client.Issues.GetIssueAttachment(ctx, owner, repo, issueNumber, attachmentID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("get issue attachment err: %v", err)
|
||||||
|
}
|
||||||
|
return att, nil
|
||||||
|
}
|
||||||
|
att, _, err := client.Issues.GetIssueCommentAttachment(ctx, owner, repo, commentID, attachmentID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("get issue comment attachment err: %v", err)
|
||||||
|
}
|
||||||
|
return att, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func slimAttachments(atts []*gitea_sdk.Attachment) []map[string]any {
|
||||||
|
out := make([]map[string]any, 0, len(atts))
|
||||||
|
for _, att := range atts {
|
||||||
|
out = append(out, slimAttachment(att))
|
||||||
|
}
|
||||||
|
return out
|
||||||
|
}
|
||||||
|
|
||||||
|
func slimAttachment(att *gitea_sdk.Attachment) map[string]any {
|
||||||
|
if att == nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
m := map[string]any{
|
||||||
|
"id": att.ID,
|
||||||
|
"name": att.Name,
|
||||||
|
"uuid": att.UUID,
|
||||||
|
"size": att.Size,
|
||||||
|
"download_count": att.DownloadCount,
|
||||||
|
"created_at": att.Created,
|
||||||
|
"mime_type": inferAttachmentMimeType(att.Name),
|
||||||
|
}
|
||||||
|
if att.DownloadURL != "" {
|
||||||
|
m["browser_download_url"] = att.DownloadURL
|
||||||
|
}
|
||||||
|
return m
|
||||||
|
}
|
||||||
|
|
||||||
|
func inferAttachmentMimeType(name string) string {
|
||||||
|
if ext := strings.ToLower(filepath.Ext(strings.TrimSpace(name))); ext != "" {
|
||||||
|
if mimeType := mime.TypeByExtension(ext); mimeType != "" {
|
||||||
|
return strings.Split(mimeType, ";")[0]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return "application/octet-stream"
|
||||||
|
}
|
||||||
|
|
||||||
|
func normalizeAttachmentContentType(contentType, name string) string {
|
||||||
|
mediaType, _, err := mime.ParseMediaType(contentType)
|
||||||
|
if err == nil && mediaType != "" && mediaType != "application/octet-stream" {
|
||||||
|
return mediaType
|
||||||
|
}
|
||||||
|
return inferAttachmentMimeType(name)
|
||||||
|
}
|
||||||
|
|
||||||
|
func defaultAttachmentPath(owner, repo, name, uuid string) string {
|
||||||
|
home, _ := os.UserHomeDir()
|
||||||
|
if home == "" {
|
||||||
|
home = os.TempDir()
|
||||||
|
}
|
||||||
|
filename := attachmentFilename(name, uuid)
|
||||||
|
ext := filepath.Ext(filename)
|
||||||
|
base := strings.TrimSuffix(filename, ext)
|
||||||
|
if uuid != "" {
|
||||||
|
filename = uuid
|
||||||
|
if base != "" && base != "attachment" {
|
||||||
|
filename = base + "-" + uuid
|
||||||
|
}
|
||||||
|
filename += ext
|
||||||
|
}
|
||||||
|
return filepath.Join(home, ".gitea-mcp", "attachments", safePathPart(owner), safePathPart(repo), filename)
|
||||||
|
}
|
||||||
|
|
||||||
|
func attachmentFilename(name, uuid string) string {
|
||||||
|
name = strings.TrimSpace(name)
|
||||||
|
if name != "" && !strings.ContainsAny(name, `/\\`) && name != "." && name != ".." {
|
||||||
|
return name
|
||||||
|
}
|
||||||
|
if uuid != "" {
|
||||||
|
return uuid + ".bin"
|
||||||
|
}
|
||||||
|
return "attachment.bin"
|
||||||
|
}
|
||||||
|
|
||||||
|
func safePathPart(name string) string {
|
||||||
|
name = strings.TrimSpace(name)
|
||||||
|
if name != "" && !strings.ContainsAny(name, `/\\`) && name != "." && name != ".." {
|
||||||
|
return name
|
||||||
|
}
|
||||||
|
return "unknown"
|
||||||
|
}
|
||||||
@@ -0,0 +1,182 @@
|
|||||||
|
package issue
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
"net/http/httptest"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"gitea.com/gitea/gitea-mcp/pkg/flag"
|
||||||
|
"gitea.com/gitea/gitea-mcp/pkg/gitea"
|
||||||
|
|
||||||
|
"github.com/mark3labs/mcp-go/mcp"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestAttachmentFilename(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
file string
|
||||||
|
uuid string
|
||||||
|
want string
|
||||||
|
}{
|
||||||
|
{"uses attachment name", "screenshot.png", "abc", "screenshot.png"},
|
||||||
|
{"falls back for traversal", "../etc/passwd", "abc", "abc.bin"},
|
||||||
|
{"falls back for empty name", "", "abc", "abc.bin"},
|
||||||
|
{"uses generic fallback", "", "", "attachment.bin"},
|
||||||
|
}
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
if got := attachmentFilename(tt.file, tt.uuid); got != tt.want {
|
||||||
|
t.Fatalf("attachmentFilename() = %q, want %q", got, tt.want)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestInferAttachmentMimeType(t *testing.T) {
|
||||||
|
if got := inferAttachmentMimeType("shot.png"); got != "image/png" {
|
||||||
|
t.Fatalf("inferAttachmentMimeType() = %q, want image/png", got)
|
||||||
|
}
|
||||||
|
if got := inferAttachmentMimeType("archive.unknownext"); got != "application/octet-stream" {
|
||||||
|
t.Fatalf("inferAttachmentMimeType() = %q, want application/octet-stream", got)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestDefaultAttachmentPath(t *testing.T) {
|
||||||
|
got := defaultAttachmentPath("octo", "demo", "shot.png", "uuid-1")
|
||||||
|
want := filepath.Join(".gitea-mcp", "attachments", "octo", "demo", "shot-uuid-1.png")
|
||||||
|
if !strings.HasSuffix(got, want) {
|
||||||
|
t.Fatalf("defaultAttachmentPath() = %q, want suffix %q", got, want)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAttachmentReadListIssueAttachments(t *testing.T) {
|
||||||
|
const owner = "octo"
|
||||||
|
const repo = "demo"
|
||||||
|
handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
switch r.URL.Path {
|
||||||
|
case "/api/v1/version":
|
||||||
|
w.Header().Set("Content-Type", "application/json")
|
||||||
|
_, _ = w.Write([]byte(`{"version":"1.12.0"}`))
|
||||||
|
case fmt.Sprintf("/api/v1/repos/%s/%s", owner, repo):
|
||||||
|
w.Header().Set("Content-Type", "application/json")
|
||||||
|
_, _ = w.Write([]byte(`{"private":false}`))
|
||||||
|
case fmt.Sprintf("/api/v1/repos/%s/%s/issues/42/assets", owner, repo):
|
||||||
|
w.Header().Set("Content-Type", "application/json")
|
||||||
|
_, _ = w.Write([]byte(`[{"id":1,"name":"shot.png","uuid":"uuid-1","size":10,"download_count":2}]`))
|
||||||
|
default:
|
||||||
|
http.NotFound(w, r)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
server := httptest.NewServer(handler)
|
||||||
|
defer server.Close()
|
||||||
|
origHost, origToken, origVersion := flag.Host, flag.Token, flag.Version
|
||||||
|
flag.Host, flag.Token, flag.Version = server.URL, "", "test"
|
||||||
|
defer func() { flag.Host, flag.Token, flag.Version = origHost, origToken, origVersion }()
|
||||||
|
|
||||||
|
res, err := attachmentReadFn(context.Background(), mcp.CallToolRequest{Params: mcp.CallToolParams{Arguments: map[string]any{
|
||||||
|
"method": "list", "owner": owner, "repo": repo, "issue_number": float64(42),
|
||||||
|
}}})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("attachmentReadFn() error = %v", err)
|
||||||
|
}
|
||||||
|
body := res.Content[0].(mcp.TextContent).Text
|
||||||
|
if !strings.Contains(body, `"mime_type":"image/png"`) || !strings.Contains(body, `"uuid":"uuid-1"`) {
|
||||||
|
t.Fatalf("unexpected body: %s", body)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestDownloadAttachmentRejectsRedirectLoopAtHopLimit(t *testing.T) {
|
||||||
|
var serverURL string
|
||||||
|
redirects := 0
|
||||||
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
if r.URL.Path != "/attachments/uuid-1" {
|
||||||
|
t.Fatalf("path = %s", r.URL.Path)
|
||||||
|
}
|
||||||
|
redirects++
|
||||||
|
http.Redirect(w, r, serverURL+"/attachments/uuid-1", http.StatusFound)
|
||||||
|
}))
|
||||||
|
defer server.Close()
|
||||||
|
serverURL = server.URL
|
||||||
|
|
||||||
|
origHost := flag.Host
|
||||||
|
flag.Host = server.URL
|
||||||
|
defer func() { flag.Host = origHost }()
|
||||||
|
|
||||||
|
_, _, _, err := gitea.DownloadAttachment(context.Background(), "/attachments/uuid-1", "*/*")
|
||||||
|
if err == nil {
|
||||||
|
t.Fatal("expected redirect limit error")
|
||||||
|
}
|
||||||
|
if !strings.Contains(err.Error(), "stopped after 10 redirects") {
|
||||||
|
t.Fatalf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
if redirects != 10 {
|
||||||
|
t.Fatalf("redirects = %d, want 10", redirects)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAttachmentReadDownloadSavesLargeAttachmentToDefaultFile(t *testing.T) {
|
||||||
|
const owner = "octo"
|
||||||
|
const repo = "demo"
|
||||||
|
const uuid = "uuid-1"
|
||||||
|
const name = "large.bin"
|
||||||
|
payload := strings.Repeat("a", 32)
|
||||||
|
|
||||||
|
home := t.TempDir()
|
||||||
|
t.Setenv("HOME", home)
|
||||||
|
|
||||||
|
handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
switch r.URL.Path {
|
||||||
|
case "/api/v1/version":
|
||||||
|
w.Header().Set("Content-Type", "application/json")
|
||||||
|
_, _ = w.Write([]byte(`{"version":"1.12.0"}`))
|
||||||
|
case fmt.Sprintf("/api/v1/repos/%s/%s", owner, repo):
|
||||||
|
w.Header().Set("Content-Type", "application/json")
|
||||||
|
_, _ = w.Write([]byte(`{"private":false}`))
|
||||||
|
case fmt.Sprintf("/api/v1/repos/%s/%s/issues/42/assets/1", owner, repo):
|
||||||
|
w.Header().Set("Content-Type", "application/json")
|
||||||
|
_ = json.NewEncoder(w).Encode(map[string]any{"id": 1, "name": name, "uuid": uuid, "size": len(payload)})
|
||||||
|
case "/attachments/" + uuid:
|
||||||
|
w.Header().Set("Content-Type", "application/octet-stream")
|
||||||
|
_, _ = w.Write([]byte(payload))
|
||||||
|
default:
|
||||||
|
http.NotFound(w, r)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
server := httptest.NewServer(handler)
|
||||||
|
defer server.Close()
|
||||||
|
|
||||||
|
origHost, origToken, origVersion, origInline := flag.Host, flag.Token, flag.Version, flag.MaxInlineAttachmentBytes
|
||||||
|
flag.Host, flag.Token, flag.Version = server.URL, "", "test"
|
||||||
|
flag.MaxInlineAttachmentBytes = 8
|
||||||
|
defer func() {
|
||||||
|
flag.Host, flag.Token, flag.Version, flag.MaxInlineAttachmentBytes = origHost, origToken, origVersion, origInline
|
||||||
|
}()
|
||||||
|
|
||||||
|
res, err := attachmentReadFn(context.Background(), mcp.CallToolRequest{Params: mcp.CallToolParams{Arguments: map[string]any{
|
||||||
|
"method": "download", "owner": owner, "repo": repo, "issue_number": float64(42), "attachment_id": float64(1),
|
||||||
|
}}})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("attachmentReadFn() error = %v", err)
|
||||||
|
}
|
||||||
|
body := res.Content[0].(mcp.TextContent).Text
|
||||||
|
wantPath := filepath.Join(home, ".gitea-mcp", "attachments", owner, repo, "large-uuid-1.bin")
|
||||||
|
if !strings.Contains(body, wantPath) {
|
||||||
|
t.Fatalf("result missing path %q: %s", wantPath, body)
|
||||||
|
}
|
||||||
|
data, err := os.ReadFile(wantPath)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("ReadFile(%q): %v", wantPath, err)
|
||||||
|
}
|
||||||
|
if string(data) != payload {
|
||||||
|
t.Fatalf("saved payload mismatch")
|
||||||
|
}
|
||||||
|
if !strings.Contains(body, `"bytes":32`) {
|
||||||
|
t.Fatalf("result missing bytes: %s", body)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -7,6 +7,8 @@ var (
|
|||||||
Version string
|
Version string
|
||||||
Mode string
|
Mode string
|
||||||
|
|
||||||
|
MaxInlineAttachmentBytes int
|
||||||
|
|
||||||
Insecure bool
|
Insecure bool
|
||||||
ReadOnly bool
|
ReadOnly bool
|
||||||
Debug bool
|
Debug bool
|
||||||
|
|||||||
@@ -0,0 +1,208 @@
|
|||||||
|
package gitea
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"net/http"
|
||||||
|
"net/http/httptest"
|
||||||
|
"net/url"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"gitea.com/gitea/gitea-mcp/pkg/flag"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestBuildAttachmentURLRejectsExternalURL(t *testing.T) {
|
||||||
|
origHost := flag.Host
|
||||||
|
flag.Host = "https://example.com"
|
||||||
|
defer func() { flag.Host = origHost }()
|
||||||
|
|
||||||
|
if _, err := buildAttachmentURL("https://evil.example.com/attachments/1"); err == nil {
|
||||||
|
t.Fatal("expected error for full URL attachment path")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestBuildAttachmentURLPreservesHostPathPrefix(t *testing.T) {
|
||||||
|
origHost := flag.Host
|
||||||
|
flag.Host = "https://example.com/gitea"
|
||||||
|
defer func() { flag.Host = origHost }()
|
||||||
|
|
||||||
|
got, err := buildAttachmentURL("/attachments/uuid-1")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("buildAttachmentURL() error = %v", err)
|
||||||
|
}
|
||||||
|
if got != "https://example.com/gitea/attachments/uuid-1" {
|
||||||
|
t.Fatalf("buildAttachmentURL() = %q", got)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestDownloadAttachmentUsesTokenAndReturnsContentType(t *testing.T) {
|
||||||
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
if r.URL.Path != "/attachments/uuid-1" {
|
||||||
|
t.Fatalf("path = %s", r.URL.Path)
|
||||||
|
}
|
||||||
|
if got := r.Header.Get("Authorization"); got != "token secret-token" {
|
||||||
|
t.Fatalf("Authorization = %q", got)
|
||||||
|
}
|
||||||
|
if got := r.Header.Get("Accept"); got != "*/*" {
|
||||||
|
t.Fatalf("Accept = %q", got)
|
||||||
|
}
|
||||||
|
w.Header().Set("Content-Type", "image/png")
|
||||||
|
_, _ = w.Write([]byte("png-bytes"))
|
||||||
|
}))
|
||||||
|
defer server.Close()
|
||||||
|
|
||||||
|
origHost, origToken := flag.Host, flag.Token
|
||||||
|
flag.Host, flag.Token = server.URL, "secret-token"
|
||||||
|
defer func() { flag.Host, flag.Token = origHost, origToken }()
|
||||||
|
|
||||||
|
raw, contentType, status, err := DownloadAttachment(context.Background(), "/attachments/uuid-1", "*/*")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("DownloadAttachment() error = %v", err)
|
||||||
|
}
|
||||||
|
if status != http.StatusOK {
|
||||||
|
t.Fatalf("DownloadAttachment() status = %d, want %d", status, http.StatusOK)
|
||||||
|
}
|
||||||
|
if contentType != "image/png" {
|
||||||
|
t.Fatalf("contentType = %q", contentType)
|
||||||
|
}
|
||||||
|
if string(raw) != "png-bytes" {
|
||||||
|
t.Fatalf("body = %q", string(raw))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestDownloadAttachmentRejectsCrossOriginRedirect(t *testing.T) {
|
||||||
|
redirected := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
t.Fatalf("cross-origin redirect should not be followed: %s", r.URL.String())
|
||||||
|
}))
|
||||||
|
defer redirected.Close()
|
||||||
|
|
||||||
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
http.Redirect(w, r, redirected.URL+"/attachments/uuid-1", http.StatusFound)
|
||||||
|
}))
|
||||||
|
defer server.Close()
|
||||||
|
|
||||||
|
origHost, origToken := flag.Host, flag.Token
|
||||||
|
flag.Host, flag.Token = server.URL, "secret-token"
|
||||||
|
defer func() { flag.Host, flag.Token = origHost, origToken }()
|
||||||
|
|
||||||
|
_, _, _, err := DownloadAttachment(context.Background(), "/attachments/uuid-1", "*/*")
|
||||||
|
if err == nil {
|
||||||
|
t.Fatal("expected error for cross-origin redirect")
|
||||||
|
}
|
||||||
|
want := "do request: Get \"" + redirected.URL + "/attachments/uuid-1\": attachment redirect changed origin"
|
||||||
|
if err.Error() != want {
|
||||||
|
t.Fatalf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestDownloadAttachmentRejectsSameOriginRedirectLoopAtHopLimit(t *testing.T) {
|
||||||
|
var serverURL string
|
||||||
|
redirects := 0
|
||||||
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
redirects++
|
||||||
|
http.Redirect(w, r, serverURL+"/attachments/uuid-1", http.StatusFound)
|
||||||
|
}))
|
||||||
|
defer server.Close()
|
||||||
|
serverURL = server.URL
|
||||||
|
|
||||||
|
origHost := flag.Host
|
||||||
|
flag.Host = server.URL
|
||||||
|
defer func() { flag.Host = origHost }()
|
||||||
|
|
||||||
|
_, _, _, err := DownloadAttachment(context.Background(), "/attachments/uuid-1", "*/*")
|
||||||
|
if err == nil {
|
||||||
|
t.Fatal("expected redirect limit error")
|
||||||
|
}
|
||||||
|
if !strings.Contains(err.Error(), "stopped after 10 redirects") {
|
||||||
|
t.Fatalf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
if redirects != 10 {
|
||||||
|
t.Fatalf("redirects = %d, want 10", redirects)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestWriteAttachmentStreamsBodyToFile(t *testing.T) {
|
||||||
|
path := filepath.Join(t.TempDir(), "attachment.bin")
|
||||||
|
written, err := WriteAttachment(strings.NewReader(strings.Repeat("z", 64)), path)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("WriteAttachment() error = %v", err)
|
||||||
|
}
|
||||||
|
if written != 64 {
|
||||||
|
t.Fatalf("written = %d, want 64", written)
|
||||||
|
}
|
||||||
|
data, err := os.ReadFile(path)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("ReadFile() error = %v", err)
|
||||||
|
}
|
||||||
|
if string(data) != strings.Repeat("z", 64) {
|
||||||
|
t.Fatalf("body mismatch")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestWriteAttachmentCreatesPrivateFile(t *testing.T) {
|
||||||
|
path := filepath.Join(t.TempDir(), "attachment.bin")
|
||||||
|
|
||||||
|
if _, err := WriteAttachment(strings.NewReader("secret"), path); err != nil {
|
||||||
|
t.Fatalf("WriteAttachment() error = %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
info, err := os.Stat(path)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Stat() error = %v", err)
|
||||||
|
}
|
||||||
|
if got := info.Mode().Perm() & 0o077; got != 0 {
|
||||||
|
t.Fatalf("permissions = %03o, want no group/other bits", info.Mode().Perm())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestWriteAttachmentTightensExistingFilePermissions(t *testing.T) {
|
||||||
|
path := filepath.Join(t.TempDir(), "attachment.bin")
|
||||||
|
if err := os.WriteFile(path, []byte("public"), 0o644); err != nil {
|
||||||
|
t.Fatalf("WriteFile() error = %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if _, err := WriteAttachment(strings.NewReader("secret"), path); err != nil {
|
||||||
|
t.Fatalf("WriteAttachment() error = %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
info, err := os.Stat(path)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Stat() error = %v", err)
|
||||||
|
}
|
||||||
|
if got := info.Mode().Perm(); got != 0o600 {
|
||||||
|
t.Fatalf("permissions = %03o, want 600", got)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestDownloadAttachmentErrorsOnNon2xx(t *testing.T) {
|
||||||
|
server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
http.Error(w, "nope", http.StatusForbidden)
|
||||||
|
}))
|
||||||
|
defer server.Close()
|
||||||
|
|
||||||
|
origHost := flag.Host
|
||||||
|
flag.Host = server.URL
|
||||||
|
defer func() { flag.Host = origHost }()
|
||||||
|
|
||||||
|
_, _, status, err := DownloadAttachment(context.Background(), "/attachments/uuid-1", "*/*")
|
||||||
|
if status != http.StatusForbidden {
|
||||||
|
t.Fatalf("status = %d, want %d", status, http.StatusForbidden)
|
||||||
|
}
|
||||||
|
var httpErr *HTTPError
|
||||||
|
if !errors.As(err, &httpErr) {
|
||||||
|
t.Fatalf("expected HTTPError, got %T", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAttachmentHTTPClientHasNoTotalTimeout(t *testing.T) {
|
||||||
|
origin := &url.URL{Scheme: "https", Host: "example.com"}
|
||||||
|
|
||||||
|
client := attachmentHTTPClient(origin)
|
||||||
|
|
||||||
|
if client.Timeout != 0 {
|
||||||
|
t.Fatalf("Timeout = %v, want 0", client.Timeout)
|
||||||
|
}
|
||||||
|
}
|
||||||
+151
-15
@@ -9,6 +9,8 @@ import (
|
|||||||
"io"
|
"io"
|
||||||
"net/http"
|
"net/http"
|
||||||
"net/url"
|
"net/url"
|
||||||
|
"os"
|
||||||
|
"path"
|
||||||
"strings"
|
"strings"
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
@@ -34,6 +36,12 @@ func (e *HTTPError) Error() string {
|
|||||||
return fmt.Sprintf("request failed with status %d: %s", e.StatusCode, e.Body)
|
return fmt.Sprintf("request failed with status %d: %s", e.StatusCode, e.Body)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type AttachmentResponse struct {
|
||||||
|
Body io.ReadCloser
|
||||||
|
ContentType string
|
||||||
|
StatusCode int
|
||||||
|
}
|
||||||
|
|
||||||
func tokenFromContext(ctx context.Context) string {
|
func tokenFromContext(ctx context.Context) string {
|
||||||
if ctx != nil {
|
if ctx != nil {
|
||||||
if token, ok := ctx.Value(mcpContext.TokenContextKey).(string); ok && token != "" {
|
if token, ok := ctx.Value(mcpContext.TokenContextKey).(string); ok && token != "" {
|
||||||
@@ -75,6 +83,40 @@ func buildAPIURL(path string, query url.Values) (string, error) {
|
|||||||
return u.String(), nil
|
return u.String(), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func buildAttachmentURL(attachmentPath string) (string, error) {
|
||||||
|
host := strings.TrimRight(flag.Host, "/")
|
||||||
|
if host == "" {
|
||||||
|
return "", errors.New("gitea host is empty")
|
||||||
|
}
|
||||||
|
baseURL, err := url.Parse(host)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
if attachmentPath == "" {
|
||||||
|
return "", errors.New("attachment path is empty")
|
||||||
|
}
|
||||||
|
if strings.Contains(attachmentPath, "://") {
|
||||||
|
return "", errors.New("attachment path must not be a URL")
|
||||||
|
}
|
||||||
|
if !strings.HasPrefix(attachmentPath, "/attachments/") {
|
||||||
|
return "", errors.New("attachment path must start with /attachments/")
|
||||||
|
}
|
||||||
|
cleanPath := path.Clean(attachmentPath)
|
||||||
|
if !strings.HasPrefix(cleanPath, "/attachments/") {
|
||||||
|
return "", errors.New("attachment path must stay within /attachments/")
|
||||||
|
}
|
||||||
|
if cleanPath == "/attachments" || cleanPath == "/attachments/" {
|
||||||
|
return "", errors.New("attachment uuid is required")
|
||||||
|
}
|
||||||
|
joinedPath := strings.TrimRight(baseURL.Path, "/") + cleanPath
|
||||||
|
if joinedPath == "" {
|
||||||
|
joinedPath = cleanPath
|
||||||
|
}
|
||||||
|
baseURL.Path = joinedPath
|
||||||
|
baseURL.RawPath = joinedPath
|
||||||
|
return baseURL.String(), nil
|
||||||
|
}
|
||||||
|
|
||||||
// DoJSON performs an API request and decodes a JSON response into respOut (if non-nil).
|
// DoJSON performs an API request and decodes a JSON response into respOut (if non-nil).
|
||||||
// It returns the HTTP status code.
|
// It returns the HTTP status code.
|
||||||
func DoJSON(ctx context.Context, method, path string, query url.Values, body, respOut any) (int, error) {
|
func DoJSON(ctx context.Context, method, path string, query url.Values, body, respOut any) (int, error) {
|
||||||
@@ -91,45 +133,142 @@ func DoJSON(ctx context.Context, method, path string, query url.Values, body, re
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return 0, err
|
return 0, err
|
||||||
}
|
}
|
||||||
|
|
||||||
req, err := http.NewRequestWithContext(ctx, method, u, bodyReader)
|
req, err := http.NewRequestWithContext(ctx, method, u, bodyReader)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return 0, fmt.Errorf("create request: %w", err)
|
return 0, fmt.Errorf("create request: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
token := tokenFromContext(ctx)
|
token := tokenFromContext(ctx)
|
||||||
if token != "" {
|
if token != "" {
|
||||||
req.Header.Set("Authorization", "token "+token)
|
req.Header.Set("Authorization", "token "+token)
|
||||||
}
|
}
|
||||||
req.Header.Set("Accept", "application/json")
|
|
||||||
if body != nil {
|
if body != nil {
|
||||||
req.Header.Set("Content-Type", "application/json")
|
req.Header.Set("Content-Type", "application/json")
|
||||||
}
|
}
|
||||||
|
if respOut != nil {
|
||||||
|
req.Header.Set("Accept", "application/json")
|
||||||
|
}
|
||||||
|
|
||||||
client := restHTTPClient()
|
resp, err := restHTTPClient().Do(req)
|
||||||
resp, err := client.Do(req)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return 0, fmt.Errorf("do request: %w", err)
|
return 0, fmt.Errorf("do request: %w", err)
|
||||||
}
|
}
|
||||||
defer resp.Body.Close()
|
defer resp.Body.Close()
|
||||||
|
|
||||||
if resp.StatusCode < 200 || resp.StatusCode >= 300 {
|
if resp.StatusCode < 200 || resp.StatusCode >= 300 {
|
||||||
bodySnippet, _ := io.ReadAll(io.LimitReader(resp.Body, errBodySnippetSize))
|
bodySnippet, err := io.ReadAll(io.LimitReader(resp.Body, errBodySnippetSize+1))
|
||||||
|
if err != nil {
|
||||||
|
return resp.StatusCode, fmt.Errorf("read response: %w", err)
|
||||||
|
}
|
||||||
|
if len(bodySnippet) > errBodySnippetSize {
|
||||||
|
bodySnippet = bodySnippet[:errBodySnippetSize]
|
||||||
|
}
|
||||||
return resp.StatusCode, &HTTPError{StatusCode: resp.StatusCode, Body: strings.TrimSpace(string(bodySnippet))}
|
return resp.StatusCode, &HTTPError{StatusCode: resp.StatusCode, Body: strings.TrimSpace(string(bodySnippet))}
|
||||||
}
|
}
|
||||||
|
|
||||||
if respOut == nil {
|
if respOut == nil {
|
||||||
_, _ = io.Copy(io.Discard, resp.Body) // best-effort
|
_, _ = io.Copy(io.Discard, resp.Body)
|
||||||
return resp.StatusCode, nil
|
return resp.StatusCode, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := json.NewDecoder(resp.Body).Decode(respOut); err != nil {
|
if err := json.NewDecoder(resp.Body).Decode(respOut); err != nil {
|
||||||
return resp.StatusCode, fmt.Errorf("decode response: %w", err)
|
return resp.StatusCode, fmt.Errorf("decode response: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
return resp.StatusCode, nil
|
return resp.StatusCode, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func attachmentHTTPClient(origin *url.URL) *http.Client {
|
||||||
|
return &http.Client{
|
||||||
|
Transport: sharedTransport(),
|
||||||
|
CheckRedirect: func(req *http.Request, via []*http.Request) error {
|
||||||
|
if err := checkRedirect(req, via); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
if req.URL.Scheme != origin.Scheme || req.URL.Host != origin.Host {
|
||||||
|
return errors.New("attachment redirect changed origin")
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func OpenAttachment(ctx context.Context, attachmentPath, accept string) (*AttachmentResponse, error) {
|
||||||
|
u, err := buildAttachmentURL(attachmentPath)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
origin, err := url.Parse(u)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
req, err := http.NewRequestWithContext(ctx, http.MethodGet, u, nil)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("create request: %w", err)
|
||||||
|
}
|
||||||
|
token := tokenFromContext(ctx)
|
||||||
|
if token != "" {
|
||||||
|
req.Header.Set("Authorization", "token "+token)
|
||||||
|
}
|
||||||
|
if accept != "" {
|
||||||
|
req.Header.Set("Accept", accept)
|
||||||
|
}
|
||||||
|
|
||||||
|
resp, err := attachmentHTTPClient(origin).Do(req)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("do request: %w", err)
|
||||||
|
}
|
||||||
|
if resp.StatusCode < 200 || resp.StatusCode >= 300 {
|
||||||
|
defer resp.Body.Close()
|
||||||
|
bodySnippet, readErr := io.ReadAll(io.LimitReader(resp.Body, errBodySnippetSize+1))
|
||||||
|
if readErr != nil {
|
||||||
|
return nil, fmt.Errorf("read response: %w", readErr)
|
||||||
|
}
|
||||||
|
if len(bodySnippet) > errBodySnippetSize {
|
||||||
|
bodySnippet = bodySnippet[:errBodySnippetSize]
|
||||||
|
}
|
||||||
|
return nil, &HTTPError{StatusCode: resp.StatusCode, Body: strings.TrimSpace(string(bodySnippet))}
|
||||||
|
}
|
||||||
|
return &AttachmentResponse{Body: resp.Body, ContentType: resp.Header.Get("Content-Type"), StatusCode: resp.StatusCode}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// DownloadAttachment downloads attachment bytes from a same-host attachment path.
|
||||||
|
func DownloadAttachment(ctx context.Context, attachmentPath, accept string) ([]byte, string, int, error) {
|
||||||
|
resp, err := OpenAttachment(ctx, attachmentPath, accept)
|
||||||
|
if err != nil {
|
||||||
|
var httpErr *HTTPError
|
||||||
|
if errors.As(err, &httpErr) {
|
||||||
|
return nil, "", httpErr.StatusCode, err
|
||||||
|
}
|
||||||
|
return nil, "", 0, err
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
respBytes, err := io.ReadAll(resp.Body)
|
||||||
|
if err != nil {
|
||||||
|
return nil, "", resp.StatusCode, fmt.Errorf("read response: %w", err)
|
||||||
|
}
|
||||||
|
return respBytes, resp.ContentType, resp.StatusCode, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func WriteAttachment(body io.Reader, outputPath string) (int64, error) {
|
||||||
|
file, err := os.OpenFile(outputPath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0o600)
|
||||||
|
if err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
defer file.Close()
|
||||||
|
if err := file.Chmod(0o600); err != nil {
|
||||||
|
return 0, err
|
||||||
|
}
|
||||||
|
|
||||||
|
written, err := io.Copy(file, body)
|
||||||
|
if err != nil {
|
||||||
|
return written, err
|
||||||
|
}
|
||||||
|
return written, file.Close()
|
||||||
|
}
|
||||||
|
|
||||||
// DoBytes performs an API request and returns the raw response bytes.
|
// DoBytes performs an API request and returns the raw response bytes.
|
||||||
// It returns the HTTP status code.
|
|
||||||
func DoBytes(ctx context.Context, method, path string, query url.Values, body any, accept string) ([]byte, int, error) {
|
func DoBytes(ctx context.Context, method, path string, query url.Values, body any, accept string) ([]byte, int, error) {
|
||||||
var bodyReader io.Reader
|
var bodyReader io.Reader
|
||||||
if body != nil {
|
if body != nil {
|
||||||
@@ -144,24 +283,23 @@ func DoBytes(ctx context.Context, method, path string, query url.Values, body an
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, 0, err
|
return nil, 0, err
|
||||||
}
|
}
|
||||||
|
|
||||||
req, err := http.NewRequestWithContext(ctx, method, u, bodyReader)
|
req, err := http.NewRequestWithContext(ctx, method, u, bodyReader)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, 0, fmt.Errorf("create request: %w", err)
|
return nil, 0, fmt.Errorf("create request: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
token := tokenFromContext(ctx)
|
token := tokenFromContext(ctx)
|
||||||
if token != "" {
|
if token != "" {
|
||||||
req.Header.Set("Authorization", "token "+token)
|
req.Header.Set("Authorization", "token "+token)
|
||||||
}
|
}
|
||||||
if accept != "" {
|
|
||||||
req.Header.Set("Accept", accept)
|
|
||||||
}
|
|
||||||
if body != nil {
|
if body != nil {
|
||||||
req.Header.Set("Content-Type", "application/json")
|
req.Header.Set("Content-Type", "application/json")
|
||||||
}
|
}
|
||||||
|
if accept != "" {
|
||||||
|
req.Header.Set("Accept", accept)
|
||||||
|
}
|
||||||
|
|
||||||
client := restHTTPClient()
|
resp, err := restHTTPClient().Do(req)
|
||||||
resp, err := client.Do(req)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, 0, fmt.Errorf("do request: %w", err)
|
return nil, 0, fmt.Errorf("do request: %w", err)
|
||||||
}
|
}
|
||||||
@@ -171,7 +309,6 @@ func DoBytes(ctx context.Context, method, path string, query url.Values, body an
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, resp.StatusCode, fmt.Errorf("read response: %w", err)
|
return nil, resp.StatusCode, fmt.Errorf("read response: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
if resp.StatusCode < 200 || resp.StatusCode >= 300 {
|
if resp.StatusCode < 200 || resp.StatusCode >= 300 {
|
||||||
bodySnippet := respBytes
|
bodySnippet := respBytes
|
||||||
if len(bodySnippet) > errBodySnippetSize {
|
if len(bodySnippet) > errBodySnippetSize {
|
||||||
@@ -179,6 +316,5 @@ func DoBytes(ctx context.Context, method, path string, query url.Values, body an
|
|||||||
}
|
}
|
||||||
return nil, resp.StatusCode, &HTTPError{StatusCode: resp.StatusCode, Body: strings.TrimSpace(string(bodySnippet))}
|
return nil, resp.StatusCode, &HTTPError{StatusCode: resp.StatusCode, Body: strings.TrimSpace(string(bodySnippet))}
|
||||||
}
|
}
|
||||||
|
|
||||||
return respBytes, resp.StatusCode, nil
|
return respBytes, resp.StatusCode, nil
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -2,6 +2,11 @@ package gitea
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
"errors"
|
||||||
|
"io"
|
||||||
|
"net/http"
|
||||||
|
"net/http/httptest"
|
||||||
|
"strings"
|
||||||
"testing"
|
"testing"
|
||||||
|
|
||||||
mcpContext "gitea.com/gitea/gitea-mcp/pkg/context"
|
mcpContext "gitea.com/gitea/gitea-mcp/pkg/context"
|
||||||
@@ -28,3 +33,32 @@ func TestTokenFromContext(t *testing.T) {
|
|||||||
}
|
}
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestDoJSON_LimitsErrorResponseBody(t *testing.T) {
|
||||||
|
payload := strings.Repeat("x", errBodySnippetSize+100)
|
||||||
|
srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||||
|
w.WriteHeader(http.StatusBadRequest)
|
||||||
|
_, _ = io.WriteString(w, payload)
|
||||||
|
}))
|
||||||
|
defer srv.Close()
|
||||||
|
|
||||||
|
origHost := flag.Host
|
||||||
|
defer func() { flag.Host = origHost }()
|
||||||
|
flag.Host = srv.URL
|
||||||
|
|
||||||
|
status, err := DoJSON(context.Background(), http.MethodGet, "repos/owner/repo", nil, nil, nil)
|
||||||
|
if err == nil {
|
||||||
|
t.Fatal("expected error")
|
||||||
|
}
|
||||||
|
if status != http.StatusBadRequest {
|
||||||
|
t.Fatalf("expected status %d, got %d", http.StatusBadRequest, status)
|
||||||
|
}
|
||||||
|
|
||||||
|
var httpErr *HTTPError
|
||||||
|
if !errors.As(err, &httpErr) {
|
||||||
|
t.Fatalf("expected HTTPError, got %T", err)
|
||||||
|
}
|
||||||
|
if len(httpErr.Body) != errBodySnippetSize {
|
||||||
|
t.Fatalf("expected body length %d, got %d", errBodySnippetSize, len(httpErr.Body))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user