client.go

package client

import (
	"context"
	"fmt"

	"cloud.google.com/go/storage"
	"google.golang.org/api/option"
)

type Writer interface {
	WriteData([]byte) error
	Close() error
}

type GCSWriter struct {
	writer *storage.Writer
}

// NewWriter creates a new GCS writer.
func NewWriter(ctx context.Context, serviceAccountJSON []byte, bucketname, filepath string) (*GCSWriter, error) {
	client, err := storage.NewClient(ctx, option.WithCredentialsJSON(serviceAccountJSON))
	if err != nil {
		return nil, fmt.Errorf("create client: %w", err)
	}

	writer := client.Bucket(bucketname).Object(filepath).NewWriter(ctx)

	return &GCSWriter{
		writer: writer,
	}, nil
}

func (c *GCSWriter) WriteData(data []byte) error {
	if _, err := c.writer.Write(data); err != nil {
		return fmt.Errorf("write data to an object: %w", err)
	}

	return nil
}

func (c *GCSWriter) Close() error {
	return c.writer.Close()
}