Move logic from genwrap and genstub to hdl_controller

Author: rodrigomelo9

hdltools/apps/genstub.py

@@ -9,11 +9,8 @@
 from hdltools.cli_parser import cli_parser
 from hdltools.hdl_controller import HDLController
 
-args = cli_parser('wrap')
-ctrl = HDLController('wrap')
-code = ctrl.generate(args.file, args.top, args.suffix)
 
-if not args.output:
-    print(code)
-else:
-    ctrl.write(args.output)
+args = cli_parser('stub')
+ctrl = HDLController('stub')
+ctrl.generate(args.file, args.top, args.suffix)
+ctrl.write(args.output)

hdltools/apps/genwrap.py

@@ -9,11 +9,8 @@
 from hdltools.cli_parser import cli_parser
 from hdltools.hdl_controller import HDLController
 
+
 args = cli_parser('wrap')
 ctrl = HDLController('wrap')
-code = ctrl.generate(args.file, args.top, args.suffix)
-
-if not args.output:
-    print(code)
-else:
-    ctrl.write(args.output)
+ctrl.generate(args.file, args.top, args.suffix)
+ctrl.write(args.output)

hdltools/hdl_controller.py

@@ -54,8 +54,11 @@ def generate(self, filepath, top=None, suffix=None):
         return self.writer.get_code()
 
     def write(self, filepath):
-        """Writes the generated HDL code to the specified file."""
-        self.writer.write_file(filepath)
+        """Writes the generated HDL code."""
+        if not filepath:
+            print(self.writer.get_code())
+        else:
+            self.writer.write_file(filepath)
 
     @staticmethod
     def _error(message, ecode=1):