adding a proper bash-completion generation

2 files changed, 36 insertions, 0 deletions

diff --git a/devscripts/bash_completion.py b/devscripts/bash_completion.py
new file mode 100644
index 000000000..1cbfa8bec
--- /dev/null
+++ b/devscripts/bash_completion.py
@@ -0,0 +1,22 @@
+#!/usr/bin/env python2
+import youtube_dl
+
+BASH_COMPLETION_FILE = "youtube-dl.bash_completion"
+BASH_COMPLETION_TEMPLATE = "devscripts/bash_completion.template"
+
+def build_completion(opt_parser):
+    opts_flag = []
+    for group in opt_parser.option_groups:
+        for option in group.option_list:
+            #for every long flag
+            opts_flag.append(option.get_opt_string())
+    with open(BASH_COMPLETION_TEMPLATE) as f:
+        template = f.read()
+    with open(BASH_COMPLETION_FILE, "w") as f:
+        #just using the special char
+        print opts_flag
+        filled_template = template.replace("{{flags}}", " ".join(opts_flag))
+        f.write(filled_template)
+
+parser = youtube_dl.parseOpts()[0]
+build_completion(parser)
diff --git a/devscripts/bash_completion.template b/devscripts/bash_completion.template
new file mode 100644
index 000000000..3b99a9614
--- /dev/null
+++ b/devscripts/bash_completion.template
@@ -0,0 +1,14 @@
+__youtube-dl()
+{
+    local cur prev opts
+    COMPREPLY=()
+    cur="${COMP_WORDS[COMP_CWORD]}"
+    opts="{{flags}}"
+
+    if [[ ${cur} == * ]] ; then
+        COMPREPLY=( $(compgen -W "${opts}" -- ${cur}) )
+        return 0
+    fi
+}
+
+complete -F __youtube-dl youtube-dl