rewrite add command to use dagwriter, moved a pinner into the dagwriter for inline pinning

pin.go

@@ -20,6 +20,14 @@ var recursePinDatastoreKey = ds.NewKey("/local/pins/recursive/keys")
 var directPinDatastoreKey = ds.NewKey("/local/pins/direct/keys")
 var indirectPinDatastoreKey = ds.NewKey("/local/pins/indirect/keys")
 
+type PinMode int
+
+const (
+	Recursive PinMode = iota
+	Direct
+	Indirect
+)
+
 type Pinner interface {
 	IsPinned(util.Key) bool
 	Pin(*mdag.Node, bool) error
@@ -27,6 +35,13 @@ type Pinner interface {
 	Flush() error
 }
 
+// ManualPinner is for manually editing the pin structure
+// Use with care
+type ManualPinner interface {
+	PinWithMode(util.Key, PinMode)
+	Pinner
+}
+
 type pinner struct {
 	lock       sync.RWMutex
 	recursePin set.BlockSet
@@ -228,3 +243,14 @@ func loadSet(d ds.Datastore, k ds.Key, val interface{}) error {
 	}
 	return json.Unmarshal(bf, val)
 }
+
+func (p *pinner) PinWithMode(k util.Key, mode PinMode) {
+	switch mode {
+	case Recursive:
+		p.recursePin.AddBlock(k)
+	case Direct:
+		p.directPin.AddBlock(k)
+	case Indirect:
+		p.indirPin.Increment(k)
+	}
+}