Move form_builder into internal pkg. (#311)

* Move form_uilder into internal pkg.

* Fix import of audio.go

* Reorganize.

* Fix import.

* Fix

---------

Co-authored-by: JoyShi <joy.shi@sap.com>

internal/form_builder.go

@@ -0,0 +1,49 @@
+package openai
+
+import (
+	"io"
+	"mime/multipart"
+	"os"
+)
+
+type FormBuilder interface {
+	CreateFormFile(fieldname string, file *os.File) error
+	WriteField(fieldname, value string) error
+	Close() error
+	FormDataContentType() string
+}
+
+type DefaultFormBuilder struct {
+	writer *multipart.Writer
+}
+
+func NewFormBuilder(body io.Writer) *DefaultFormBuilder {
+	return &DefaultFormBuilder{
+		writer: multipart.NewWriter(body),
+	}
+}
+
+func (fb *DefaultFormBuilder) CreateFormFile(fieldname string, file *os.File) error {
+	fieldWriter, err := fb.writer.CreateFormFile(fieldname, file.Name())
+	if err != nil {
+		return err
+	}
+
+	_, err = io.Copy(fieldWriter, file)
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+func (fb *DefaultFormBuilder) WriteField(fieldname, value string) error {
+	return fb.writer.WriteField(fieldname, value)
+}
+
+func (fb *DefaultFormBuilder) Close() error {
+	return fb.writer.Close()
+}
+
+func (fb *DefaultFormBuilder) FormDataContentType() string {
+	return fb.writer.FormDataContentType()
+}

internal/form_builder_test.go

@@ -0,0 +1,54 @@
+package openai //nolint:testpackage // testing private field
+
+import (
+	"github.com/sashabaranov/go-openai/internal/test"
+	"github.com/sashabaranov/go-openai/internal/test/checks"
+
+	"bytes"
+	"errors"
+	"os"
+	"testing"
+)
+
+type failingWriter struct {
+}
+
+var errMockFailingWriterError = errors.New("mock writer failed")
+
+func (*failingWriter) Write([]byte) (int, error) {
+	return 0, errMockFailingWriterError
+}
+
+func TestFormBuilderWithFailingWriter(t *testing.T) {
+	dir, cleanup := test.CreateTestDirectory(t)
+	defer cleanup()
+
+	file, err := os.CreateTemp(dir, "")
+	if err != nil {
+		t.Errorf("Error creating tmp file: %v", err)
+	}
+	defer file.Close()
+	defer os.Remove(file.Name())
+
+	builder := NewFormBuilder(&failingWriter{})
+	err = builder.CreateFormFile("file", file)
+	checks.ErrorIs(t, err, errMockFailingWriterError, "formbuilder should return error if writer fails")
+}
+
+func TestFormBuilderWithClosedFile(t *testing.T) {
+	dir, cleanup := test.CreateTestDirectory(t)
+	defer cleanup()
+
+	file, err := os.CreateTemp(dir, "")
+	if err != nil {
+		t.Errorf("Error creating tmp file: %v", err)
+	}
+	file.Close()
+	defer os.Remove(file.Name())
+
+	body := &bytes.Buffer{}
+	builder := NewFormBuilder(body)
+	err = builder.CreateFormFile("file", file)
+	checks.HasError(t, err, "formbuilder should return error if file is closed")
+	checks.ErrorIs(t, err, os.ErrClosed, "formbuilder should return error if file is closed")
+}