coder/codersdk/files.go

package codersdk

import (
	"context"
	"encoding/json"
	"fmt"
	"net/http"
	"net/url"

	"github.com/coder/coder/coderd"
)

const (
	ContentTypeTar = "application/x-tar"
)

func (c *Client) UploadFile(ctx context.Context, contentType string, content []byte) (coderd.UploadFileResponse, error) {
	res, err := c.request(ctx, http.MethodPost, "/api/v2/upload", content, func(r *http.Request) {
		r.Header.Set("Content-Type", contentType)
	})
	if err != nil {
		return coderd.UploadFileResponse{}, err
	}
	defer res.Body.Close()
	if res.StatusCode != http.StatusCreated && res.StatusCode != http.StatusOK {
		return coderd.UploadFileResponse{}, readBodyAsError(res)
	}
	var resp coderd.UploadFileResponse
	return resp, json.NewDecoder(res.Body).Decode(&resp)
}

// DownloadURL returns the download URL for the specified asset
func (c *Client) DownloadURL(asset string) (*url.URL, error) {
	return c.URL.Parse(fmt.Sprintf("/api/v2/downloads/%s", asset))
}