summaryrefslogtreecommitdiffstats
path: root/modules/uri
diff options
context:
space:
mode:
Diffstat (limited to 'modules/uri')
-rw-r--r--modules/uri/uri.go42
-rw-r--r--modules/uri/uri_test.go19
2 files changed, 61 insertions, 0 deletions
diff --git a/modules/uri/uri.go b/modules/uri/uri.go
new file mode 100644
index 0000000..768bc66
--- /dev/null
+++ b/modules/uri/uri.go
@@ -0,0 +1,42 @@
+// Copyright 2020 The Gitea Authors. All rights reserved.
+// SPDX-License-Identifier: MIT
+
+package uri
+
+import (
+ "fmt"
+ "io"
+ "net/http"
+ "net/url"
+ "os"
+ "strings"
+)
+
+// ErrURISchemeNotSupported represents a scheme error
+type ErrURISchemeNotSupported struct {
+ Scheme string
+}
+
+func (e ErrURISchemeNotSupported) Error() string {
+ return fmt.Sprintf("Unsupported scheme: %v", e.Scheme)
+}
+
+// Open open a local file or a remote file
+func Open(uriStr string) (io.ReadCloser, error) {
+ u, err := url.Parse(uriStr)
+ if err != nil {
+ return nil, err
+ }
+ switch strings.ToLower(u.Scheme) {
+ case "http", "https":
+ f, err := http.Get(uriStr)
+ if err != nil {
+ return nil, err
+ }
+ return f.Body, nil
+ case "file":
+ return os.Open(u.Path)
+ default:
+ return nil, ErrURISchemeNotSupported{Scheme: u.Scheme}
+ }
+}
diff --git a/modules/uri/uri_test.go b/modules/uri/uri_test.go
new file mode 100644
index 0000000..71a8985
--- /dev/null
+++ b/modules/uri/uri_test.go
@@ -0,0 +1,19 @@
+// Copyright 2020 The Gitea Authors. All rights reserved.
+// SPDX-License-Identifier: MIT
+
+package uri
+
+import (
+ "path/filepath"
+ "testing"
+
+ "github.com/stretchr/testify/require"
+)
+
+func TestReadURI(t *testing.T) {
+ p, err := filepath.Abs("./uri.go")
+ require.NoError(t, err)
+ f, err := Open("file://" + p)
+ require.NoError(t, err)
+ defer f.Close()
+}