summaryrefslogtreecommitdiffstats
path: root/modules/storage/storage.go
blob: e355d2459f50b6bd8e391936b4abd0afe2976017 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
// Copyright 2020 The Gitea Authors. All rights reserved.
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.

package storage

import (
	"context"
	"errors"
	"fmt"
	"io"
	"net/url"
	"time"

	"code.gitea.io/gitea/modules/setting"
)

var (
	// ErrURLNotSupported represents url is not supported
	ErrURLNotSupported = errors.New("url method not supported")
)

// Object represents the object on the storage
type Object interface {
	io.ReadCloser
	io.Seeker
}

// ObjectInfo represents the object info on the storage
type ObjectInfo interface {
	Name() string
	Size() int64
	ModTime() time.Time
}

// ObjectStorage represents an object storage to handle a bucket and files
type ObjectStorage interface {
	Open(path string) (Object, error)
	Save(path string, r io.Reader) (int64, error)
	Stat(path string) (ObjectInfo, error)
	Delete(path string) error
	URL(path, name string) (*url.URL, error)
}

// Copy copys a file from source ObjectStorage to dest ObjectStorage
func Copy(dstStorage ObjectStorage, dstPath string, srcStorage ObjectStorage, srcPath string) (int64, error) {
	f, err := srcStorage.Open(srcPath)
	if err != nil {
		return 0, err
	}
	defer f.Close()

	return dstStorage.Save(dstPath, f)
}

var (
	// Attachments represents attachments storage
	Attachments ObjectStorage

	// LFS represents lfs storage
	LFS ObjectStorage
)

// Init init the stoarge
func Init() error {
	if err := initAttachments(); err != nil {
		return err
	}

	return initLFS()
}

func initAttachments() error {
	var err error
	switch setting.Attachment.StoreType {
	case "local":
		Attachments, err = NewLocalStorage(setting.Attachment.Path)
	case "minio":
		minio := setting.Attachment.Minio
		Attachments, err = NewMinioStorage(
			context.Background(),
			minio.Endpoint,
			minio.AccessKeyID,
			minio.SecretAccessKey,
			minio.Bucket,
			minio.Location,
			minio.BasePath,
			minio.UseSSL,
		)
	default:
		return fmt.Errorf("Unsupported attachment store type: %s", setting.Attachment.StoreType)
	}

	if err != nil {
		return err
	}

	return nil
}

func initLFS() error {
	var err error
	switch setting.LFS.StoreType {
	case "local":
		LFS, err = NewLocalStorage(setting.LFS.ContentPath)
	case "minio":
		minio := setting.LFS.Minio
		LFS, err = NewMinioStorage(
			context.Background(),
			minio.Endpoint,
			minio.AccessKeyID,
			minio.SecretAccessKey,
			minio.Bucket,
			minio.Location,
			minio.BasePath,
			minio.UseSSL,
		)
	default:
		return fmt.Errorf("Unsupported LFS store type: %s", setting.LFS.StoreType)
	}

	if err != nil {
		return err
	}

	return nil
}