3dcb3e9073
* Second attempt at preventing zombies * Ensure that the pipes are closed in ssh.go * Ensure that a cancellable context is passed up in cmd/* http requests * Make cmd.fail return properly so defers are obeyed * Ensure that something is sent to stdout in case of blocks here Signed-off-by: Andrew Thornton <art27@cantab.net> * placate lint Signed-off-by: Andrew Thornton <art27@cantab.net> * placate lint 2 Signed-off-by: Andrew Thornton <art27@cantab.net> * placate lint 3 Signed-off-by: Andrew Thornton <art27@cantab.net> * fixup Signed-off-by: Andrew Thornton <art27@cantab.net> * Apply suggestions from code review Co-authored-by: 6543 <6543@obermui.de> Co-authored-by: Lauris BH <lauris@nix.lv>
110 lines
2.9 KiB
Go
110 lines
2.9 KiB
Go
// Copyright 2019 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 private
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
"net/url"
|
|
|
|
"code.gitea.io/gitea/models"
|
|
"code.gitea.io/gitea/modules/setting"
|
|
jsoniter "github.com/json-iterator/go"
|
|
)
|
|
|
|
// KeyAndOwner is the response from ServNoCommand
|
|
type KeyAndOwner struct {
|
|
Key *models.PublicKey `json:"key"`
|
|
Owner *models.User `json:"user"`
|
|
}
|
|
|
|
// ServNoCommand returns information about the provided key
|
|
func ServNoCommand(ctx context.Context, keyID int64) (*models.PublicKey, *models.User, error) {
|
|
reqURL := setting.LocalURL + fmt.Sprintf("api/internal/serv/none/%d",
|
|
keyID)
|
|
resp, err := newInternalRequest(ctx, reqURL, "GET").Response()
|
|
if err != nil {
|
|
return nil, nil, err
|
|
}
|
|
defer resp.Body.Close()
|
|
if resp.StatusCode != http.StatusOK {
|
|
return nil, nil, fmt.Errorf("%s", decodeJSONError(resp).Err)
|
|
}
|
|
|
|
var keyAndOwner KeyAndOwner
|
|
json := jsoniter.ConfigCompatibleWithStandardLibrary
|
|
if err := json.NewDecoder(resp.Body).Decode(&keyAndOwner); err != nil {
|
|
return nil, nil, err
|
|
}
|
|
return keyAndOwner.Key, keyAndOwner.Owner, nil
|
|
}
|
|
|
|
// ServCommandResults are the results of a call to the private route serv
|
|
type ServCommandResults struct {
|
|
IsWiki bool
|
|
IsDeployKey bool
|
|
KeyID int64
|
|
KeyName string
|
|
UserName string
|
|
UserEmail string
|
|
UserID int64
|
|
OwnerName string
|
|
RepoName string
|
|
RepoID int64
|
|
}
|
|
|
|
// ErrServCommand is an error returned from ServCommmand.
|
|
type ErrServCommand struct {
|
|
Results ServCommandResults
|
|
Err string
|
|
StatusCode int
|
|
}
|
|
|
|
func (err ErrServCommand) Error() string {
|
|
return err.Err
|
|
}
|
|
|
|
// IsErrServCommand checks if an error is a ErrServCommand.
|
|
func IsErrServCommand(err error) bool {
|
|
_, ok := err.(ErrServCommand)
|
|
return ok
|
|
}
|
|
|
|
// ServCommand preps for a serv call
|
|
func ServCommand(ctx context.Context, keyID int64, ownerName, repoName string, mode models.AccessMode, verbs ...string) (*ServCommandResults, error) {
|
|
reqURL := setting.LocalURL + fmt.Sprintf("api/internal/serv/command/%d/%s/%s?mode=%d",
|
|
keyID,
|
|
url.PathEscape(ownerName),
|
|
url.PathEscape(repoName),
|
|
mode)
|
|
for _, verb := range verbs {
|
|
if verb != "" {
|
|
reqURL += fmt.Sprintf("&verb=%s", url.QueryEscape(verb))
|
|
}
|
|
}
|
|
|
|
resp, err := newInternalRequest(ctx, reqURL, "GET").Response()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
defer resp.Body.Close()
|
|
json := jsoniter.ConfigCompatibleWithStandardLibrary
|
|
if resp.StatusCode != http.StatusOK {
|
|
var errServCommand ErrServCommand
|
|
if err := json.NewDecoder(resp.Body).Decode(&errServCommand); err != nil {
|
|
return nil, err
|
|
}
|
|
errServCommand.StatusCode = resp.StatusCode
|
|
return nil, errServCommand
|
|
}
|
|
var results ServCommandResults
|
|
if err := json.NewDecoder(resp.Body).Decode(&results); err != nil {
|
|
return nil, err
|
|
}
|
|
return &results, nil
|
|
|
|
}
|