forked from mirror/gitea
a5df7ba6bf
If http.Get() returns an error return nil and err before attempting to use the broken file. Thanks to walker xiong for spotting this bug. Signed-off-by: Andrew Thornton <art27@cantab.net>
44 lines
883 B
Go
44 lines
883 B
Go
// 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 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}
|
|
}
|
|
}
|