From 93f89bc7d68cbb1c6c23d19cb5f434ec68ce93c8 Mon Sep 17 00:00:00 2001
From: XANTRONIX Development <dev@xantronix.com>
Date: Sun, 5 Jan 2025 00:15:24 -0500
Subject: [PATCH] Make bin/xenu-nntp-server includeable

---
 bin/xenu-nntp-server | 28 ++++++++++++++++------------
 1 file changed, 16 insertions(+), 12 deletions(-)

diff --git a/bin/xenu-nntp-server b/bin/xenu-nntp-server
index 26487ba..9a07bac 100755
--- a/bin/xenu-nntp-server
+++ b/bin/xenu-nntp-server
@@ -7,19 +7,23 @@ from xenu_nntp.config import Config
 from xenu_nntp.server import Server
 from xenu_nntp.daemon import Daemon
 
-parser = argparse.ArgumentParser(description='Tiny NNTP server')
-parser.add_argument('--daemon',      '-d', action='store_true', help='Run NNTP server as daemon in background')
-parser.add_argument('--config-file', '-f', type=str,            help='Specify a configuration file location')
+def main():
+    parser = argparse.ArgumentParser(description='Tiny NNTP server')
+    parser.add_argument('--daemon',      '-d', action='store_true', help='Run NNTP server as daemon in background')
+    parser.add_argument('--config-file', '-f', type=str,            help='Specify a configuration file location')
 
-args = parser.parse_args()
+    args = parser.parse_args()
 
-config = Config.load(args.config_file)
-server = Server(config)
+    config = Config.load(args.config_file)
+    server = Server(config)
 
-if args.daemon:
-    Daemon.init(config)
+    if args.daemon:
+        Daemon.init(config)
 
-try:
-    server.run()
-except KeyboardInterrupt:
-    server.stop()
+    try:
+        server.run()
+    except KeyboardInterrupt:
+        server.stop()
+
+if __name__ == '__main__':
+    exit(main())