Rev 4: Use getcwd() if no path is supplied. in http://bazaar.launchpad.net/~jameinel/my_nick

[John Arbash Meinel]

At http://bazaar.launchpad.net/~jameinel/my_nick

------------------------------------------------------------
revno: 4
revision-id: john at arbash-meinel.com-20100521214831-l6ak8dphymf3vgvy
parent: john at arbash-meinel.com-20100520181406-ytz3511wx8hz8soy
committer: John Arbash Meinel <john at arbash-meinel.com>
branch nick: my_nick
timestamp: Fri 2010-05-21 16:48:31 -0500
message:
  Use getcwd() if no path is supplied.
-------------- next part --------------
=== modified file 'my_nick.cpp'
--- a/my_nick.cpp	2010-05-20 18:14:06 +0000
+++ b/my_nick.cpp	2010-05-21 21:48:31 +0000
@@ -2,6 +2,8 @@
 /* A simple version that probes the FS to find what branch we're in.
  */
 
+#include <direct.h>
+#include <stdlib.h>
 #include <iostream>
 #include <fstream>
 #include <string>
@@ -170,14 +172,26 @@
     return path_to_short(path);
 }
 
+string
+get_current_path()
+{
+    char buf[_MAX_PATH];
+    getcwd(buf, _MAX_PATH);
+    return string(buf);
+}
+
 int
 main(int argc, char **argv)
 {
-    if (argc != 2) {
-        cout << "<invalid request>" << endl;
+    string path;
+    if (argc == 1) {
+        path = get_current_path();
+    } else if (argc == 2) {
+        path = string(argv[1]);
+    } else {
+        cout << "<invalid>" << endl;
         return 2;
     }
-    string path(argv[1]);
     path = find_containing_bzrdir(path);
     if (path.empty()) {
         return 0;