Merge branch 'master' of https://github.com/iiab/iiab

roles/kiwix/templates/iiab-make-kiwix-lib

@@ -18,6 +18,8 @@ if flock -n -e 200; then :
   echo "Now running iiab-make-kiwix-lib.py"
   # write to {{ kiwix_library_xml }}.tmp to minimize kiwix down
   # zim map could be out of sync for a few seconds
+  # using new version that does deltas
+  cp $KIWIXLIB $KIWIXLIB.tmp
   /usr/bin/iiab-make-kiwix-lib.py
   {{ systemctl_program }} stop kiwix-serve
   rm $KIWIXLIB
@@ -27,4 +29,5 @@ else
   echo "Can't get wait lock for iiab-make-kiwix-lib.py";
   exit 1;
 fi
+echo 'Finished making Kiwix library.xml'
 exit 0

roles/kiwix/templates/iiab-make-kiwix-lib.py

@@ -3,6 +3,7 @@
 """
 
    Creates temp library.xml file for kiwix from contents of /zims/content and index
+   Updated to handle incremental additions and deletions
 
    Author: Tim Moody <tim(at)timmoody(dot)com>
    Contributors: Jerry Vonau <jvonau3(at)gmail.com>
@@ -19,68 +20,110 @@ import re
 import subprocess
 import shlex
 import ConfigParser
+import xml.etree.ElementTree as ET
+import argparse
+
 IIAB_PATH='/etc/iiab'
 if not IIAB_PATH in sys.path:
     sys.path.append(IIAB_PATH)
 from iiab_env import get_iiab_env
 
 # Config Files
-iiab_config_file = "{{ iiab_config_file }}"
+# iiab_config_file should be in /etc/iiab/iiab.env
+iiab_config_file = "{{ iiab_config_file }}" # nominally /etc/iiab/iiab.ini
+# iiab_config_file = "/etc/iiab/iiab.ini" # comment out after testing
+
+IIAB_INI = get_iiab_env('IIAB_INI') # future
+if IIAB_INI:
+    iiab_config_file = IIAB_INI
 
 # Variables that should be read from config file
 # All of these variables will be read from config files and recomputed in init()
-iiab_zim_path = "{{ iiab_zim_path }}"
+zim_path = "/library/zims"
 
-# Later we will append .tmp to file name
-kiwix_library_xml = "{{ kiwix_library_xml }}"
-
-iiab_base_path = "{{ iiab_base }}"
+iiab_base_path = "/opt/iiab"
 kiwix_manage = iiab_base_path + "/kiwix/bin/kiwix-manage"
 doc_root = get_iiab_env('WWWROOT')
-zim_version_idx = doc_root + "/common/assets/zim_version_idx.json"
-zim_versions = {}
+zim_version_idx_dir = doc_root + "/common/assets/"
+zim_version_idx_file = "zim_version_idx.json"
+
 old_zim_map = {"bad.zim" : "unparseable name"}
 
+# Working variables
+# zim_files - list of zims and possible index from file system
+# path_to_array_map - list of zims in current library.xml with array index number (for delete)
+zim_versions = {} # map of zim's generic name to version installed, e.g. wikipedia_es_all to wikipedia_es_all_2017-01
+
 def main():
     """Server routine"""
     global kiwix_library_xml
-    init()
-    kiwix_library_xml += '.tmp' # write to temp file    
+    global zim_path
+    global zim_version_idx_dir
+    global zim_version_idx_file
 
-    # remove existing file
+    init()
+    args = parse_args()
+    if args.device: # allow override of path
+        zim_path = args.device + zim_path
+        zim_version_idx_dir = args.device + zim_version_idx_dir
+
+    kiwix_library_xml = zim_path + "/library.xml"
+    if not args.no_tmp: # don't append .tmp
+        kiwix_library_xml += ".tmp"
+
+    # remove existing file if force
+    if args.force:
         try:
             os.remove(kiwix_library_xml)
         except OSError:
             pass
+        zims_installed = {}
+        path_to_array_map = {}
+    else:
+          zims_installed, path_to_array_map = read_library_xml(kiwix_library_xml)
 
-    # add each file in /library/zims/content with corresponding index
-    # only add a single .zim for each .zimxx file
+    zim_files = get_zim_list(zim_path)
 
+    # Remove zims not in file system from library.xml
+    remove_list_str = ""
+    for item in path_to_array_map:
+          if item not in zim_files:
+              remove_list_str += str(path_to_array_map[item]) + " "
+    if remove_list_str:
+        rem_libr_xml(remove_list_str)
+
+    # Add zims from file system that are not in library.xml
+    for item in zim_files:
+          if item not in path_to_array_map:
+              add_libr_xml(kiwix_library_xml, zim_path, item, zim_files[item])
+
+    # Write Version Map
+    if os.path.isdir(zim_version_idx_dir):
+        with open(zim_version_idx_dir + zim_version_idx_file, 'w') as fp:
+            json.dump(zim_versions, fp)
+    else:
+        print zim_version_idx_dir + " not found."
+    sys.exit()
+
+def get_zim_list(path):
     files_processed = {}
-    content = iiab_zim_path + "/content/"
-    index = iiab_zim_path + "/index/"
-
+    zim_list = []
+    content = path + "/content/"
+    index = path + "/index/"
     flist = os.listdir(content)
     flist.sort()
     for filename in flist:
         zimpos = filename.find(".zim")
         if zimpos != -1:
             filename = filename[:zimpos]
-            if filename not in files_processed:
-                files_processed[filename] = True
+            zimname = "content/" + filename + ".zim"
+            zimidx = "index/" + filename + ".zim.idx"
+            if zimname not in files_processed:
+                if not os.path.isdir (path + "/" + zimidx): # only declare index if exists (could be embedded)
+                    zimidx = None
+                files_processed[zimname] = zimidx
                 zimname = content + filename + ".zim"
                 zimidx = index + filename + ".zim.idx"
-                command = kiwix_manage + " " + kiwix_library_xml + " add " + zimname
-                if os.path.isdir (zimidx): # only declare index if exists (could be embedded)
-                    command += " -i " + zimidx
-                #print command
-                args = shlex.split(command)
-                try:
-                    outp = subprocess.check_output(args)
-
-                    # create map of generic zim name to actual, assumes pattern of <name>_<yyyy-mm>
-                    # all current files follow this pattern, but some older ones, no longer in the catalog, do not
-
                 if filename in old_zim_map: # handle old names that don't parse
                     wiki_name = old_zim_map[filename]
                 else:
@@ -89,32 +132,75 @@ def main():
                     if "gutenberg_" in filename:
                         ulpos = filename[:ulpos].rfind("_")
                     wiki_name = filename[:ulpos]
-
                 zim_versions[wiki_name] = filename # if there are multiples, last should win
+    return files_processed
+
+def read_library_xml(lib_xml_file, kiwix_exclude_attr=[""]): # duplicated from iiab-cmdsrv
+    kiwix_exclude_attr.append("id") # don't include id
+    kiwix_exclude_attr.append("favicon") # don't include large favicon
+    zims_installed = {}
+    path_to_array_map = {}
+    try:
+        tree = ET.parse(lib_xml_file)
+        root = tree.getroot()
+        xml_item_no = 0
+        for child in root:
+            xml_item_no += 1 # hopefully this is the array number
+            attributes = {}
+            if 'id' not in child.attrib: # is this necessary? implies there are records with no book id which would break index for removal
+                  print "xml record missing Book Id"
+            id = child.attrib['id']
+            for attr in child.attrib:
+                if attr not in kiwix_exclude_attr:
+                    attributes[attr] = child.attrib[attr] # copy if not id or in exclusion list
+            zims_installed[id] = attributes
+            path_to_array_map[child.attrib['path']] = xml_item_no
+    except IOError:
+        zims_installed = {}
+    return zims_installed, path_to_array_map
+
+def rem_libr_xml(list_str):
+    command = kiwix_manage + " " + kiwix_library_xml + " remove " + list_str
+    print command
+    args = shlex.split(command)
+
+    outp = subprocess.check_output(args)
+
+def add_libr_xml(kiwix_library_xml, zim_path, zimname, zimidx):
+    command = kiwix_manage + " " + kiwix_library_xml + " add " + zim_path + "/" + zimname
+    if zimidx:
+          command += " -i " + zim_path + "/" + zimidx
+    print command
+    args = shlex.split(command)
+    try:
+        outp = subprocess.check_output(args)
 
     except: #skip things that don't work
         print 'skipping ' + filename
         pass
 
-    with open(zim_version_idx, 'w') as fp:
-        json.dump(zim_versions, fp)
-
-    sys.exit()
-
 def init():
 
     global iiab_base_path
-    global iiab_zim_path
+    global zim_path
     global kiwix_library_xml
     global kiwix_manage
 
     config = ConfigParser.SafeConfigParser()
     config.read(iiab_config_file)
     iiab_base_path = config.get('location','iiab_base')
-    iiab_zim_path = config.get('kiwix','iiab_zim_path')
+    zim_path = config.get('kiwix','iiab_zim_path')
     kiwix_library_xml = config.get('kiwix','kiwix_library_xml')
     kiwix_manage = iiab_base_path + "/kiwix/bin/kiwix-manage"
 
+def parse_args():
+    parser = argparse.ArgumentParser(description="Create library.xml for Kiwix.")
+    parser.add_argument("--device", help="no trailing /. change the target device from internal storage to something else like /media/usb0")
+    parser.add_argument("--no_tmp", help="don't append .tmp to the library.xml name", action="store_true")
+    parser.add_argument("-f", "--force", help="force complete rebuild of library.xml", action="store_true")
+    parser.add_argument("-v", "--verbose", help="Print messages.", action="store_true")
+    return parser.parse_args()
+
 # Now start the application
 
 if __name__ == "__main__":

roles/openvpn/tasks/main.yml

@@ -87,6 +87,18 @@
     # Obsolete & unused for ~2 years as of August 2018:
     #- { src: 'iiab-vpn.j2', dest: '/usr/bin/iiab-vpn', mode: '0755' }
 
+- name: Create iiab-vpn-on (symlink to iiab-remote-on for now)
+  file:
+    src: /usr/bin/iiab-remote-on
+    path: /usr/bin/iiab-vpn-on
+    state: link
+
+- name: Create iiab-vpn-off (symlink to iiab-remote-off for now)
+  file:
+    src: /usr/bin/iiab-remote-off
+    path: /usr/bin/iiab-vpn-off
+    state: link
+
 # up_wan was being installed twice (also above) and was unused for ~2 years
 # as of August 2018: (see 15-openvpn below)
 #- name: Put up_wan in place (debuntu)

roles/openvpn/templates/iiab-remote-off

@@ -1,17 +1,29 @@
 #!/bin/bash
-# script to turn on openvpn
 
-# do nothing if it is not installed
+# /usr/bin/iiab-remote-off should fully turn off multiple remote support
+# services like OpenVPN and others, to reduce risk of remote attacks.
+
+echo -e '\nWARNING: To disable OpenVPN long-term, it'"'"'s recommended you:\n'
+
+echo -e '1) Set this variable in /etc/local/local_vars.yml'
+echo -e '   openvpn_enabled: False\n'
+
+echo -e '2) Run:'
+echo -e '   cd /opt/iiab/iiab'
+echo -e '   sudo ./runrole openvpn\n'
+
+# Do nothing if OpenVPN not installed
 which openvpn
 if [ $? -ne 0 ]; then
-    echo Cannot find the OpenVPN program (openvpn).
+    echo 'Cannot find the OpenVPN program (openvpn).'
     exit 1
 fi
+
 systemctl disable openvpn
 systemctl stop openvpn
 
 sleep 5
-ps -e|grep vpn
+ps -e | grep openvpn    # 2018-09-05: "ps -e | grep vpn" no longer works (nor would "pgrep vpn") when invoked from iiab-vpn-off (as filename itself causes [multiple] "vpn" instances to appear in process list!)
 if [ $? -eq 0 ]; then
     echo OpenVPN failed to stop.
 else

roles/openvpn/templates/iiab-remote-on.j2

@@ -1,12 +1,25 @@
 #!/bin/bash
-# script to turn on openvpn
 
-# do nothing if it is not installed
+# /usr/bin/iiab-remote-on should turn on multiple remote support services like
+# OpenVPN and others, for remote support, so they work even after reboot.
+
+echo -e '\nWARNING: To enable OpenVPN long-term, it'"'"'s recommended you:\n'
+
+echo -e '1) Set these variables in /etc/local/local_vars.yml'
+echo -e '   openvpn_install: True'
+echo -e '   openvpn_enabled: True\n'
+
+echo -e '2) Run:'
+echo -e '   cd /opt/iiab/iiab'
+echo -e '   sudo ./runrole openvpn\n'
+
+# Do nothing if OpenVPN not installed
 which openvpn
 if [ $? -ne 0 ]; then
-    echo Cannot find the OpenVPN program (openvpn).
+    echo 'Cannot find the OpenVPN program (openvpn).'
     exit 1
 fi
+
 systemctl enable openvpn
 systemctl start openvpn