go-tty.Close() on exit. Should fix #21

pry/pry.go

@@ -21,7 +21,12 @@ var (
 	tty *gotty.TTY
 )
 
-func init() {
+// Pry does nothing. It only exists so running code without go-pry doesn't throw an error.
+func Pry(v ...interface{}) {
+}
+
+// Apply drops into a pry shell in the location required.
+func Apply(scope *Scope) {
 	if runtime.GOOS == "windows" {
 		out = colorable.NewColorableStdout()
 	}
@@ -30,23 +35,16 @@ func init() {
 	if err != nil {
 		panic(err)
 	}
-}
+	defer tty.Close()
 
-// Pry does nothing. It only exists so running code without go-pry doesn't throw an error.
-func Pry(v ...interface{}) {
-}
-
-// Apply drops into a pry shell in the location required.
-func Apply(scope *Scope) {
 	if scope.Files == nil {
 		scope.Files = map[string]*ast.File{}
 	}
 
 	_, filePathRaw, lineNum, _ := runtime.Caller(1)
 	filePath := filepath.Dir(filePathRaw) + "/." + filepath.Base(filePathRaw) + "pry"
 
-	err := scope.ConfigureTypes(filePath, lineNum)
-	if err != nil {
+	if err := scope.ConfigureTypes(filePath, lineNum); err != nil {
 		panic(err)
 	}
 
@@ -189,7 +187,7 @@ func Apply(scope *Scope) {
 			index = 0
 		case 4: // Ctrl-D
 			fmt.Fprintln(out)
-			os.Exit(0)
+			return
 		}
 	}
 }