pic/weed.go

package pic

import (
	"fmt"
	"io"
	"net/http"
	"os"
	"strings"
	"time"

	//"log"
	"github.com/linxGnu/goseaweedfs"
)

var sw *goseaweedfs.Seaweed

func init() {
	masterURL := "http://192.168.0.105:9333"
	filer := []string{}
	sw, _ = goseaweedfs.NewSeaweed(
		masterURL, // master server
		filer,     // нету у меня filer
		8096,      // размер chunk
		&http.Client{Timeout: 5 * time.Minute})

	_, err := sw.Status()
	if err != nil {
		panic(err)
	}
}

// загружает файл filename на weed и возвращает url загруженного файла
func upload(filename string) (string, error) {
	fid, err := sw.Submit(filename, "", "")
	if err != nil {
		return "", err
	}
	return fid.FileURL, nil
}

// удаляет файл из weed
func delete(url string) error {
	fid, err := getFID(url)
	if err != nil {
		return err
	}

	err = sw.DeleteFile(fid, nil)
	return err
}

// загружает файл из weed в filename
func download(url string, filename string) error {
	fid, err := getFID(url)
	if err != nil {
		return err
	}

	_, err = sw.Download(fid, nil, func(r io.Reader) (err error) {
		data, err := io.ReadAll(r)
		if err != nil {
			return err
		}

		file, err := os.OpenFile(filename, os.O_CREATE, 0666)
		if err != nil {
			return err
		}
		_, err = file.Write(data)
		if err != nil {
			return err
		}

		err = file.Close()
		return err
	})

	return err
}

func getFID(url string) (string, error) {
	items := strings.Split(url, "/")
	if len(items) < 2 {
		return "", fmt.Errorf("url должен быть в формате: <http://address/fid.ext>")
	}
	return items[len(items)-1], nil
}

func getFilename(url string) (string, error) {
	fn, err := getFID(url)
	if err != nil {
		return "", err
	}

	items := strings.Split(fn, ",")
	return items[len(items)-1], nil
}