From 0f42cdd63936fcc4c805c55e3b9c666d2bd64439 Mon Sep 17 00:00:00 2001
From: Andrew Murray <radarhere@users.noreply.github.com>
Date: Fri, 23 Nov 2018 18:57:17 +1100
Subject: [PATCH] Added functions to Read The Docs sidebar

---
 docs/_static/js/script.js | 60 +++++++++++++++++++++++++++++++++++++++
 docs/conf.py              |  3 ++
 2 files changed, 63 insertions(+)
 create mode 100644 docs/_static/js/script.js

diff --git a/docs/_static/js/script.js b/docs/_static/js/script.js
new file mode 100644
index 000000000..2cd7335ec
--- /dev/null
+++ b/docs/_static/js/script.js
@@ -0,0 +1,60 @@
+jQuery(document).ready(function ($) {
+    setTimeout(function () {
+        var sectionID = 'base';
+        var search = function ($section, $sidebarItem) {
+            $section.children('.section, .function, .method').each(function () {
+                if ($(this).hasClass('section')) {
+                    sectionID = $(this).attr('id');
+                    search($(this), $sidebarItem.parent().find('[href=#'+sectionID+']'));
+                } else {
+                    var $dt = $(this).children('dt');
+                    var id = $dt.attr('id');
+                    if (id === undefined) {
+                        return;
+                    }
+
+                    var $functionsUL = $sidebarItem.siblings('[data-sectionID='+sectionID+']');
+                    if (!$functionsUL.length) {
+                        $functionsUL = $('<ul />').attr('data-sectionID', sectionID);
+                        $functionsUL.insertAfter($sidebarItem);
+                    }
+
+                    var $li = $('<li />');
+                    var $a = $('<a />').css('font-size', '11.5px');
+                    var $upperA = $sidebarItem.parent().children('a');
+                    var $upperAParent = $upperA.parent();
+                    if ($upperAParent.hasClass('toctree-l2')) {
+                        $a.css('background-color', '#c9c9c9');
+                        $a.css('padding-left', '4em');
+                    } else if ($upperAParent.hasClass('toctree-l3')) {
+                        if (!$upperA.find('.toctree-expand').length) {
+                            $upperA.prepend($('<span />').addClass('toctree-expand'));
+                        }
+                        $a.css('background-color', '#c9c9c9');
+                        $a.css('padding-left', '5em');
+                    } else {
+                        $a.css('background-color', '#bdbdbd');
+                        $a.css('padding-left', '6.25em');
+                    }
+                    $a.attr('href', '#'+id);
+                    $a.text('- '+$dt.find('code').text());
+                    $a.click(function () {
+                        setTimeout(function () {
+                            $a.css('font-weight', 'bold');
+                        }, 0);
+                    });
+                    $li.append($a);
+                    $functionsUL.append($li);
+                }
+            });
+        };
+        search($('[itemprop=articleBody] > .section'), $('.wy-nav-side a[href=#]'));
+    }, 0);
+    $(window).on('hashchange', function () {
+        $('ul[data-sectionID]').each(function () {
+            $(this).find('a').each(function () {
+                $(this).css('font-weight', 'normal');
+            });
+        });
+    });
+});
diff --git a/docs/conf.py b/docs/conf.py
index b788febee..f569f5bdc 100644
--- a/docs/conf.py
+++ b/docs/conf.py
@@ -286,3 +286,6 @@ texinfo_documents = [
 
 # If true, do not generate a @detailmenu in the "Top" node's menu.
 # texinfo_no_detailmenu = False
+
+def setup(app):
+    app.add_javascript('js/script.js')