mv more fkt to http-client
This commit is contained in:
parent
310d740cee
commit
587bd07372
2 changed files with 35 additions and 15 deletions
|
@ -1,7 +1,7 @@
|
||||||
// Copyright 2022, 2023 The Forgejo & Gitea Authors. All rights reserved.
|
// Copyright 2022, 2023 The Forgejo & Gitea Authors. All rights reserved.
|
||||||
// SPDX-License-Identifier: MIT
|
// SPDX-License-Identifier: MIT
|
||||||
|
|
||||||
// TODO: Think about whether this should be moved to services/actifitypub (compare to exosy/services/activitypub/client.go)
|
// TODO: Think about whether this should be moved to services/activitypub (compare to exosy/services/activitypub/client.go)
|
||||||
package activitypub
|
package activitypub
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
@ -11,11 +11,13 @@ import (
|
||||||
"crypto/x509"
|
"crypto/x509"
|
||||||
"encoding/pem"
|
"encoding/pem"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"io"
|
||||||
"net/http"
|
"net/http"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
user_model "code.gitea.io/gitea/models/user"
|
user_model "code.gitea.io/gitea/models/user"
|
||||||
|
"code.gitea.io/gitea/modules/log"
|
||||||
"code.gitea.io/gitea/modules/proxy"
|
"code.gitea.io/gitea/modules/proxy"
|
||||||
"code.gitea.io/gitea/modules/setting"
|
"code.gitea.io/gitea/modules/setting"
|
||||||
|
|
||||||
|
@ -134,3 +136,35 @@ func (c *Client) Get(to string) (resp *http.Response, err error) { // ToDo: we m
|
||||||
resp, err = c.client.Do(req)
|
resp, err = c.client.Do(req)
|
||||||
return resp, err
|
return resp, err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Create an http GET request with forgejo/gitea specific headers
|
||||||
|
func (c *Client) GetBody(uri string) ([]byte, error) {
|
||||||
|
response, err := c.Get(uri)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
log.Debug("Client: got status: %v", response.Status)
|
||||||
|
if response.StatusCode != 200 {
|
||||||
|
err = fmt.Errorf("got non 200 status code for id: %v", uri)
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
defer response.Body.Close()
|
||||||
|
body, err := io.ReadAll(response.Body)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
log.Debug("Client: got body: %v", charLimiter(string(body), 120))
|
||||||
|
return body, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Limit number of characters in a string (useful to prevent log injection attacks and overly long log outputs)
|
||||||
|
// Thanks to https://www.socketloop.com/tutorials/golang-characters-limiter-example
|
||||||
|
func charLimiter(s string, limit int) string {
|
||||||
|
reader := strings.NewReader(s)
|
||||||
|
buff := make([]byte, limit)
|
||||||
|
n, _ := io.ReadAtLeast(reader, buff, limit)
|
||||||
|
if n != 0 {
|
||||||
|
return fmt.Sprint(string(buff), "...")
|
||||||
|
}
|
||||||
|
return s
|
||||||
|
}
|
||||||
|
|
|
@ -4,9 +4,7 @@
|
||||||
package utils
|
package utils
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
|
||||||
"html"
|
"html"
|
||||||
"io"
|
|
||||||
"net/url"
|
"net/url"
|
||||||
"strings"
|
"strings"
|
||||||
|
|
||||||
|
@ -30,15 +28,3 @@ func IsExternalURL(rawURL string) bool {
|
||||||
}
|
}
|
||||||
return false
|
return false
|
||||||
}
|
}
|
||||||
|
|
||||||
// Limit number of characters in a string (useful to prevent log injection attacks and overly long log outputs)
|
|
||||||
// Thanks to https://www.socketloop.com/tutorials/golang-characters-limiter-example
|
|
||||||
func CharLimiter(s string, limit int) string {
|
|
||||||
reader := strings.NewReader(s)
|
|
||||||
buff := make([]byte, limit)
|
|
||||||
n, _ := io.ReadAtLeast(reader, buff, limit)
|
|
||||||
if n != 0 {
|
|
||||||
return fmt.Sprint(string(buff), "...")
|
|
||||||
}
|
|
||||||
return s
|
|
||||||
}
|
|
||||||
|
|
Loading…
Reference in a new issue