From e0e8fc867a739158c27ea567d2bd5878500fb8df Mon Sep 17 00:00:00 2001
From: Jorropo <jorropo.pgm@gmail.com>
Date: Fri, 2 Jun 2023 18:16:46 +0200
Subject: [PATCH] pinning: fix pin ls --type=indirect

---
 core/commands/pin/pin.go | 15 +++++++++------
 core/coreapi/pin.go      | 23 +++++++++++++++--------
 2 files changed, 24 insertions(+), 14 deletions(-)

diff --git a/core/commands/pin/pin.go b/core/commands/pin/pin.go
index 626c536dbd2a..1465a6a4d9b9 100644
--- a/core/commands/pin/pin.go
+++ b/core/commands/pin/pin.go
@@ -342,14 +342,17 @@ Example:
 		}
 
 		// For backward compatibility, we accumulate the pins in the same output type as before.
-		emit := res.Emit
+		var emit func(*PinLsOutputWrapper) error
 		lgcList := map[string]PinLsType{}
 		if !stream {
-			emit = func(v interface{}) error {
-				obj := v.(*PinLsOutputWrapper)
-				lgcList[obj.PinLsObject.Cid] = PinLsType{Type: obj.PinLsObject.Type}
+			emit = func(v *PinLsOutputWrapper) error {
+				lgcList[v.PinLsObject.Cid] = PinLsType{Type: v.PinLsObject.Type}
 				return nil
 			}
+		} else {
+			emit = func(v *PinLsOutputWrapper) error {
+				return res.Emit(v)
+			}
 		}
 
 		if len(req.Arguments) > 0 {
@@ -432,7 +435,7 @@ type PinLsObject struct {
 	Type string `json:",omitempty"`
 }
 
-func pinLsKeys(req *cmds.Request, typeStr string, api coreiface.CoreAPI, emit func(value interface{}) error) error {
+func pinLsKeys(req *cmds.Request, typeStr string, api coreiface.CoreAPI, emit func(value *PinLsOutputWrapper) error) error {
 	enc, err := cmdenv.GetCidEncoder(req)
 	if err != nil {
 		return err
@@ -484,7 +487,7 @@ func pinLsKeys(req *cmds.Request, typeStr string, api coreiface.CoreAPI, emit fu
 	return nil
 }
 
-func pinLsAll(req *cmds.Request, typeStr string, api coreiface.CoreAPI, emit func(value interface{}) error) error {
+func pinLsAll(req *cmds.Request, typeStr string, api coreiface.CoreAPI, emit func(value *PinLsOutputWrapper) error) error {
 	enc, err := cmdenv.GetCidEncoder(req)
 	if err != nil {
 		return err
diff --git a/core/coreapi/pin.go b/core/coreapi/pin.go
index 98be600b2e4d..0559ef73cd92 100644
--- a/core/coreapi/pin.go
+++ b/core/coreapi/pin.go
@@ -358,13 +358,27 @@ func (api *PinAPI) pinLsAll(ctx context.Context, typeStr string) <-chan coreifac
 					return
 				}
 				emittedSet.Add(streamedCid.C)
+				rkeys = append(rkeys, streamedCid.C)
 			}
 
 			walkingSet := cid.NewSet()
 			for _, k := range rkeys {
 				err = merkledag.Walk(
 					ctx, merkledag.GetLinksWithDAG(api.dag), k,
-					walkingSet.Visit,
+					func(c cid.Cid) bool {
+						if !walkingSet.Visit(c) {
+							return false
+						}
+						if emittedSet.Has(c) {
+							return true // skipped
+						}
+						err := AddToResultKeys(c, "indirect")
+						if err != nil {
+							out <- &pinInfo{err: err}
+							return false
+						}
+						return true
+					},
 					merkledag.SkipRoot(), merkledag.Concurrent(),
 				)
 				if err != nil {
@@ -372,13 +386,6 @@ func (api *PinAPI) pinLsAll(ctx context.Context, typeStr string) <-chan coreifac
 					return
 				}
 			}
-			err = emittedSet.ForEach(func(c cid.Cid) error {
-				return AddToResultKeys(c, "indirect")
-			})
-			if err != nil {
-				out <- &pinInfo{err: err}
-				return
-			}
 		}
 	}()