common/utils/ipfs/ipfs.go

package ipfs

import (
	"context"
	"fmt"
	"io"

	shell "github.com/ipfs/go-ipfs-api"
	myio "gitlink.org.cn/cloudream/common/utils/io"
)

type IPFS struct {
	shell *shell.Shell
}

func NewIPFS(cfg *Config) (*IPFS, error) {
	ipfsAddr := fmt.Sprintf("localhost:%d", cfg.Port)
	sh := shell.NewShell(ipfsAddr)

	// 检测连通性
	if !sh.IsUp() {
		return nil, fmt.Errorf("cannot connect to %s", ipfsAddr)
	}

	return &IPFS{
		shell: sh,
	}, nil
}

func (fs *IPFS) IsUp() bool {
	return fs.shell.IsUp()
}

func (fs *IPFS) CreateFile() (myio.PromiseWriteCloser[string], error) {
	pr, pw := io.Pipe()

	ipfsWriter := ipfsWriter{
		writer:   pw,
		finished: make(chan any, 1),
	}

	go func() {
		hash, err := fs.shell.Add(pr)
		ipfsWriter.finishErr = err
		ipfsWriter.fileHash = hash
		close(ipfsWriter.finished)
		pr.CloseWithError(err)
	}()

	return &ipfsWriter, nil
}

func (fs *IPFS) OpenRead(hash string) (io.ReadCloser, error) {
	return fs.shell.Cat(hash)
}

func (fs *IPFS) Pin(hash string) error {
	return fs.shell.Pin(hash)
}

func (fs *IPFS) Unpin(hash string) error {
	return fs.shell.Unpin(hash)
}

func (fs *IPFS) GetPinnedFiles() (map[string]shell.PinInfo, error) {
	return fs.shell.PinsOfType(context.Background(), shell.RecursivePin)
}

type ipfsWriter struct {
	writer    *io.PipeWriter
	finished  chan any
	finishErr error
	fileHash  string
}

func (p *ipfsWriter) Write(data []byte) (n int, err error) {
	return p.writer.Write(data)
}

// 设置一个error中断写入
func (w *ipfsWriter) Abort(err error) {
	w.writer.CloseWithError(err)
}

// Finish 结束写入，并获得返回值（文件哈希值）
func (w *ipfsWriter) Finish() (string, error) {
	w.writer.CloseWithError(io.EOF)

	<-w.finished

	return w.fileHash, w.finishErr
}

func IPFSRemoteDeamonDetector() { //探测本地IPFS Deamon与目的地IPFS Deamon的连接状态

}