1
0
Fork 0
mirror of https://github.com/vbatts/go-mtree.git synced 2025-02-16 14:07:52 +00:00

walk: when directory is root, use "."

Signed-off-by: Vincent Batts <vbatts@hashbangbash.com>
This commit is contained in:
Vincent Batts 2016-04-13 13:55:34 -04:00
parent 9f0a9fd6c0
commit 410c0d60c1

View file

@ -36,6 +36,7 @@ func Walk(root string, exlcudes []ExcludeFunc, keywords []string) (*DirectoryHie
} }
} }
entryPathName := filepath.Base(path)
if info.IsDir() { if info.IsDir() {
creator.DH.Entries = append(creator.DH.Entries, Entry{ creator.DH.Entries = append(creator.DH.Entries, Entry{
Type: BlankType, Type: BlankType,
@ -46,13 +47,14 @@ func Walk(root string, exlcudes []ExcludeFunc, keywords []string) (*DirectoryHie
if creator.curDir != nil { if creator.curDir != nil {
creator.DH.Entries = append(creator.DH.Entries, Entry{ creator.DH.Entries = append(creator.DH.Entries, Entry{
Pos: len(creator.DH.Entries), Pos: len(creator.DH.Entries),
Raw: "# " + filepath.Join(creator.curDir.Path(), filepath.Base(path)), Raw: "# " + filepath.Join(creator.curDir.Path(), entryPathName),
Type: CommentType, Type: CommentType,
}) })
} else { } else {
entryPathName = "."
creator.DH.Entries = append(creator.DH.Entries, Entry{ creator.DH.Entries = append(creator.DH.Entries, Entry{
Pos: len(creator.DH.Entries), Pos: len(creator.DH.Entries),
Raw: "# " + filepath.Base(path), Raw: "# .",
Type: CommentType, Type: CommentType,
}) })
} }
@ -105,7 +107,7 @@ func Walk(root string, exlcudes []ExcludeFunc, keywords []string) (*DirectoryHie
} }
e := Entry{ e := Entry{
Name: filepath.Base(path), Name: entryPathName,
Pos: len(creator.DH.Entries), Pos: len(creator.DH.Entries),
Type: RelativeType, Type: RelativeType,
Set: creator.curSet, Set: creator.curSet,