mirror of
https://github.com/go-gitea/gitea
synced 2024-09-19 10:16:03 +00:00
62e6c9bc6c
* Add a storage layer for attachments * Fix some bug * fix test * Fix copyright head and lint * Fix bug * Add setting for minio and flags for migrate-storage * Add documents * fix lint * Add test for minio store type on attachments * fix test * fix test * Apply suggestions from code review Co-authored-by: guillep2k <18600385+guillep2k@users.noreply.github.com> * Add warning when storage migrated successfully * Fix drone * fix test * rebase * Fix test * display the error on console * Move minio test to amd64 since minio docker don't support arm64 * refactor the codes * add trace * Fix test * remove log on xorm * Fi download bug * Add a storage layer for attachments * Add setting for minio and flags for migrate-storage * fix lint * Add test for minio store type on attachments * Apply suggestions from code review Co-authored-by: guillep2k <18600385+guillep2k@users.noreply.github.com> * Fix drone * fix test * Fix test * display the error on console * Move minio test to amd64 since minio docker don't support arm64 * refactor the codes * add trace * Fix test * Add URL function to serve attachments directly from S3/Minio * Add ability to enable/disable redirection in attachment configuration * Fix typo * Add a storage layer for attachments * Add setting for minio and flags for migrate-storage * fix lint * Add test for minio store type on attachments * Apply suggestions from code review Co-authored-by: guillep2k <18600385+guillep2k@users.noreply.github.com> * Fix drone * fix test * Fix test * display the error on console * Move minio test to amd64 since minio docker don't support arm64 * don't change unrelated files * Fix lint * Fix build * update go.mod and go.sum * Use github.com/minio/minio-go/v6 * Remove unused function * Upgrade minio to v7 and some other improvements * fix lint * Fix go mod Co-authored-by: guillep2k <18600385+guillep2k@users.noreply.github.com> Co-authored-by: Tyler <tystuyfzand@gmail.com>
343 lines
8.0 KiB
Go
Vendored
343 lines
8.0 KiB
Go
Vendored
/*
|
|
* MinIO Cloud Storage, (C) 2020 MinIO, Inc.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
package tags
|
|
|
|
import (
|
|
"encoding/xml"
|
|
"io"
|
|
"net/url"
|
|
"strings"
|
|
"unicode/utf8"
|
|
)
|
|
|
|
// Error contains tag specific error.
|
|
type Error interface {
|
|
error
|
|
Code() string
|
|
}
|
|
|
|
type errTag struct {
|
|
code string
|
|
message string
|
|
}
|
|
|
|
// Code contains error code.
|
|
func (err errTag) Code() string {
|
|
return err.code
|
|
}
|
|
|
|
// Error contains error message.
|
|
func (err errTag) Error() string {
|
|
return err.message
|
|
}
|
|
|
|
var (
|
|
errTooManyObjectTags = &errTag{"BadRequest", "Tags cannot be more than 10"}
|
|
errTooManyTags = &errTag{"BadRequest", "Tags cannot be more than 50"}
|
|
errInvalidTagKey = &errTag{"InvalidTag", "The TagKey you have provided is invalid"}
|
|
errInvalidTagValue = &errTag{"InvalidTag", "The TagValue you have provided is invalid"}
|
|
errDuplicateTagKey = &errTag{"InvalidTag", "Cannot provide multiple Tags with the same key"}
|
|
)
|
|
|
|
// Tag comes with limitation as per
|
|
// https://docs.aws.amazon.com/AmazonS3/latest/dev/object-tagging.html amd
|
|
// https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/Using_Tags.html#tag-restrictions
|
|
const (
|
|
maxKeyLength = 128
|
|
maxValueLength = 256
|
|
maxObjectTagCount = 10
|
|
maxTagCount = 50
|
|
)
|
|
|
|
func checkKey(key string) error {
|
|
if len(key) == 0 || utf8.RuneCountInString(key) > maxKeyLength || strings.Contains(key, "&") {
|
|
return errInvalidTagKey
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func checkValue(value string) error {
|
|
if utf8.RuneCountInString(value) > maxValueLength || strings.Contains(value, "&") {
|
|
return errInvalidTagValue
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Tag denotes key and value.
|
|
type Tag struct {
|
|
Key string `xml:"Key"`
|
|
Value string `xml:"Value"`
|
|
}
|
|
|
|
func (tag Tag) String() string {
|
|
return tag.Key + "=" + tag.Value
|
|
}
|
|
|
|
// IsEmpty returns whether this tag is empty or not.
|
|
func (tag Tag) IsEmpty() bool {
|
|
return tag.Key == ""
|
|
}
|
|
|
|
// Validate checks this tag.
|
|
func (tag Tag) Validate() error {
|
|
if err := checkKey(tag.Key); err != nil {
|
|
return err
|
|
}
|
|
|
|
return checkValue(tag.Value)
|
|
}
|
|
|
|
// MarshalXML encodes to XML data.
|
|
func (tag Tag) MarshalXML(e *xml.Encoder, start xml.StartElement) error {
|
|
if err := tag.Validate(); err != nil {
|
|
return err
|
|
}
|
|
|
|
type subTag Tag // to avoid recursively calling MarshalXML()
|
|
return e.EncodeElement(subTag(tag), start)
|
|
}
|
|
|
|
// UnmarshalXML decodes XML data to tag.
|
|
func (tag *Tag) UnmarshalXML(d *xml.Decoder, start xml.StartElement) error {
|
|
type subTag Tag // to avoid recursively calling UnmarshalXML()
|
|
var st subTag
|
|
if err := d.DecodeElement(&st, &start); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := Tag(st).Validate(); err != nil {
|
|
return err
|
|
}
|
|
|
|
*tag = Tag(st)
|
|
return nil
|
|
}
|
|
|
|
// tagSet represents list of unique tags.
|
|
type tagSet struct {
|
|
tagMap map[string]string
|
|
isObject bool
|
|
}
|
|
|
|
func (tags tagSet) String() string {
|
|
s := []string{}
|
|
for key, value := range tags.tagMap {
|
|
s = append(s, key+"="+value)
|
|
}
|
|
|
|
return strings.Join(s, "&")
|
|
}
|
|
|
|
func (tags *tagSet) remove(key string) {
|
|
delete(tags.tagMap, key)
|
|
}
|
|
|
|
func (tags *tagSet) set(key, value string, failOnExist bool) error {
|
|
if failOnExist {
|
|
if _, found := tags.tagMap[key]; found {
|
|
return errDuplicateTagKey
|
|
}
|
|
}
|
|
|
|
if err := checkKey(key); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := checkValue(value); err != nil {
|
|
return err
|
|
}
|
|
|
|
if tags.isObject {
|
|
if len(tags.tagMap) == maxObjectTagCount {
|
|
return errTooManyObjectTags
|
|
}
|
|
} else if len(tags.tagMap) == maxTagCount {
|
|
return errTooManyTags
|
|
}
|
|
|
|
tags.tagMap[key] = value
|
|
return nil
|
|
}
|
|
|
|
func (tags tagSet) toMap() map[string]string {
|
|
m := make(map[string]string)
|
|
for key, value := range tags.tagMap {
|
|
m[key] = value
|
|
}
|
|
return m
|
|
}
|
|
|
|
// MarshalXML encodes to XML data.
|
|
func (tags tagSet) MarshalXML(e *xml.Encoder, start xml.StartElement) error {
|
|
tagList := struct {
|
|
Tags []Tag `xml:"Tag"`
|
|
}{}
|
|
|
|
for key, value := range tags.tagMap {
|
|
tagList.Tags = append(tagList.Tags, Tag{key, value})
|
|
}
|
|
|
|
return e.EncodeElement(tagList, start)
|
|
}
|
|
|
|
// UnmarshalXML decodes XML data to tag list.
|
|
func (tags *tagSet) UnmarshalXML(d *xml.Decoder, start xml.StartElement) error {
|
|
tagList := struct {
|
|
Tags []Tag `xml:"Tag"`
|
|
}{}
|
|
|
|
if err := d.DecodeElement(&tagList, &start); err != nil {
|
|
return err
|
|
}
|
|
|
|
if tags.isObject {
|
|
if len(tagList.Tags) > maxObjectTagCount {
|
|
return errTooManyObjectTags
|
|
}
|
|
} else if len(tagList.Tags) > maxTagCount {
|
|
return errTooManyTags
|
|
}
|
|
|
|
m := map[string]string{}
|
|
for _, tag := range tagList.Tags {
|
|
if _, found := m[tag.Key]; found {
|
|
return errDuplicateTagKey
|
|
}
|
|
|
|
m[tag.Key] = tag.Value
|
|
}
|
|
|
|
tags.tagMap = m
|
|
return nil
|
|
}
|
|
|
|
type tagging struct {
|
|
XMLName xml.Name `xml:"Tagging"`
|
|
TagSet *tagSet `xml:"TagSet"`
|
|
}
|
|
|
|
// Tags is list of tags of XML request/response as per
|
|
// https://docs.aws.amazon.com/AmazonS3/latest/API/API_GetBucketTagging.html#API_GetBucketTagging_RequestBody
|
|
type Tags tagging
|
|
|
|
func (tags Tags) String() string {
|
|
return tags.TagSet.String()
|
|
}
|
|
|
|
// Remove removes a tag by its key.
|
|
func (tags *Tags) Remove(key string) {
|
|
tags.TagSet.remove(key)
|
|
}
|
|
|
|
// Set sets new tag.
|
|
func (tags *Tags) Set(key, value string) error {
|
|
return tags.TagSet.set(key, value, false)
|
|
}
|
|
|
|
// ToMap returns copy of tags.
|
|
func (tags Tags) ToMap() map[string]string {
|
|
return tags.TagSet.toMap()
|
|
}
|
|
|
|
// MapToObjectTags converts an input map of key and value into
|
|
// *Tags data structure with validation.
|
|
func MapToObjectTags(tagMap map[string]string) (*Tags, error) {
|
|
return NewTags(tagMap, true)
|
|
}
|
|
|
|
// MapToBucketTags converts an input map of key and value into
|
|
// *Tags data structure with validation.
|
|
func MapToBucketTags(tagMap map[string]string) (*Tags, error) {
|
|
return NewTags(tagMap, false)
|
|
}
|
|
|
|
// NewTags creates Tags from tagMap, If isObject is set, it validates for object tags.
|
|
func NewTags(tagMap map[string]string, isObject bool) (*Tags, error) {
|
|
tagging := &Tags{
|
|
TagSet: &tagSet{
|
|
tagMap: make(map[string]string),
|
|
isObject: isObject,
|
|
},
|
|
}
|
|
|
|
for key, value := range tagMap {
|
|
if err := tagging.TagSet.set(key, value, true); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
return tagging, nil
|
|
}
|
|
|
|
func unmarshalXML(reader io.Reader, isObject bool) (*Tags, error) {
|
|
tagging := &Tags{
|
|
TagSet: &tagSet{
|
|
tagMap: make(map[string]string),
|
|
isObject: isObject,
|
|
},
|
|
}
|
|
|
|
if err := xml.NewDecoder(reader).Decode(tagging); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return tagging, nil
|
|
}
|
|
|
|
// ParseBucketXML decodes XML data of tags in reader specified in
|
|
// https://docs.aws.amazon.com/AmazonS3/latest/API/API_PutBucketTagging.html#API_PutBucketTagging_RequestSyntax.
|
|
func ParseBucketXML(reader io.Reader) (*Tags, error) {
|
|
return unmarshalXML(reader, false)
|
|
}
|
|
|
|
// ParseObjectXML decodes XML data of tags in reader specified in
|
|
// https://docs.aws.amazon.com/AmazonS3/latest/API/API_PutObjectTagging.html#API_PutObjectTagging_RequestSyntax
|
|
func ParseObjectXML(reader io.Reader) (*Tags, error) {
|
|
return unmarshalXML(reader, true)
|
|
}
|
|
|
|
// Parse decodes HTTP query formatted string into tags which is limited by isObject.
|
|
// A query formatted string is like "key1=value1&key2=value2".
|
|
func Parse(s string, isObject bool) (*Tags, error) {
|
|
values, err := url.ParseQuery(s)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
tagging := &Tags{
|
|
TagSet: &tagSet{
|
|
tagMap: make(map[string]string),
|
|
isObject: isObject,
|
|
},
|
|
}
|
|
|
|
for key := range values {
|
|
if err := tagging.TagSet.set(key, values.Get(key), true); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
return tagging, nil
|
|
}
|
|
|
|
// ParseObjectTags decodes HTTP query formatted string into tags. A query formatted string is like "key1=value1&key2=value2".
|
|
func ParseObjectTags(s string) (*Tags, error) {
|
|
return Parse(s, true)
|
|
}
|