1 files changed, 52 insertions, 9 deletions
diff --git a/poezio/args.py b/poezio/args.py
index d0005d82..3907fc88 100644
--- a/poezio/args.py
+++ b/poezio/args.py
@@ -1,10 +1,16 @@
 """
 Module related to the argument parsing
-
-There is a fallback to the deprecated optparse if argparse is not found
 """
+import pkg_resources
+import stat
+import sys
+from argparse import ArgumentParser, SUPPRESS, Namespace
 from pathlib import Path
-from argparse import ArgumentParser, SUPPRESS
+from shutil import copy2
+from typing import Tuple
+
+from poezio.version import __version__
+from poezio import xdg
 
 
 def parse_args(CONFIG_PATH: Path):
@@ -33,11 +39,48 @@ def parse_args(CONFIG_PATH: Path):
         help="The config file you want to use",
         metavar="CONFIG_FILE")
     parser.add_argument(
-        "-v",
-        "--version",
-        dest="version",
+        '-v',
+        '--version',
+        action='version',
+        version='Poezio v%s' % __version__,
+    )
+    parser.add_argument(
+        "--custom-version",
+        dest="custom_version",
         help=SUPPRESS,
         metavar="VERSION",
-        default="0.13-dev")
-    options = parser.parse_args()
-    return options
+        default=__version__
+    )
+    return parser.parse_args()
+
+
+def run_cmdline_args() -> Tuple[Namespace, bool]:
+    "Parse the command line arguments"
+    options = parse_args(xdg.CONFIG_HOME)
+    firstrun = False
+
+    # Copy a default file if none exists
+    if not options.filename.is_file():
+        try:
+            options.filename.parent.mkdir(parents=True, exist_ok=True)
+        except OSError as e:
+            sys.stderr.write(
+                'Poezio was unable to create the config directory: %s\n' % e)
+            sys.exit(1)
+        default = Path(__file__).parent / '..' / 'data' / 'default_config.cfg'
+        other = Path(
+            pkg_resources.resource_filename('poezio', 'default_config.cfg'))
+        if default.is_file():
+            copy2(str(default), str(options.filename))
+        elif other.is_file():
+            copy2(str(other), str(options.filename))
+
+        # Inside the nixstore and possibly other distributions, the reference
+        # file is readonly, so is the copy.
+        # Make it writable by the user who just created it.
+        if options.filename.exists():
+            options.filename.chmod(options.filename.stat().st_mode
+                                   | stat.S_IWUSR)
+        firstrun = True
+
+    return (options, firstrun)