feat(fsnode): add type helper

mfs_test.go

@@ -1182,3 +1182,31 @@ func TestTruncateAndWrite(t *testing.T) {
 		}
 	}
 }
+
+func TestFSNodeType(t *testing.T) {
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+	ds, rt := setupRoot(ctx, t)
+
+	// check for IsDir
+	nd := dag.NodeWithData(ft.FolderPBData())
+	di, err := NewDirectory(ctx, "test", nd, rt.GetDirectory(), ds)
+	if err != nil {
+		t.Fatal(err)
+	}
+	ret := IsDir(di)
+	if !ret {
+		t.Fatal("FSNode type should be dir, but not")
+	}
+
+	// check for IsFile
+	fnd := dag.NodeWithData(ft.FilePBData(nil, 0))
+	fi, err := NewFile("test", fnd, rt.GetDirectory(), ds)
+	if err != nil {
+		t.Fatal(err)
+	}
+	ret = IsFile(fi)
+	if !ret {
+		t.Fatal("FSNode type should be file, but not")
+	}
+}

system.go

@@ -48,6 +48,16 @@ type FSNode interface {
 	Type() NodeType
 }
 
+// IsDir checks whether the FSNode is dir type
+func IsDir(fsn FSNode) bool {
+	return fsn.Type() == TDir
+}
+
+// IsFile checks whether the FSNode is file type
+func IsFile(fsn FSNode) bool {
+	return fsn.Type() == TFile
+}
+
 // Root represents the root of a filesystem tree.
 type Root struct {