-from __future__ import unicode_literals
+#!/usr/bin/env python3
+
+# Allow direct execution
+import os
+import sys
+
+sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
+
+import contextlib
+import subprocess
+import sys
from datetime import datetime
-# import urllib.request
-# response = urllib.request.urlopen('https://blackjack4494.github.io/youtube-dlc/update/LATEST_VERSION')
-# old_version = response.read().decode('utf-8')
+from devscripts.utils import read_version, write_file
+
+
+def get_new_version(revision):
+ version = datetime.utcnow().strftime('%Y.%m.%d')
+
+ if revision:
+ assert revision.isdigit(), 'Revision must be a number'
+ else:
+ old_version = read_version().split('.')
+ if version.split('.') == old_version[:3]:
+ revision = str(int((old_version + [0])[3]) + 1)
+
+ return f'{version}.{revision}' if revision else version
+
+
+def get_git_head():
+ with contextlib.suppress(Exception):
+ sp = subprocess.Popen(['git', 'rev-parse', '--short', 'HEAD'], stdout=subprocess.PIPE)
+ return sp.communicate()[0].decode().strip() or None
-exec(compile(open('youtube_dlc/version.py').read(), 'youtube_dlc/version.py', 'exec'))
-old_version = locals()['__version__']
-old_version_list = old_version.split(".", 4)
+VERSION = get_new_version((sys.argv + [''])[1])
+GIT_HEAD = get_git_head()
-old_ver = '.'.join(old_version_list[:3])
-old_rev = old_version_list[3] if len(old_version_list) > 3 else ''
+VERSION_FILE = f'''\
+# Autogenerated by devscripts/update-version.py
-ver = datetime.utcnow().strftime("%Y.%m.%d")
-rev = str(int(old_rev or 0) + 1) if old_ver == ver else ''
+__version__ = {VERSION!r}
-VERSION = '.'.join((ver, rev)) if rev else ver
-# VERSION_LIST = [(int(v) for v in ver.split(".") + [rev or 0])]
+RELEASE_GIT_HEAD = {GIT_HEAD!r}
-print('::set-output name=ytdlc_version::' + VERSION)
+VARIANT = None
-file_version_py = open('youtube_dlc/version.py', 'rt')
-data = file_version_py.read()
-data = data.replace(old_version, VERSION)
-file_version_py.close()
+UPDATE_HINT = None
+'''
-file_version_py = open('youtube_dlc/version.py', 'wt')
-file_version_py.write(data)
-file_version_py.close()
+write_file('yt_dlp/version.py', VERSION_FILE)
+print(f'::set-output name=ytdlp_version::{VERSION}')
+print(f'\nVersion = {VERSION}, Git HEAD = {GIT_HEAD}')