Merge pull request #1469 from p-l-/add-view-plugins

Support plugins for db2view

ivre/view.py

@@ -1,7 +1,7 @@
 #! /usr/bin/env python
 
 # This file is part of IVRE.
-# Copyright 2011 - 2022 Pierre LALET <pierre@droids-corp.org>
+# Copyright 2011 - 2023 Pierre LALET <pierre@droids-corp.org>
 #
 # IVRE is free software: you can redistribute it and/or modify it
 # under the terms of the GNU General Public License as published by
@@ -24,6 +24,14 @@
 from datetime import datetime
 from textwrap import wrap
 
+try:
+    from importlib.metadata import entry_points
+except ImportError:
+    HAS_PLUGINS = False
+else:
+    HAS_PLUGINS = True
+
+
 from ivre import utils
 from ivre.active.cpe import add_cpe_values
 from ivre.active.data import (
@@ -40,6 +48,12 @@
 from ivre.xmlnmap import add_service_hostname
 
 
+def load_plugins():
+    for entry_point in entry_points(group="ivre.plugins.view"):
+        if entry_point.name.startswith("_install_"):
+            entry_point.load()(globals())
+
+
 def _extract_passive_HTTP_CLIENT_HEADER_SERVER(rec):
     """Handle http client header about server."""
     return {
@@ -950,3 +964,7 @@ def prepare_record(rec):
             cur_addr = min(next_addrs, key=utils.ip2int)
     if cur_rec is not None:
         yield prepare_record(cur_rec)
+
+
+if HAS_PLUGINS:
+    load_plugins()