Add aggressive linting
Some checks failed
ci/woodpecker/push/woodpecker Pipeline failed

This commit is contained in:
Torjus Håkestad 2022-01-24 20:25:52 +01:00
parent 763d691b6c
commit e7b0c5fa33
18 changed files with 188 additions and 32 deletions

34
.golangci.yml Normal file
View File

@ -0,0 +1,34 @@
run:
tests: false
linters:
enable:
- deadcode
- errcheck
- gosimple
- govet
- ineffassign
- staticcheck
- structcheck
- typecheck
- unused
- varcheck
- gosec
- asciicheck
- bidichk
- bodyclose
- gomnd
- ifshort
- misspell
- prealloc
- tagliatelle
- ireturn
- gocritic
- whitespace
- wsl
- stylecheck
- exportloopref
- godot
linters-settings:
gomnd:
ignored-functions:
- "strconv.ParseUint"

View File

@ -10,6 +10,13 @@ pipeline:
branch: master branch: master
event: [push, pull_request, tag, deployment] event: [push, pull_request, tag, deployment]
lint:
image: golangci/golangci-lint:latest
commands:
- golangci-lint run
when:
event: [push, pull_request]
image-latest: image-latest:
image: plugins/docker image: plugins/docker
settings: settings:

View File

@ -18,6 +18,8 @@ import (
"go.uber.org/zap" "go.uber.org/zap"
) )
const multipartMaxMemory = 1024 * 1024 * 100
type HTTPServer struct { type HTTPServer struct {
Files files.FileStore Files files.FileStore
Users users.UserStore Users users.UserStore
@ -41,8 +43,8 @@ func NewHTTPServer(cfg *gpaste.ServerConfig) *HTTPServer {
// Create initial user // Create initial user
// TODO: Do properly // TODO: Do properly
user := &users.User{Username: "admin", Role: users.RoleAdmin} user := &users.User{Username: "admin", Role: users.RoleAdmin}
user.SetPassword("admin") _ = user.SetPassword("admin")
srv.Users.Store(user) _ = srv.Users.Store(user)
r := chi.NewRouter() r := chi.NewRouter()
r.Use(middleware.RealIP) r.Use(middleware.RealIP)
@ -82,9 +84,12 @@ func (s *HTTPServer) HandlerAPIFilePost(w http.ResponseWriter, r *http.Request)
if err != nil { if err != nil {
w.WriteHeader(http.StatusInternalServerError) w.WriteHeader(http.StatusInternalServerError)
s.Logger.Warnw("Error storing file.", "req_id", reqID, "error", err, "id", f.ID, "remote_addr", r.RemoteAddr) s.Logger.Warnw("Error storing file.", "req_id", reqID, "error", err, "id", f.ID, "remote_addr", r.RemoteAddr)
return return
} }
s.Logger.Infow("Stored file.", "req_id", reqID, "id", f.ID, "remote_addr", r.RemoteAddr) s.Logger.Infow("Stored file.", "req_id", reqID, "id", f.ID, "remote_addr", r.RemoteAddr)
fileURL := path.Join(s.config.URL, "/api/file", f.ID) fileURL := path.Join(s.config.URL, "/api/file", f.ID)
resp := &ResponseAPIFilePost{ resp := &ResponseAPIFilePost{
Message: "OK", Message: "OK",
@ -95,7 +100,9 @@ func (s *HTTPServer) HandlerAPIFilePost(w http.ResponseWriter, r *http.Request)
}, },
}, },
} }
w.WriteHeader(http.StatusAccepted) w.WriteHeader(http.StatusAccepted)
encoder := json.NewEncoder(w) encoder := json.NewEncoder(w)
if err := encoder.Encode(&resp); err != nil { if err := encoder.Encode(&resp); err != nil {
s.Logger.Warnw("Error encoding response to client.", "req_id", reqID, "error", err, "remote_addr", r.RemoteAddr) s.Logger.Warnw("Error encoding response to client.", "req_id", reqID, "error", err, "remote_addr", r.RemoteAddr)
@ -117,6 +124,7 @@ func (s *HTTPServer) HandlerAPIFileGet(w http.ResponseWriter, r *http.Request) {
} }
w.WriteHeader(http.StatusOK) w.WriteHeader(http.StatusOK)
if _, err := io.Copy(w, f.Body); err != nil { if _, err := io.Copy(w, f.Body); err != nil {
reqID := middleware.GetReqID(r.Context()) reqID := middleware.GetReqID(r.Context())
s.Logger.Warnw("Error writing file to client.", "req_id", reqID, "error", err, "remote_addr", r.RemoteAddr) s.Logger.Warnw("Error writing file to client.", "req_id", reqID, "error", err, "remote_addr", r.RemoteAddr)
@ -136,6 +144,7 @@ func (s *HTTPServer) HandlerAPIFileDelete(w http.ResponseWriter, r *http.Request
w.WriteHeader(http.StatusBadRequest) w.WriteHeader(http.StatusBadRequest)
return return
} }
reqID := middleware.GetReqID(r.Context()) reqID := middleware.GetReqID(r.Context())
s.Logger.Infow("Deleted file", "id", id, "req_id", reqID) s.Logger.Infow("Deleted file", "id", id, "req_id", reqID)
} }
@ -145,15 +154,17 @@ func (s *HTTPServer) processMultiPartFormUpload(w http.ResponseWriter, r *http.R
var resp ResponseAPIFilePost var resp ResponseAPIFilePost
if err := r.ParseMultipartForm(1024 * 1024 * 10); err != nil { if err := r.ParseMultipartForm(multipartMaxMemory); err != nil {
s.Logger.Warnw("Error parsing multipart form.", "req_id", reqID, "err", err) s.Logger.Warnw("Error parsing multipart form.", "req_id", reqID, "err", err)
} }
for k := range r.MultipartForm.File { for k := range r.MultipartForm.File {
ff, fh, err := r.FormFile(k) ff, fh, err := r.FormFile(k)
if err != nil { if err != nil {
s.Logger.Warnw("Error reading file from multipart form.", "req_id", reqID, "error", err) s.Logger.Warnw("Error reading file from multipart form.", "req_id", reqID, "error", err)
return return
} }
f := fileFromParams(r) f := fileFromParams(r)
f.ID = uuid.NewString() f.ID = uuid.NewString()
f.OriginalFilename = fh.Filename f.OriginalFilename = fh.Filename
@ -162,18 +173,20 @@ func (s *HTTPServer) processMultiPartFormUpload(w http.ResponseWriter, r *http.R
if err := s.Files.Store(f); err != nil { if err := s.Files.Store(f); err != nil {
w.WriteHeader(http.StatusInternalServerError) w.WriteHeader(http.StatusInternalServerError)
s.Logger.Warnw("Error storing file.", "req_id", reqID, "error", err, "id", f.ID, "remote_addr", r.RemoteAddr) s.Logger.Warnw("Error storing file.", "req_id", reqID, "error", err, "id", f.ID, "remote_addr", r.RemoteAddr)
return return
} }
s.Logger.Infow("Stored file.", "req_id", reqID, "id", f.ID, "filename", f.OriginalFilename, "remote_addr", r.RemoteAddr) s.Logger.Infow("Stored file.", "req_id", reqID, "id", f.ID, "filename", f.OriginalFilename, "remote_addr", r.RemoteAddr)
fileURL := path.Join(s.config.URL, "/api/file", f.ID) fileURL := path.Join(s.config.URL, "/api/file", f.ID)
fileResponse := ResponseAPIFilePostFiles{ID: f.ID, URL: fileURL} fileResponse := ResponseAPIFilePostFiles{ID: f.ID, URL: fileURL}
resp.Files = append(resp.Files, fileResponse) resp.Files = append(resp.Files, fileResponse)
} }
w.WriteHeader(http.StatusAccepted) w.WriteHeader(http.StatusAccepted)
encoder := json.NewEncoder(w) encoder := json.NewEncoder(w)
if err := encoder.Encode(&resp); err != nil { if err := encoder.Encode(&resp); err != nil {
s.Logger.Warnw("Error encoding response to client.", "req_id", reqID, "error", err, "remote_addr", r.RemoteAddr) s.Logger.Warnw("Error encoding response to client.", "req_id", reqID, "error", err, "remote_addr", r.RemoteAddr)
} }
@ -181,9 +194,12 @@ func (s *HTTPServer) processMultiPartFormUpload(w http.ResponseWriter, r *http.R
func (s *HTTPServer) HandlerAPILogin(w http.ResponseWriter, r *http.Request) { func (s *HTTPServer) HandlerAPILogin(w http.ResponseWriter, r *http.Request) {
reqID := middleware.GetReqID(r.Context()) reqID := middleware.GetReqID(r.Context())
var expectedRequest RequestAPILogin var expectedRequest RequestAPILogin
decoder := json.NewDecoder(r.Body) decoder := json.NewDecoder(r.Body)
defer r.Body.Close() defer r.Body.Close()
if err := decoder.Decode(&expectedRequest); err != nil { if err := decoder.Decode(&expectedRequest); err != nil {
w.WriteHeader(http.StatusBadRequest) w.WriteHeader(http.StatusBadRequest)
return return
@ -218,10 +234,12 @@ func (s *HTTPServer) HandlerAPIUserCreate(w http.ResponseWriter, r *http.Request
} }
var req RequestAPIUserCreate var req RequestAPIUserCreate
decoder := json.NewDecoder(r.Body) decoder := json.NewDecoder(r.Body)
if err := decoder.Decode(&req); err != nil { if err := decoder.Decode(&req); err != nil {
s.Logger.Debugw("Error parsing request.", "req_id", reqID, "error", err, "remote_addr", r.RemoteAddr) s.Logger.Debugw("Error parsing request.", "req_id", reqID, "error", err, "remote_addr", r.RemoteAddr)
w.WriteHeader(http.StatusBadRequest) w.WriteHeader(http.StatusBadRequest)
return return
} }
@ -230,14 +248,17 @@ func (s *HTTPServer) HandlerAPIUserCreate(w http.ResponseWriter, r *http.Request
if err := user.SetPassword(req.Password); err != nil { if err := user.SetPassword(req.Password); err != nil {
s.Logger.Warnw("Error setting user password.", "req_id", reqID, "error", err, "remote_addr", r.RemoteAddr) s.Logger.Warnw("Error setting user password.", "req_id", reqID, "error", err, "remote_addr", r.RemoteAddr)
w.WriteHeader(http.StatusBadRequest) w.WriteHeader(http.StatusBadRequest)
return return
} }
if err := s.Users.Store(user); err != nil { if err := s.Users.Store(user); err != nil {
s.Logger.Warnw("Error setting user password.", "req_id", reqID, "error", err, "remote_addr", r.RemoteAddr) s.Logger.Warnw("Error setting user password.", "req_id", reqID, "error", err, "remote_addr", r.RemoteAddr)
w.WriteHeader(http.StatusInternalServerError) w.WriteHeader(http.StatusInternalServerError)
return return
} }
w.WriteHeader(http.StatusAccepted) w.WriteHeader(http.StatusAccepted)
s.Logger.Infow("Created user.", "req_id", reqID, "remote_addr", r.RemoteAddr, "username", req.Username) s.Logger.Infow("Created user.", "req_id", reqID, "remote_addr", r.RemoteAddr, "username", req.Username)
} }
@ -249,6 +270,7 @@ func (s *HTTPServer) HandlerAPIUserList(w http.ResponseWriter, r *http.Request)
if err != nil { if err != nil {
s.Logger.Warnw("Error listing users.", "req_id", reqID, "error", err) s.Logger.Warnw("Error listing users.", "req_id", reqID, "error", err)
w.WriteHeader(http.StatusInternalServerError) w.WriteHeader(http.StatusInternalServerError)
return return
} }
@ -263,12 +285,13 @@ func fileFromParams(r *http.Request) *files.File {
keyMaxViews = "max_views" keyMaxViews = "max_views"
keyExpiresOn = "exp" keyExpiresOn = "exp"
) )
var f files.File var f files.File
q := r.URL.Query() q := r.URL.Query()
if q.Has(keyMaxViews) { if q.Has(keyMaxViews) {
views, err := strconv.ParseUint(q.Get(keyMaxViews), 10, 64) views, err := strconv.ParseUint(q.Get(keyMaxViews), 10, 64) // nolint: gomnd
if err == nil { if err == nil {
f.MaxViews = uint(views) f.MaxViews = uint(views)
} }

View File

@ -39,31 +39,37 @@ func (s *HTTPServer) MiddlewareAccessLogger(next http.Handler) http.Handler {
next.ServeHTTP(ww, r) next.ServeHTTP(ww, r)
} }
return http.HandlerFunc(fn) return http.HandlerFunc(fn)
} }
func (s *HTTPServer) MiddlewareAuthentication(next http.Handler) http.Handler { func (s *HTTPServer) MiddlewareAuthentication(next http.Handler) http.Handler {
fn := func(w http.ResponseWriter, r *http.Request) { fn := func(w http.ResponseWriter, r *http.Request) {
reqID := middleware.GetReqID(r.Context()) reqID := middleware.GetReqID(r.Context())
header := r.Header.Get("Authorization") header := r.Header.Get("Authorization")
if header == "" { if header == "" {
s.Logger.Debugw("Request has no auth header.", "req_id", reqID) s.Logger.Debugw("Request has no auth header.", "req_id", reqID)
next.ServeHTTP(w, r) next.ServeHTTP(w, r)
return return
} }
splitHeader := strings.Split(header, "Bearer ") splitHeader := strings.Split(header, "Bearer ")
if len(splitHeader) != 2 { if len(splitHeader) != 2 { // nolint: gomnd
s.Logger.Debugw("Request has invalid token.", "req_id", reqID) s.Logger.Debugw("Request has invalid token.", "req_id", reqID)
next.ServeHTTP(w, r) next.ServeHTTP(w, r)
return return
} }
token := splitHeader[1] token := splitHeader[1]
claims, err := s.Auth.ValidateToken(token) claims, err := s.Auth.ValidateToken(token)
if err != nil { if err != nil {
s.Logger.Debugw("Request has invalid token.", "req_id", reqID) s.Logger.Debugw("Request has invalid token.", "req_id", reqID)
next.ServeHTTP(w, r) next.ServeHTTP(w, r)
return return
} }
@ -71,6 +77,7 @@ func (s *HTTPServer) MiddlewareAuthentication(next http.Handler) http.Handler {
ctx = context.WithValue(ctx, authCtxAuthLevel, claims.Role) ctx = context.WithValue(ctx, authCtxAuthLevel, claims.Role)
ctx = context.WithValue(ctx, authCtxClaims, claims) ctx = context.WithValue(ctx, authCtxClaims, claims)
withCtx := r.WithContext(ctx) withCtx := r.WithContext(ctx)
s.Logger.Debugw("Request is authenticated.", "req_id", reqID, "username", claims.Subject, "role", claims.Role) s.Logger.Debugw("Request is authenticated.", "req_id", reqID, "username", claims.Subject, "role", claims.Role)
next.ServeHTTP(w, withCtx) next.ServeHTTP(w, withCtx)
@ -84,10 +91,12 @@ func UsernameFromRequest(r *http.Request) (string, error) {
if rawUsername == nil { if rawUsername == nil {
return "", fmt.Errorf("no username") return "", fmt.Errorf("no username")
} }
username, ok := rawUsername.(string) username, ok := rawUsername.(string)
if !ok { if !ok {
return "", fmt.Errorf("no username") return "", fmt.Errorf("no username")
} }
return username, nil return username, nil
} }
@ -96,10 +105,12 @@ func RoleFromRequest(r *http.Request) (users.Role, error) {
if rawLevel == nil { if rawLevel == nil {
return users.RoleUnset, fmt.Errorf("no username") return users.RoleUnset, fmt.Errorf("no username")
} }
level, ok := rawLevel.(users.Role) level, ok := rawLevel.(users.Role)
if !ok { if !ok {
return users.RoleUnset, fmt.Errorf("no username") return users.RoleUnset, fmt.Errorf("no username")
} }
return level, nil return level, nil
} }
@ -108,9 +119,11 @@ func ClaimsFromRequest(r *http.Request) *gpaste.Claims {
if rawClaims == nil { if rawClaims == nil {
return nil return nil
} }
claims, ok := rawClaims.(*gpaste.Claims) claims, ok := rawClaims.(*gpaste.Claims)
if !ok { if !ok {
return nil return nil
} }
return claims return claims
} }

View File

@ -44,6 +44,7 @@ func (as *AuthService) Login(username, password string) (string, error) {
claims.Role = user.Role claims.Role = user.Role
token := jwt.NewWithClaims(jwt.GetSigningMethod("HS256"), claims) token := jwt.NewWithClaims(jwt.GetSigningMethod("HS256"), claims)
signed, err := token.SignedString(as.hmacSecret) signed, err := token.SignedString(as.hmacSecret)
if err != nil { if err != nil {
return "", err return "", err
@ -57,9 +58,11 @@ func (as *AuthService) ValidateToken(rawToken string) (*Claims, error) {
token, err := jwt.ParseWithClaims(rawToken, claims, func(t *jwt.Token) (interface{}, error) { token, err := jwt.ParseWithClaims(rawToken, claims, func(t *jwt.Token) (interface{}, error) {
return as.hmacSecret, nil return as.hmacSecret, nil
}) })
if err != nil { if err != nil {
return nil, err return nil, err
} }
if !token.Valid { if !token.Valid {
return nil, fmt.Errorf("invalid token") return nil, fmt.Errorf("invalid token")
} }

View File

@ -18,9 +18,11 @@ import (
"github.com/kirsle/configdir" "github.com/kirsle/configdir"
) )
const defaultTimeout = 10 * time.Second
type Client struct { type Client struct {
BaseURL string `json:"base_url"` BaseURL string `json:"baseUrl"`
AuthToken string `json:"auth_token"` AuthToken string `json:"authToken"`
httpClient http.Client httpClient http.Client
} }
@ -36,7 +38,9 @@ func (c *Client) WriteConfig() error {
if err != nil { if err != nil {
return err return err
} }
path := filepath.Join(dir, "client.json") path := filepath.Join(dir, "client.json")
f, err := os.Create(path) f, err := os.Create(path)
if err != nil { if err != nil {
return err return err
@ -49,6 +53,7 @@ func (c *Client) WriteConfig() error {
func (c *Client) LoadConfig() error { func (c *Client) LoadConfig() error {
dir := configdir.LocalCache("gpaste") dir := configdir.LocalCache("gpaste")
path := filepath.Join(dir, "client.json") path := filepath.Join(dir, "client.json")
f, err := os.Open(path) f, err := os.Open(path)
if err != nil { if err != nil {
return err return err
@ -66,7 +71,7 @@ func (c *Client) LoadConfigFromReader(r io.Reader) error {
func (c *Client) Login(ctx context.Context, username, password string) error { func (c *Client) Login(ctx context.Context, username, password string) error {
url := fmt.Sprintf("%s/api/login", c.BaseURL) url := fmt.Sprintf("%s/api/login", c.BaseURL)
// TODO: Change timeout // TODO: Change timeout
ctx, cancel := context.WithTimeout(ctx, 10*time.Second) ctx, cancel := context.WithTimeout(ctx, defaultTimeout)
defer cancel() defer cancel()
body := new(bytes.Buffer) body := new(bytes.Buffer)
@ -74,10 +79,12 @@ func (c *Client) Login(ctx context.Context, username, password string) error {
Username: username, Username: username,
Password: password, Password: password,
} }
encoder := json.NewEncoder(body) encoder := json.NewEncoder(body)
if err := encoder.Encode(&requestData); err != nil { if err := encoder.Encode(&requestData); err != nil {
return fmt.Errorf("error encoding response: %w", err) return fmt.Errorf("error encoding response: %w", err)
} }
req, err := http.NewRequestWithContext(ctx, http.MethodPost, url, body) req, err := http.NewRequestWithContext(ctx, http.MethodPost, url, body)
if err != nil { if err != nil {
return fmt.Errorf("error creating request: %w", err) return fmt.Errorf("error creating request: %w", err)
@ -99,6 +106,7 @@ func (c *Client) Login(ctx context.Context, username, password string) error {
if err := decoder.Decode(&responseData); err != nil { if err := decoder.Decode(&responseData); err != nil {
return fmt.Errorf("unable to parse response: %s", err) return fmt.Errorf("unable to parse response: %s", err)
} }
c.AuthToken = responseData.Token c.AuthToken = responseData.Token
return nil return nil
@ -112,6 +120,7 @@ func (c *Client) UserCreate(ctx context.Context, username, password string) erro
Username: username, Username: username,
Password: password, Password: password,
} }
encoder := json.NewEncoder(body) encoder := json.NewEncoder(body)
if err := encoder.Encode(requestData); err != nil { if err := encoder.Encode(requestData); err != nil {
return fmt.Errorf("error encoding response: %w", err) return fmt.Errorf("error encoding response: %w", err)
@ -164,9 +173,6 @@ func (c *Client) Upload(ctx context.Context, files ...*files.File) (*api.Respons
client := &http.Client{} client := &http.Client{}
// TODO: Change timeout // TODO: Change timeout
ctx, cancel := context.WithTimeout(ctx, 10*time.Minute)
defer cancel()
// TODO: Improve buffering // TODO: Improve buffering
buf := &bytes.Buffer{} buf := &bytes.Buffer{}
mw := multipart.NewWriter(buf) mw := multipart.NewWriter(buf)
@ -176,16 +182,21 @@ func (c *Client) Upload(ctx context.Context, files ...*files.File) (*api.Respons
if err != nil { if err != nil {
return nil, err return nil, err
} }
if _, err := io.Copy(fw, file.Body); err != nil { if _, err := io.Copy(fw, file.Body); err != nil {
return nil, err return nil, err
} }
file.Body.Close() file.Body.Close()
} }
mw.Close() mw.Close()
req, err := http.NewRequestWithContext(ctx, http.MethodPost, url, buf) req, err := http.NewRequestWithContext(ctx, http.MethodPost, url, buf)
if err != nil { if err != nil {
return nil, err return nil, err
} }
req.Header.Add("Content-Type", mw.FormDataContentType()) req.Header.Add("Content-Type", mw.FormDataContentType())
resp, err := client.Do(req) resp, err := client.Do(req)
@ -213,10 +224,12 @@ func (c *Client) Delete(ctx context.Context, id string) error {
} }
req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", c.AuthToken)) req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", c.AuthToken))
resp, err := c.httpClient.Do(req) resp, err := c.httpClient.Do(req)
if err != nil { if err != nil {
return fmt.Errorf("unable to perform request: %s", err) return fmt.Errorf("unable to perform request: %s", err)
} }
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK { if resp.StatusCode != http.StatusOK {
return fmt.Errorf("got non-ok response from server: %s", resp.Status) return fmt.Errorf("got non-ok response from server: %s", resp.Status)

View File

@ -166,7 +166,7 @@ func TestClient(t *testing.T) {
}) })
t.Run("Save", func(t *testing.T) { t.Run("Save", func(t *testing.T) {
c := client.Client{BaseURL: "http://example.org/gpaste", AuthToken: "tokenpls"} c := client.Client{BaseURL: "http://example.org/gpaste", AuthToken: "tokenpls"}
expectedConfig := "{\"base_url\":\"http://example.org/gpaste\",\"auth_token\":\"tokenpls\"}\n" expectedConfig := "{\"baseUrl\":\"http://example.org/gpaste\",\"authToken\":\"tokenpls\"}\n"
buf := new(bytes.Buffer) buf := new(bytes.Buffer)
err := c.WriteConfigToWriter(buf) err := c.WriteConfigToWriter(buf)
if err != nil { if err != nil {
@ -179,7 +179,7 @@ func TestClient(t *testing.T) {
}) })
t.Run("Load", func(t *testing.T) { t.Run("Load", func(t *testing.T) {
c := client.Client{} c := client.Client{}
config := "{\"base_url\":\"http://pasta.example.org\",\"auth_token\":\"tokenpls\"}\n" config := "{\"baseUrl\":\"http://pasta.example.org\",\"authToken\":\"tokenpls\"}\n"
expectedClient := client.Client{BaseURL: "http://pasta.example.org", AuthToken: "tokenpls"} expectedClient := client.Client{BaseURL: "http://pasta.example.org", AuthToken: "tokenpls"}
sr := strings.NewReader(config) sr := strings.NewReader(config)
if err := c.LoadConfigFromReader(sr); err != nil { if err := c.LoadConfigFromReader(sr); err != nil {

View File

@ -15,27 +15,34 @@ import (
"golang.org/x/term" "golang.org/x/term"
) )
const defaultTimeout = 10 * time.Second
func ActionUpload(c *cli.Context) error { func ActionUpload(c *cli.Context) error {
clnt := client.Client{ clnt := client.Client{
BaseURL: c.String("url"), BaseURL: c.String("url"),
} }
for _, arg := range c.Args().Slice() { for _, arg := range c.Args().Slice() {
f, err := os.Open(arg) f, err := os.Open(arg)
if err != nil { if err != nil {
return err return err
} }
defer f.Close() defer f.Close()
file := &files.File{ file := &files.File{
OriginalFilename: arg, OriginalFilename: arg,
Body: f, Body: f,
} }
resp, err := clnt.Upload(c.Context, file) resp, err := clnt.Upload(c.Context, file)
if err != nil { if err != nil {
errmsg := fmt.Sprintf("Error uploading file: %s", err) errmsg := fmt.Sprintf("Error uploading file: %s", err)
return cli.Exit(errmsg, 1) return cli.Exit(errmsg, 1)
} }
fmt.Printf("Uploaded file %s - %s", file.OriginalFilename, resp.Files[0].URL) fmt.Printf("Uploaded file %s - %s", file.OriginalFilename, resp.Files[0].URL)
} }
return nil return nil
} }
@ -43,15 +50,19 @@ func ActionDelete(c *cli.Context) error {
clnt := client.Client{ clnt := client.Client{
BaseURL: c.String("url"), BaseURL: c.String("url"),
} }
for _, arg := range c.Args().Slice() { for _, arg := range c.Args().Slice() {
ctx, cancel := context.WithTimeout(c.Context, 5*time.Second) ctx, cancel := context.WithTimeout(c.Context, defaultTimeout)
defer cancel() defer cancel()
if err := clnt.Delete(ctx, arg); err != nil { if err := clnt.Delete(ctx, arg); err != nil {
fmt.Printf("Error deleting file %s\n", arg) fmt.Printf("Error deleting file %s\n", arg)
fmt.Printf("%s\n", err) fmt.Printf("%s\n", err)
} }
fmt.Printf("Deleted %s\n", arg) fmt.Printf("Deleted %s\n", arg)
} }
return nil return nil
} }
@ -60,6 +71,7 @@ func ActionLogin(c *cli.Context) error {
if username == "" { if username == "" {
return cli.Exit("USERNAME not supplied.", 1) return cli.Exit("USERNAME not supplied.", 1)
} }
password, err := readPassword() password, err := readPassword()
if err != nil { if err != nil {
return fmt.Errorf("error reading password: %w", err) return fmt.Errorf("error reading password: %w", err)
@ -72,6 +84,7 @@ func ActionLogin(c *cli.Context) error {
errmsg := fmt.Sprintf("Error logging in: %s", err) errmsg := fmt.Sprintf("Error logging in: %s", err)
return cli.Exit(errmsg, 1) return cli.Exit(errmsg, 1)
} }
if err := clnt.WriteConfig(); err != nil { if err := clnt.WriteConfig(); err != nil {
errMsg := fmt.Sprintf("Failed to write config: %s", err) errMsg := fmt.Sprintf("Failed to write config: %s", err)
return cli.Exit(errMsg, 1) return cli.Exit(errMsg, 1)
@ -85,7 +98,9 @@ func ActionLogin(c *cli.Context) error {
func ActionUserCreate(c *cli.Context) error { func ActionUserCreate(c *cli.Context) error {
// TODO: Needs to supply auth token to actually work // TODO: Needs to supply auth token to actually work
fmt.Println("Need to be logged in to create user") fmt.Println("Need to be logged in to create user")
username := readString("Enter username: ") username := readString("Enter username: ")
password, err := readPassword() password, err := readPassword()
if err != nil { if err != nil {
return fmt.Errorf("error reading password: %w", err) return fmt.Errorf("error reading password: %w", err)
@ -94,7 +109,8 @@ func ActionUserCreate(c *cli.Context) error {
clnt := client.Client{ clnt := client.Client{
BaseURL: c.String("url"), BaseURL: c.String("url"),
} }
ctx, cancel := context.WithTimeout(c.Context, 10*time.Second)
ctx, cancel := context.WithTimeout(c.Context, defaultTimeout)
defer cancel() defer cancel()
if err := clnt.Login(ctx, username, password); err != nil { if err := clnt.Login(ctx, username, password); err != nil {
@ -103,7 +119,9 @@ func ActionUserCreate(c *cli.Context) error {
} }
fmt.Println("User to create:") fmt.Println("User to create:")
username = readString("Enter username: ") username = readString("Enter username: ")
password, err = readPassword() password, err = readPassword()
if err != nil { if err != nil {
return fmt.Errorf("error reading password: %w", err) return fmt.Errorf("error reading password: %w", err)
@ -121,20 +139,24 @@ func ActionUserCreate(c *cli.Context) error {
func readPassword() (string, error) { func readPassword() (string, error) {
fmt.Print("Enter Password: ") fmt.Print("Enter Password: ")
bytePassword, err := term.ReadPassword(int(syscall.Stdin)) bytePassword, err := term.ReadPassword(int(syscall.Stdin))
if err != nil { if err != nil {
return "", err return "", err
} }
password := string(bytePassword) password := string(bytePassword)
return strings.TrimSpace(password), nil return strings.TrimSpace(password), nil
} }
func readString(prompt string) string { func readString(prompt string) string {
fmt.Print(prompt) fmt.Print(prompt)
scanner := bufio.NewScanner(os.Stdin) scanner := bufio.NewScanner(os.Stdin)
for scanner.Scan() { for scanner.Scan() {
return scanner.Text() return scanner.Text()
} }
return "" return ""
} }

View File

@ -64,5 +64,5 @@ func main() {
}, },
} }
app.Run(os.Args) _ = app.Run(os.Args)
} }

View File

@ -22,8 +22,11 @@ func ActionServe(c *cli.Context) error {
configPath = c.String("config") configPath = c.String("config")
} }
var cfg *gpaste.ServerConfig var (
var r io.ReadCloser cfg *gpaste.ServerConfig
r io.ReadCloser
)
r, err := os.Open(configPath) r, err := os.Open(configPath)
if err != nil { if err != nil {
cfg = &gpaste.ServerConfig{ cfg = &gpaste.ServerConfig{
@ -52,8 +55,10 @@ func ActionServe(c *cli.Context) error {
// Setup contexts for clean shutdown // Setup contexts for clean shutdown
rootCtx, rootCancel := signal.NotifyContext(context.Background(), os.Interrupt) rootCtx, rootCancel := signal.NotifyContext(context.Background(), os.Interrupt)
defer rootCancel() defer rootCancel()
httpCtx, httpCancel := context.WithCancel(rootCtx) httpCtx, httpCancel := context.WithCancel(rootCtx)
defer httpCancel() defer httpCancel()
httpShutdownCtx, httpShutdownCancel := context.WithCancel(context.Background()) httpShutdownCtx, httpShutdownCancel := context.WithCancel(context.Background())
defer httpShutdownCancel() defer httpShutdownCancel()
@ -66,14 +71,18 @@ func ActionServe(c *cli.Context) error {
// Wait for cancel // Wait for cancel
go func() { go func() {
<-httpCtx.Done() <-httpCtx.Done()
timeoutCtx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
timeoutCtx, cancel := context.WithTimeout(context.Background(), 10*time.Second) // nolint: gomnd
defer cancel() defer cancel()
srv.Shutdown(timeoutCtx)
_ = srv.Shutdown(timeoutCtx)
}() }()
serverLogger.Infow("Starting HTTP server.", "addr", cfg.ListenAddr) serverLogger.Infow("Starting HTTP server.", "addr", cfg.ListenAddr)
if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed { if err := srv.ListenAndServe(); err != nil && err != http.ErrServerClosed {
serverLogger.Errorw("Error during shutdown.", "error", err) serverLogger.Errorw("Error during shutdown.", "error", err)
} }
serverLogger.Infow("HTTP server shutdown complete.", "addr", cfg.ListenAddr) serverLogger.Infow("HTTP server shutdown complete.", "addr", cfg.ListenAddr)
httpShutdownCancel() httpShutdownCancel()
}() }()

View File

@ -29,5 +29,5 @@ func main() {
Action: actions.ActionServe, Action: actions.ActionServe,
} }
app.Run(os.Args) _ = app.Run(os.Args)
} }

View File

@ -33,6 +33,7 @@ func ServerConfigFromReader(r io.Reader) (*ServerConfig, error) {
FS: &ServerStoreFSStoreConfig{}, FS: &ServerStoreFSStoreConfig{},
}, },
} }
if err := decoder.Decode(&c); err != nil { if err := decoder.Decode(&c); err != nil {
return nil, fmt.Errorf("error decoding server config: %w", err) return nil, fmt.Errorf("error decoding server config: %w", err)
} }

View File

@ -7,13 +7,13 @@ import (
type File struct { type File struct {
ID string `json:"id"` ID string `json:"id"`
OriginalFilename string `json:"original_filename"` OriginalFilename string `json:"originalFilename"`
MaxViews uint `json:"max_views"` MaxViews uint `json:"maxViews"`
ExpiresOn time.Time `json:"expires_on"` ExpiresOn time.Time `json:"expiresOn"`
Body io.ReadCloser Body io.ReadCloser
FileSize int64 `json:"file_size"` FileSize int64 `json:"fileSize"`
} }
type FileStore interface { type FileStore interface {

View File

@ -34,6 +34,7 @@ func (s *FSFileStore) Store(f *File) error {
} }
path := filepath.Join(s.dir, f.ID) path := filepath.Join(s.dir, f.ID)
dst, err := os.Create(path) dst, err := os.Create(path)
if err != nil { if err != nil {
return err return err
@ -44,12 +45,15 @@ func (s *FSFileStore) Store(f *File) error {
if err != nil { if err != nil {
return err return err
} }
s.metadata[f.ID] = metadata s.metadata[f.ID] = metadata
s.metadata[f.ID].FileSize = n s.metadata[f.ID].FileSize = n
if err := s.writeMetadata(); err != nil { if err := s.writeMetadata(); err != nil {
delete(s.metadata, f.ID) delete(s.metadata, f.ID)
return err return err
} }
return nil return nil
} }
@ -60,11 +64,14 @@ func (s *FSFileStore) Get(id string) (*File, error) {
} }
path := filepath.Join(s.dir, id) path := filepath.Join(s.dir, id)
f, err := os.Open(path) f, err := os.Open(path)
if err != nil { if err != nil {
return nil, err return nil, err
} }
metadata.Body = f metadata.Body = f
return metadata, nil return metadata, nil
} }
@ -73,20 +80,24 @@ func (s *FSFileStore) Delete(id string) error {
if err := os.Remove(path); err != nil { if err := os.Remove(path); err != nil {
return err return err
} }
delete(s.metadata, id) delete(s.metadata, id)
return nil return nil
} }
func (s *FSFileStore) List() ([]string, error) { func (s *FSFileStore) List() ([]string, error) {
var results []string results := make([]string, 0, len(s.metadata))
for k := range s.metadata { for k := range s.metadata {
results = append(results, k) results = append(results, k)
} }
return results, nil return results, nil
} }
func (s *FSFileStore) writeMetadata() error { func (s *FSFileStore) writeMetadata() error {
path := filepath.Join(s.dir, "metadata.json") path := filepath.Join(s.dir, "metadata.json")
f, err := os.Create(path) f, err := os.Create(path)
if err != nil { if err != nil {
return err return err
@ -97,11 +108,13 @@ func (s *FSFileStore) writeMetadata() error {
if err := encoder.Encode(s.metadata); err != nil { if err := encoder.Encode(s.metadata); err != nil {
return err return err
} }
return nil return nil
} }
func (s *FSFileStore) readMetadata() error { func (s *FSFileStore) readMetadata() error {
path := filepath.Join(s.dir, "metadata.json") path := filepath.Join(s.dir, "metadata.json")
f, err := os.Open(path) f, err := os.Open(path)
if err != nil { if err != nil {
// TODO: Handle errors better // TODO: Handle errors better
@ -113,5 +126,6 @@ func (s *FSFileStore) readMetadata() error {
if err := decoder.Decode(&s.metadata); err != nil { if err := decoder.Decode(&s.metadata); err != nil {
return err return err
} }
return nil return nil
} }

View File

@ -29,7 +29,6 @@ func NewMemoryFileStore() *MemoryFileStore {
} }
func (s *MemoryFileStore) Store(f *File) error { func (s *MemoryFileStore) Store(f *File) error {
data := &fileData{ data := &fileData{
ID: f.ID, ID: f.ID,
MaxViews: f.MaxViews, MaxViews: f.MaxViews,
@ -45,6 +44,7 @@ func (s *MemoryFileStore) Store(f *File) error {
defer s.lock.Unlock() defer s.lock.Unlock()
s.data[f.ID] = data s.data[f.ID] = data
return err return err
} }
@ -56,6 +56,7 @@ func (s *MemoryFileStore) Get(id string) (*File, error) {
if !ok { if !ok {
return nil, fmt.Errorf("no such item") return nil, fmt.Errorf("no such item")
} }
f := &File{ f := &File{
ID: fd.ID, ID: fd.ID,
MaxViews: fd.MaxViews, MaxViews: fd.MaxViews,
@ -70,17 +71,21 @@ func (s *MemoryFileStore) Get(id string) (*File, error) {
func (s *MemoryFileStore) Delete(id string) error { func (s *MemoryFileStore) Delete(id string) error {
s.lock.Lock() s.lock.Lock()
defer s.lock.Unlock() defer s.lock.Unlock()
delete(s.data, id) delete(s.data, id)
return nil return nil
} }
func (s *MemoryFileStore) List() ([]string, error) { func (s *MemoryFileStore) List() ([]string, error) {
var ids []string ids := make([]string, 0, len(s.data))
s.lock.RLock() s.lock.RLock()
defer s.lock.RUnlock() defer s.lock.RUnlock()
for id := range s.data { for id := range s.data {
ids = append(ids, id) ids = append(ids, id)
} }
return ids, nil return ids, nil
} }

View File

@ -16,7 +16,7 @@ const (
type User struct { type User struct {
Username string `json:"username"` Username string `json:"username"`
HashedPassword []byte `json:"hashed_password"` HashedPassword []byte `json:"hashedPassword"`
Role Role `json:"role"` Role Role `json:"role"`
} }
@ -38,6 +38,8 @@ func (u *User) SetPassword(password string) error {
if err != nil { if err != nil {
return err return err
} }
u.HashedPassword = hashed u.HashedPassword = hashed
return nil return nil
} }

View File

@ -15,7 +15,7 @@ type BoltUserStore struct {
} }
func NewBoltUserStore(path string) (*BoltUserStore, error) { func NewBoltUserStore(path string) (*BoltUserStore, error) {
db, err := bbolt.Open(path, 0666, nil) db, err := bbolt.Open(path, 0666, nil) // nolint: gomnd
if err != nil { if err != nil {
return nil, err return nil, err
} }
@ -36,6 +36,7 @@ func (s *BoltUserStore) Close() error {
func (s *BoltUserStore) Get(username string) (*User, error) { func (s *BoltUserStore) Get(username string) (*User, error) {
var user User var user User
err := s.db.View(func(tx *bbolt.Tx) error { err := s.db.View(func(tx *bbolt.Tx) error {
bkt := tx.Bucket(keyUsers) bkt := tx.Bucket(keyUsers)
rawUser := bkt.Get([]byte(username)) rawUser := bkt.Get([]byte(username))
@ -44,9 +45,11 @@ func (s *BoltUserStore) Get(username string) (*User, error) {
} }
return nil return nil
}) })
if err != nil { if err != nil {
return nil, err return nil, err
} }
return &user, nil return &user, nil
} }
@ -86,5 +89,6 @@ func (s *BoltUserStore) List() ([]string, error) {
if err != nil { if err != nil {
return nil, err return nil, err
} }
return ids, nil return ids, nil
} }

View File

@ -18,7 +18,9 @@ func NewMemoryUserStore() *MemoryUserStore {
func (s *MemoryUserStore) Get(username string) (*User, error) { func (s *MemoryUserStore) Get(username string) (*User, error) {
s.lock.Lock() s.lock.Lock()
defer s.lock.Unlock() defer s.lock.Unlock()
user, ok := s.users[username] user, ok := s.users[username]
if !ok { if !ok {
return nil, fmt.Errorf("no such user: %s", username) return nil, fmt.Errorf("no such user: %s", username)
} }
@ -29,21 +31,25 @@ func (s *MemoryUserStore) Get(username string) (*User, error) {
func (s *MemoryUserStore) Store(user *User) error { func (s *MemoryUserStore) Store(user *User) error {
s.lock.Lock() s.lock.Lock()
defer s.lock.Unlock() defer s.lock.Unlock()
s.users[user.Username] = user s.users[user.Username] = user
return nil return nil
} }
func (s *MemoryUserStore) Delete(username string) error { func (s *MemoryUserStore) Delete(username string) error {
s.lock.Lock() s.lock.Lock()
defer s.lock.Unlock() defer s.lock.Unlock()
delete(s.users, username) delete(s.users, username)
return nil return nil
} }
func (s *MemoryUserStore) List() ([]string, error) { func (s *MemoryUserStore) List() ([]string, error) {
s.lock.Lock() s.lock.Lock()
defer s.lock.Unlock() defer s.lock.Unlock()
var ids []string ids := make([]string, 0, len(s.users))
for k := range s.users { for k := range s.users {
ids = append(ids, k) ids = append(ids, k)
} }