gitlabhq/gitlab-shell

package gitauditevent

import (
    "context"
    "fmt"

    pb "gitlab.com/gitlab-org/gitaly/v16/proto/go/gitalypb"
    "gitlab.com/gitlab-org/gitlab-shell/v14/client"
    "gitlab.com/gitlab-org/gitlab-shell/v14/internal/command/commandargs"
    "gitlab.com/gitlab-org/gitlab-shell/v14/internal/config"
    "gitlab.com/gitlab-org/gitlab-shell/v14/internal/gitlabnet"
)

const uri = "/api/v4/internal/shellhorse/git_audit_event"

type Client struct {
    config *config.Config
    client *client.GitlabNetClient
}

func NewClient(config *config.Config) (*Client, error) {
    client, err := gitlabnet.GetClient(config)
    if err != nil {
        return nil, fmt.Errorf("error creating http client: %w", err)
    }

    return &Client{config: config, client: client}, nil
}

type Request struct {
    Action        commandargs.CommandType           `json:"action"`
    Protocol      string                            `json:"protocol"`
    Repo          string                            `json:"gl_repository"`
    Username      string                            `json:"username"`
    PackfileStats *pb.PackfileNegotiationStatistics `json:"packfile_stats,omitempty"`
}

func (c *Client) Audit(ctx context.Context, username string, action commandargs.CommandType, repo string, packfileStats *pb.PackfileNegotiationStatistics) error {
    request := &Request{
        Action:        action,
        Repo:          repo,
        Protocol:      "ssh",
        Username:      username,
        PackfileStats: packfileStats,
    }

    response, err := c.client.Post(ctx, uri, request)
    if err != nil {
        return err
    }
    defer response.Body.Close()
    return nil
}