Initial commit.

bin/findtargets.py

@@ -0,0 +1,30 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+import configparser
+import os
+import sys
+
+if len(sys.argv) < 3:
+    print("Syntax: {} BUILD_DIR FILE_LIST".format(sys.argv[0]))
+    sys.exit(1)
+
+source_dirs = sys.argv[2:]
+dst_dir = sys.argv[1]
+
+for src_dir in source_dirs:
+    pkg_name = os.path.basename(src_dir)
+
+    info = {
+        "name": pkg_name,
+        "version": 1.0
+    }
+
+    if os.path.isfile(src_dir + "/baseinfo.ini"):
+        cfg = configparser.ConfigParser()
+        cfg.read(src_dir + "/baseinfo.ini")
+        for key in cfg["info"]:
+            info[key] = cfg["info"][key]
+
+    dst_file = os.path.normpath(dst_dir + "/" + "{}-{}.mkp".format(info["name"], info["version"]))
+    print(dst_file)

bin/makemkp.py

@@ -0,0 +1,131 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+import configparser
+import json
+import os
+import sys
+import pprint
+import subprocess
+import tarfile
+import time
+from io import BytesIO
+
+if len(sys.argv) < 3:
+    print("Syntax: {} SRCDIR DESTDIR".format(sys.argv[0]))
+    sys.exit(1)
+
+src_dir = sys.argv[1]
+dst_dir = sys.argv[2]
+
+if not os.path.isdir(src_dir):
+    print("{} is not a directory!".format(src_dir))
+    sys.exit(2)
+
+if not os.path.isdir(dst_dir):
+    print("{} is not a directory!".format(dst_dir))
+    sys.exit(3)
+
+pkg_name = os.path.basename(src_dir)
+
+info = {
+    "title"                : "Title of {}".format(pkg_name),
+    "name"                 : pkg_name,
+    "description"          : "Please add a description here",
+    "version"              : "1.0",
+    "version.packaged"     : "1.4.0",
+    "version.min_required" : "1.4.0",
+    "packaged_by"          : "makemkp.py",
+    "author"               : "Add your name here",
+    "download_url"         : "http://example.com/{}/".format(pkg_name),
+    "files"                : {}
+}
+
+cfg = configparser.ConfigParser()
+cfg.read(src_dir + "/baseinfo.ini")
+
+for key in cfg["info"]:
+    info[key] = cfg["info"][key].encode("utf-8")
+
+dst_file = os.path.normpath(dst_dir + "/" + "{}-{}.mkp".format(info["name"], info["version"]))
+
+print("Packaging {} v{} into {}...".format(info["name"], info["version"], dst_file))
+
+tar = tarfile.open(name=dst_file, mode="w:gz")
+
+
+# COLLECT FILES
+
+package_parts = [ (part, title, perm) for part, title, perm in [
+  ( "checks",        "Checks",                    0644 ),
+  ( "notifications", "Notification scripts",      0755 ),
+  ( "inventory",     "Inventory plugins",         0644 ),
+  ( "checkman",      "Checks' man pages",         0644 ),
+  ( "agents",        "Agents",                    0755 ),
+  ( "web",           "Multisite extensions",      0644 ),
+  ( "pnp-templates", "PNP4Nagios templates",      0644 ),
+  ( "doc",           "Documentation files",       0644 ),
+  ( "bin",           "Binaries",                  0755 ),
+  ( "lib",           "Libraries",                 0644 ),
+  ( "mibs",          "SNMP MIBs",                 0644 ),
+]]
+
+def files_in_dir(dir, prefix = ""):
+    if dir == None or not os.path.exists(dir):
+        return []
+
+    result = []
+    files = os.listdir(dir)
+    for f in files:
+        if f in [ '.', '..' ] or f.startswith('.') or f.endswith('~'):
+            continue
+
+        path = dir + "/" + f
+        if os.path.isdir(path):
+            result += files_in_dir(path, prefix + f + "/")
+        else:
+            result.append(prefix + f)
+    result.sort()
+    return result
+
+def create_tar_info(filename, size):
+    info = tarfile.TarInfo()
+    info.mtime = time.time()
+    info.uid = 0
+    info.gid = 0
+    info.size = size
+    info.mode = 0644
+    info.type = tarfile.REGTYPE
+    info.name = filename
+    return info
+
+def tar_from_string(tar, filename, payload):
+    data_stream = BytesIO(payload)
+    tarinfo = create_tar_info(filename, len(data_stream.getvalue()))
+    tar.addfile(tarinfo, data_stream)
+
+files = {}
+num_files = 0
+for part, title, perm in package_parts:
+    files_list = files_in_dir(src_dir + "/" + part)
+    files[part] = files_list
+    num_files += len(files_list)
+
+info["files"] = files
+info["num_files"] = num_files
+
+info_file = pprint.pformat(info)
+info_json = json.dumps(info)
+tar_from_string(tar, "info", info_file)
+tar_from_string(tar, "info.json", info_json)
+
+
+for part in info["files"]:
+    filenames = info["files"][part]
+    if len(filenames) > 0:
+        subtarname = part + ".tar"
+        subdata = subprocess.check_output(["tar", "cf", "-", "--dereference", "--force-local",
+                                           "-C", src_dir + "/" + part] + filenames)
+        tar_from_string(tar, subtarname, subdata)
+
+tar.close()