From a741583a118c018c245fe425ad50d7953f7a6c1a Mon Sep 17 00:00:00 2001
From: Earl Hathaway <github@earlh.com>
Date: Fri, 7 Apr 2017 17:03:50 -0700
Subject: [PATCH] parse / distribute arguments for a single command

---
 dvc/runtime.py | 10 +++++--
 dvc2.py        | 71 ++++++++++++++++++++++++++++++++++++++++++++++++--
 2 files changed, 77 insertions(+), 4 deletions(-)

diff --git a/dvc/runtime.py b/dvc/runtime.py
index 3c04557a1..04aa0d66d 100644
--- a/dvc/runtime.py
+++ b/dvc/runtime.py
@@ -16,7 +16,13 @@ class Runtime(object):
         return os.path.realpath(os.path.join(git_dir, Runtime.CONFIG))
 
     @staticmethod
-    def run(cmd_class, parse_config=True):
+    def run(cmd_class, parse_config=True, args_start_loc=1):
+        """
+
+        Arguments:
+            args_start_loc (int): where the arguments this command should use start
+        """
+
         try:
             runtime_git = GitWrapper()
 
@@ -25,7 +31,7 @@ class Runtime(object):
             else:
                 runtime_config = ConfigI()
 
-            args = sys.argv[1:]
+            args = sys.argv[args_start_loc:]
 
             instance = cmd_class(Settings(args, runtime_git, runtime_config))
             sys.exit(instance.run())
diff --git a/dvc2.py b/dvc2.py
index cd2503002..6b906f73b 100755
--- a/dvc2.py
+++ b/dvc2.py
@@ -5,8 +5,75 @@ main entry point / argument parsing for dvc
 NB: for ${REASONS}, cannot be named dvc.py.  trust me.
 """
 
+import sys
+
 from dvc import *
+from dvc.runtime import Runtime
+from dvc.command.data_import import CmdDataImport
+from dvc.command.remove import CmdDataRemove
+from dvc.command.run import CmdRun
+from dvc.command.repro import CmdRepro
+from dvc.command.data_sync import CmdDataSync
+from dvc.command.data_import import CmdDataImport
+
+
+
+def print_usage():
+    usage = ('',
+            'These are common dvc commands:',
+            '',
+            'start a working area',
+            '    init    initialize dvc control over a dir (should already be a git dir)',
+            '    run TODO',
+            '    sync TODO',
+            '',
+            'synchronize data between remote and local',
+            '    data sync    TODO',
+            '    data remove  TODO',
+            '    data import  TODO',
+            '')
+    print('\n'.join(usage))
 
 if __name__ == '__main__':
-    print('hello, world')
-print('hello world, outside main')
+    cmds = ['init', 'run', 'sync', 'data', 'data-sync', 'data-remove', 'data-import', 'cloud', \
+            'cloud', 'cloud-run', 'cloud-instance-create', 'cloud-instance-remove', 'cloud-instance-describe']
+    cmds_expand = {'data':  ['sync', 'remove', 'import'],
+                   'cloud': ['run', 'instance-create', 'instance-remove', 'instance-describe']
+                  }
+
+    if len(sys.argv) < 2 or sys.argv[1] not in cmds:
+        print_usage()
+        sys.exit(-1)
+
+    cmd = sys.argv[1]
+    subcmd = None
+    if cmd in cmds_expand:
+        if (len(sys.argv) < 3 or sys.argv[2] not in cmds_expand[cmd]):
+            print('for command %s, eligible actions are %s' % (cmd, cmds_expand[cmd]))
+            print_usage()
+            sys.exit(-1)
+        else:
+            subcmd = sys.argv[2]
+
+    if cmd == 'init':
+        Runtime.run(CmdRun, args_start_loc=2)
+    elif cmd == 'run':
+        Runtime.run(CmdDataImport, args_start_loc=2)
+    elif cmd == 'repro':
+        Runtime.run(CmdRepro, args_start_loc=2)
+    elif cmd == 'data-sync' or (cmd == 'data' and subcmd == 'sync'):
+        Runtime.run(CmdDataSync, args_start_loc=2)
+    elif cmd == 'data-import' or (cmd == 'data' and subcmd == 'import'):
+        Runtime.run(CmdDataImport, args_start_loc=2)
+    elif cmd == 'cloud-run' or (cmd == 'cloud' and subcmd == 'run'):
+        print('cloud-run unimplemented')
+    elif cmd == 'cloud-instance-create' or (cmd == 'cloud' and subcmd == 'instance-create'):
+        print('cloud-instance-create unimplemented')
+    elif cmd == 'clould-instance-remove' or (cmd == 'cloud' and subcmd == 'instance-remove'):
+        print('cloud-instance-remove unimplemented')
+    elif cmd == 'cloud-instance-describe' or (cmd == 'cloud' and subcmd == 'instance-describe'):
+        print('cloud-instance-describe unimplemented')
+    else:
+        print('Unimplemented or unrecognized command.  sorry!')
+        print_usage()
+        sys.exit(-1)