Handle relative paths, resolves #1

Author: KyleBanks

Makefile

@@ -11,7 +11,7 @@ include github.com/KyleBanks/make/git/precommit
 
 # Runs a number of depth commands as examples of what's possible.
 example: | install
-	depth github.com/KyleBanks/depth/cmd/depth strings
+	depth github.com/KyleBanks/depth/cmd/depth strings ./
 
 	depth -internal strings 
 
@@ -22,4 +22,8 @@ example: | install
 	depth -test -internal strings
 
 	depth -test -internal -max 3 strings
+
+	depth .
+
+	depth ./cmd/depth
 .PHONY: example

README.md

@@ -20,7 +20,7 @@ go get github.com/KyleBanks/depth/cmd/depth
 
 ### Command-Line
 
-Simply execute `depth` with one or more package names to visualize:
+Simply execute `depth` with one or more package names to visualize. You can use the fully qualified import path of the package, like so:
 
 ```sh
 $ depth github.com/KyleBanks/depth/cmd/depth
@@ -40,6 +40,14 @@ github.com/KyleBanks/depth/cmd/depth
     └ strings
 ```
 
+Or you can use a relative path, for example:
+
+```sh
+$ depth .
+$ depth ./cmd/depth
+$ depth ../
+```
+
 You can also use `depth` on the Go standard library:
 
 ```sh
@@ -54,7 +62,7 @@ strings
 Visualizing multiple packages at a time is supported by simply naming the packages you'd like to visualize:
 
 ```sh
-$ depth strings github.com/KyleBanks/depth
+$ depth strings github.com/KyleBanks/depth 
 strings
   ├ errors
   ├ io
@@ -178,10 +186,25 @@ if err != nil {
 log.Printf("'%v' has %v dependencies.", t.Root.Name, len(t.Root.Deps)) 
 ```
 
+For additional customization, simply set the appropriate flags on the `Tree` before resolving:
+
+```go
+import "github.com/KyleBanks/depth"
+
+t := depth.Tree {
+  ResolveInternal: true,
+  ResolveTest: true,
+  MaxDepth: 10,
+}
+
+
+err := t.Resolve("strings")
+```
+
 ## Author
 
 `depth` was developed by [Kyle Banks](https://twitter.com/kylewbanks).
 
 ## License
 
-`depth` is available under [MIT](./LICENSE)
+`depth` is available under the [MIT](./LICENSE) license.

depth.go

@@ -3,23 +3,34 @@
 //
 // For example, the dependencies of the stdlib `strings` package can be resolved like so:
 //
-// 		```go
-// 		import "github.com/KyleBanks/depth"
+// 	import "github.com/KyleBanks/depth"
 //
-//		var t depth.Tree
-// 		err := t.Resolve("strings")
-// 		if err != nil {
-//     		log.Fatal(err)
-// 		}
+//	var t depth.Tree
+// 	err := t.Resolve("strings")
+// 	if err != nil {
+// 		log.Fatal(err)
+// 	}
 //
-// 		// Output: "strings has 4 dependencies."
-// 		log.Printf("%v has %v dependencies.", t.Root.Name, len(t.Root.Deps))
-// 		```
+// 	// Output: "strings has 4 dependencies."
+// 	log.Printf("%v has %v dependencies.", t.Root.Name, len(t.Root.Deps))
+//
+// For additional customization, simply set the appropriate flags on the `Tree` before resolving:
+//
+// 	import "github.com/KyleBanks/depth"
+//
+// 	t := depth.Tree {
+//  	ResolveInternal: true,
+//   	ResolveTest: true,
+//   	MaxDepth: 10,
+// 	}
+//
+// err := t.Resolve("strings")
 package depth
 
 import (
 	"errors"
 	"go/build"
+	"os"
 )
 
 // ErrRootPkgNotResolved is returned when the root Pkg of the Tree cannot be resolved,
@@ -48,7 +59,16 @@ type Tree struct {
 // Resolve recursively finds all dependencies for the root Pkg name provided,
 // and the packages it depends on.
 func (t *Tree) Resolve(name string) error {
-	t.Root = &Pkg{Name: name, Tree: t}
+	pwd, err := os.Getwd()
+	if err != nil {
+		return err
+	}
+
+	t.Root = &Pkg{
+		Name:   name,
+		Tree:   t,
+		SrcDir: pwd,
+	}
 
 	// Reset the import cache each time to ensure a reused Tree doesn't
 	// reuse the same cache.

depth_test.go

@@ -24,10 +24,13 @@ func TestTree_Resolve(t *testing.T) {
 	if err := tr.Resolve("strings"); err != nil {
 		t.Fatal(err)
 	}
+
 	if tr.Root == nil || tr.Root.Name != "strings" {
 		t.Fatalf("Unexpected Root, expected=%v, got=%v", "strings", tr.Root)
 	} else if len(tr.Root.Deps) == 0 {
 		t.Fatal("Expected positive number of Deps")
+	} else if len(tr.Root.SrcDir) == 0 {
+		t.Fatal("Expected SrcDir to be populated")
 	}
 
 	// Reuse the same tree and the same package to ensure that the internal pkg cache

pkg.go

@@ -45,6 +45,9 @@ func (p *Pkg) Resolve(i Importer) {
 		return
 	}
 
+	// Update the name with the fully qualified import path.
+	p.Name = pkg.ImportPath
+
 	// If this is an internal dependency, we may need to skip it.
 	if pkg.Goroot {
 		p.Internal = true