91 lines
2.3 KiB
Go
91 lines
2.3 KiB
Go
// VulcanizeDB
|
|
// Copyright © 2019 Vulcanize
|
|
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Affero General Public License for more details.
|
|
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package ipfs
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
|
|
"github.com/ipfs/go-blockservice"
|
|
"github.com/ipfs/go-ipfs/core"
|
|
"github.com/ipfs/go-ipfs/plugin/loader"
|
|
"github.com/ipfs/go-ipfs/repo/fsrepo"
|
|
ipld "github.com/ipfs/go-ipld-format"
|
|
)
|
|
|
|
// InitIPFSPlugins is used to initialized IPFS plugins before creating a new IPFS node
|
|
// This should only be called once
|
|
func InitIPFSPlugins() error {
|
|
logrus.Debug("initializing IPFS plugins")
|
|
l, err := loader.NewPluginLoader("")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
err = l.Initialize()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return l.Inject()
|
|
}
|
|
|
|
// InitIPFSBlockService is used to configure and return a BlockService using an ipfs repo path (e.g. ~/.ipfs)
|
|
func InitIPFSBlockService(ipfsPath string) (blockservice.BlockService, error) {
|
|
logrus.Debug("initializing IPFS block service interface")
|
|
r, openErr := fsrepo.Open(ipfsPath)
|
|
if openErr != nil {
|
|
return nil, openErr
|
|
}
|
|
ctx := context.Background()
|
|
cfg := &core.BuildCfg{
|
|
Online: false,
|
|
Repo: r,
|
|
}
|
|
ipfsNode, newNodeErr := core.NewNode(ctx, cfg)
|
|
if newNodeErr != nil {
|
|
return nil, newNodeErr
|
|
}
|
|
return ipfsNode.Blocks, nil
|
|
}
|
|
|
|
type IPFS struct {
|
|
n *core.IpfsNode
|
|
ctx context.Context
|
|
}
|
|
|
|
func (ipfs IPFS) Add(node ipld.Node) error {
|
|
return ipfs.n.DAG.Add(ipfs.n.Context(), node)
|
|
}
|
|
|
|
func InitIPFSNode(repoPath string) (*IPFS, error) {
|
|
logrus.Debug("initializing IPFS node interface")
|
|
r, err := fsrepo.Open(repoPath)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
ctx := context.Background()
|
|
cfg := &core.BuildCfg{
|
|
Online: false,
|
|
Repo: r,
|
|
}
|
|
ipfsNode, err := core.NewNode(ctx, cfg)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &IPFS{n: ipfsNode, ctx: ctx}, nil
|
|
}
|