gpaste/client/client.go

package client

import (
	"bytes"
	"context"
	"encoding/json"
	"fmt"
	"io"
	"mime/multipart"
	"net/http"
	"time"

	"git.t-juice.club/torjus/gpaste/api"
	"git.t-juice.club/torjus/gpaste/files"
	"github.com/google/uuid"
)

type Client struct {
	BaseURL   string
	AuthToken string

	httpClient http.Client
}

func (c *Client) Login(ctx context.Context, username, password string) error {
	url := fmt.Sprintf("%s/api/login", c.BaseURL)
	// TODO: Change timeout
	ctx, cancel := context.WithTimeout(ctx, 10*time.Second)
	defer cancel()

	body := new(bytes.Buffer)
	requestData := api.RequestAPILogin{
		Username: username,
		Password: password,
	}
	encoder := json.NewEncoder(body)
	if err := encoder.Encode(&requestData); err != nil {
		return fmt.Errorf("error encoding response: %w", err)
	}
	req, err := http.NewRequestWithContext(ctx, http.MethodPost, url, body)
	if err != nil {
		return fmt.Errorf("error creating request: %w", err)
	}

	resp, err := c.httpClient.Do(req)
	if err != nil {
		return fmt.Errorf("unable to perform request: %s", err)
	}
	defer resp.Body.Close()

	if resp.StatusCode != http.StatusOK {
		return fmt.Errorf("got non-ok response from server: %s", resp.Status)
	}

	var responseData api.ResponseAPILogin

	decoder := json.NewDecoder(resp.Body)
	if err := decoder.Decode(&responseData); err != nil {
		return fmt.Errorf("unable to parse response: %s", err)
	}
	c.AuthToken = responseData.Token

	return nil
}

func (c *Client) UserCreate(ctx context.Context, username, password string) error {
	url := fmt.Sprintf("%s/api/user", c.BaseURL)
	body := new(bytes.Buffer)

	requestData := &api.RequestAPIUserCreate{
		Username: username,
		Password: password,
	}
	encoder := json.NewEncoder(body)
	if err := encoder.Encode(requestData); err != nil {
		return fmt.Errorf("error encoding response: %w", err)
	}

	req, err := http.NewRequestWithContext(ctx, http.MethodPost, url, body)
	if err != nil {
		return fmt.Errorf("error creating request: %w", err)
	}

	req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", c.AuthToken))

	resp, err := c.httpClient.Do(req)
	if err != nil {
		return fmt.Errorf("unable to perform request: %s", err)
	}
	defer resp.Body.Close()

	if resp.StatusCode != http.StatusAccepted {
		return fmt.Errorf("got non-ok response from server: %s", resp.Status)
	}

	return nil
}

func (c *Client) Download(ctx context.Context, id string) (io.ReadCloser, error) {
	url := fmt.Sprintf("%s/api/file/%s", c.BaseURL, id)

	req, err := http.NewRequestWithContext(ctx, http.MethodGet, url, nil)
	if err != nil {
		return nil, fmt.Errorf("error creating request: %w", err)
	}

	req.Header.Set("Authorization", fmt.Sprintf("Bearer %s", c.AuthToken))

	resp, err := c.httpClient.Do(req)
	if err != nil {
		return nil, fmt.Errorf("unable to perform request: %s", err)
	}

	if resp.StatusCode != http.StatusOK {
		return nil, fmt.Errorf("got non-ok response from server: %s", resp.Status)
	}

	return resp.Body, nil
}

func (c *Client) Upload(ctx context.Context, files ...*files.File) ([]api.ResponseAPIFilePost, error) {
	url := fmt.Sprintf("%s/api/file", c.BaseURL)
	client := &http.Client{}

	// TODO: Change timeout
	ctx, cancel := context.WithTimeout(ctx, 10*time.Minute)
	defer cancel()

	// TODO: Improve buffering
	buf := &bytes.Buffer{}
	mw := multipart.NewWriter(buf)

	for _, file := range files {
		fw, err := mw.CreateFormFile(uuid.Must(uuid.NewRandom()).String(), file.OriginalFilename)
		if err != nil {
			return nil, err
		}
		if _, err := io.Copy(fw, file.Body); err != nil {
			return nil, err
		}
		file.Body.Close()
	}
	mw.Close()
	req, err := http.NewRequestWithContext(ctx, http.MethodPost, url, buf)
	if err != nil {
		return nil, err
	}
	req.Header.Add("Content-Type", mw.FormDataContentType())

	resp, err := client.Do(req)
	if err != nil {
		return nil, err
	}
	defer resp.Body.Close()

	var expectedResp []api.ResponseAPIFilePost

	decoder := json.NewDecoder(resp.Body)
	if err := decoder.Decode(&expectedResp); err != nil {
		return nil, fmt.Errorf("error decoding response: %w", err)
	}

	return expectedResp, nil
}