agecem-org/apirequest/document.go

package apirequest

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

	"codeberg.org/vlbeaudoin/voki"
	"codeberg.org/vlbeaudoin/voki/request"
	"git.agecem.com/agecem/agecem-org/apiresponse"
)

var _ request.Requester[apiresponse.V1DocumentsPOST] = V1DocumentsPOST{}

type V1DocumentsPOST struct {
	Data struct {
		Documents []*multipart.FileHeader `json:"documents"`
	}
	Params struct {
		Bucket string `json:"bucket"`
	}
}

func NewV1DocumentsPOST(bucket string, documents ...*multipart.FileHeader) (request V1DocumentsPOST, err error) {
	if bucket == "" {
		err = fmt.Errorf("NewV1DocumentsPOST requires non-nil bucket name")
		return
	}

	request.Params.Bucket = bucket

	if documents == nil {
		err = fmt.Errorf("NewV1DocumentsPOST requires non-nil documents")
		return
	}

	for _, document := range documents {
		if document == nil {
			err = fmt.Errorf("NewV1DocumentsPOST requires non-nil documents")
			return
		}
	}

	request.Data.Documents = documents

	return
}

func (request V1DocumentsPOST) Complete() bool {
	if request.Data.Documents == nil {
		return false
	}
	for _, document := range request.Data.Documents {
		if document == nil {
			return false
		}
	}

	return request.Params.Bucket != ""
}

func (request V1DocumentsPOST) Request(v *voki.Voki) (response apiresponse.V1DocumentsPOST, err error) {
	if !request.Complete() {
		err = fmt.Errorf("Incomplete V1DocumentsPOST request")
		return
	}

	var buf bytes.Buffer
	if err = json.NewEncoder(&buf).Encode(request.Data); err != nil {
		return
	}

	return response, v.UnmarshalIfComplete(http.MethodPost, fmt.Sprintf("/v1/bucket/%s/many", request.Params.Bucket), &buf, true, &response)
}

var _ request.Requester[apiresponse.V1DocumentPOST] = V1DocumentPOST{}

type V1DocumentPOST struct {
	Data struct {
		Document *multipart.FileHeader `document`
	}
	Params struct {
		Bucket string `json:"bucket"`
	}
}

func NewV1DocumentPOST(bucket string, document *multipart.FileHeader) (request V1DocumentPOST, err error) {
	if bucket == "" {
		err = fmt.Errorf("NewV1DocumentPOST requires non-nil bucket name")
		return
	}

	request.Params.Bucket = bucket

	if document == nil {
		err = fmt.Errorf("NewV1DocumentPOST requires non-nil document")
		return
	}

	request.Data.Document = document

	return
}

func (request V1DocumentPOST) Complete() bool {
	return request.Params.Bucket != "" && request.Data.Document != nil
}

func (request V1DocumentPOST) Request(v *voki.Voki) (response apiresponse.V1DocumentPOST, err error) {
	if !request.Complete() {
		err = fmt.Errorf("Incomplete V1DocumentPOST request")
		return
	}

	var buf bytes.Buffer
	if err = json.NewEncoder(&buf).Encode(request.Data); err != nil {
		return
	}

	return response, v.UnmarshalIfComplete(http.MethodPost, fmt.Sprintf("/v1/bucket/%s", request.Params.Bucket), &buf, true, &response)
}

var _ request.Requester[apiresponse.V1DocumentDELETE] = V1DocumentDELETE{}

type V1DocumentDELETE struct {
	Params struct {
		Bucket   string `json:"bucket"`
		Document string `json:"document"`
	}
}

func NewV1DocumentDELETE(bucket, document string) (request V1DocumentDELETE, err error) {
	if bucket == "" {
		err = fmt.Errorf("NewV1DocumentDELETE requires non-nil bucket name")
		return
	}

	request.Params.Bucket = bucket

	if document == "" {
		err = fmt.Errorf("NewV1DocumentDELETE requires non-nil document name")
		return
	}

	request.Params.Document = document

	return
}

func (request V1DocumentDELETE) Complete() bool {
	return request.Params.Bucket != "" && request.Params.Document != ""
}

func (request V1DocumentDELETE) Request(v *voki.Voki) (response apiresponse.V1DocumentDELETE, err error) {
	if !request.Complete() {
		err = fmt.Errorf("Incomplete V1DocumentDELETE request")
		return
	}

	return response, v.UnmarshalIfComplete(http.MethodDelete, fmt.Sprintf("/v1/bucket/%s/%s", request.Params.Bucket, request.Params.Document), nil, true, &response)
}