From a810bd53ae8bb863f298f758dd41f98c6d0e96a6 Mon Sep 17 00:00:00 2001
From: Daniel Rojas <github@danielrojas.net>
Date: Sat, 2 Oct 2021 11:40:36 +0200
Subject: [PATCH] Add `wv_cli.py`, add Click requirement, change entry point

---
 requirements.txt      |  1 +
 setup.py              |  5 ++++-
 src/wireviz/wv_cli.py | 29 +++++++++++++++++++++++++++++
 3 files changed, 34 insertions(+), 1 deletion(-)
 create mode 100644 src/wireviz/wv_cli.py

diff --git a/requirements.txt b/requirements.txt
index 36f048c..07564c3 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,3 +1,4 @@
+click
 graphviz
 pillow
 pyyaml
diff --git a/setup.py b/setup.py
index d339386..f7971e0 100644
--- a/setup.py
+++ b/setup.py
@@ -17,6 +17,7 @@ setup(
     long_description=open(README_PATH).read(),
     long_description_content_type='text/markdown',
     install_requires=[
+        'click',
         'pyyaml',
         'pillow',
         'graphviz',
@@ -27,7 +28,9 @@ setup(
     package_dir={'': 'src'},
     packages=find_packages('src'),
     entry_points={
-        'console_scripts': ['wireviz=wireviz.wireviz:main'],
+        'console_scripts': [
+            'wireviz=wireviz.wv_cli:main',
+            ],
         },
     classifiers=[
         'Development Status :: 4 - Beta',
diff --git a/src/wireviz/wv_cli.py b/src/wireviz/wv_cli.py
new file mode 100644
index 0000000..f38ad97
--- /dev/null
+++ b/src/wireviz/wv_cli.py
@@ -0,0 +1,29 @@
+import os
+import sys
+
+import click
+
+import wireviz.wireviz
+
+@click.command()
+@click.argument('filepath', nargs=-1)
+@click.option('-p', '--prepend', default=None)
+@click.option('-o', '--output', default='hpst')
+@click.option('-V', '--version', is_flag=True, default=False)
+def main(filepath, prepend, output, version):
+    print('WireViz!')
+    # get list of files
+    try:
+        _ = iter(filepath)
+    except TypeError:
+        filepaths = [filepath]
+    else:
+        filepaths = list(filepath)
+
+    for f in filepaths:
+        print(f)
+    print()
+
+if __name__ == '__main__':
+    sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..'))
+    wireviz()