Commit 440c90be authored by Brian Tiger Chow's avatar Brian Tiger Chow Committed by Juan Batiz-Benet

extract unpin

@mappum
parent 92d20e42
package commands package commands
import ( import (
"errors"
"fmt" "fmt"
cmds "github.com/jbenet/go-ipfs/commands" cmds "github.com/jbenet/go-ipfs/commands"
"github.com/jbenet/go-ipfs/core" "github.com/jbenet/go-ipfs/core"
"github.com/jbenet/go-ipfs/core/commands2/internal"
"github.com/jbenet/go-ipfs/merkledag" "github.com/jbenet/go-ipfs/merkledag"
) )
...@@ -37,7 +37,7 @@ var pinCmd = &cmds.Command{ ...@@ -37,7 +37,7 @@ var pinCmd = &cmds.Command{
} }
}*/ }*/
paths, err := toStrings(req.Arguments()) paths, err := internal.ToStrings(req.Arguments())
if err != nil { if err != nil {
res.SetError(err, cmds.ErrNormal) res.SetError(err, cmds.ErrNormal)
return return
...@@ -66,28 +66,13 @@ var unpinCmd = &cmds.Command{ ...@@ -66,28 +66,13 @@ var unpinCmd = &cmds.Command{
opt, _ := req.Option("recursive") opt, _ := req.Option("recursive")
recursive, _ := opt.(bool) // false if cast fails. recursive, _ := opt.(bool) // false if cast fails.
for _, arg := range req.Arguments() { paths, err := internal.ToStrings(req.Arguments())
path, ok := arg.(string) if err != nil {
if !ok { res.SetError(err, cmds.ErrNormal)
res.SetError(errors.New("cast error"), cmds.ErrNormal) return
return
}
dagnode, err := n.Resolver.ResolvePath(path)
if err != nil {
res.SetError(fmt.Errorf("pin error: %v", err), cmds.ErrNormal)
return
}
k, _ := dagnode.Key()
err = n.Pinning.Unpin(k, recursive)
if err != nil {
res.SetError(fmt.Errorf("pin: %v", err), cmds.ErrNormal)
return
}
} }
err := n.Pinning.Flush() _, err = unpin(n, paths, recursive)
if err != nil { if err != nil {
res.SetError(err, cmds.ErrNormal) res.SetError(err, cmds.ErrNormal)
} }
...@@ -100,7 +85,6 @@ func pin(n *core.IpfsNode, paths []string, recursive bool) ([]*merkledag.Node, e ...@@ -100,7 +85,6 @@ func pin(n *core.IpfsNode, paths []string, recursive bool) ([]*merkledag.Node, e
dagnodes := make([]*merkledag.Node, 0) dagnodes := make([]*merkledag.Node, 0)
for _, path := range paths { for _, path := range paths {
dagnode, err := n.Resolver.ResolvePath(path) dagnode, err := n.Resolver.ResolvePath(path)
if err != nil { if err != nil {
return nil, fmt.Errorf("pin error: %v", err) return nil, fmt.Errorf("pin error: %v", err)
...@@ -122,3 +106,29 @@ func pin(n *core.IpfsNode, paths []string, recursive bool) ([]*merkledag.Node, e ...@@ -122,3 +106,29 @@ func pin(n *core.IpfsNode, paths []string, recursive bool) ([]*merkledag.Node, e
return dagnodes, nil return dagnodes, nil
} }
func unpin(n *core.IpfsNode, paths []string, recursive bool) ([]*merkledag.Node, error) {
dagnodes := make([]*merkledag.Node, 0)
for _, path := range paths {
dagnode, err := n.Resolver.ResolvePath(path)
if err != nil {
return nil, err
}
dagnodes = append(dagnodes, dagnode)
}
for _, dagnode := range dagnodes {
k, _ := dagnode.Key()
err := n.Pinning.Unpin(k, recursive)
if err != nil {
return nil, err
}
}
err := n.Pinning.Flush()
if err != nil {
return nil, err
}
return dagnodes, nil
}
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment