node: Added -S option to set the split size

Signed-off-by: Hubert Figuière <[email protected]>

Author: hfiguiere

node/flatpak_node_generator/main.py

@@ -95,6 +95,14 @@ async def _async_main() -> None:
         action='store_true',
         help='Split the sources file to fit onto GitHub.',
     )
+    parser.add_argument(
+        '-S',
+        '--split-size',
+        type=int,
+        default=0,
+        dest='split_size',
+        help='If splitting the sources file, split at this size in KB. Default is 49000KB.',
+    )
     parser.add_argument(
         '--node-chromedriver-from-electron',
         help='Use the ChromeDriver version associated with the given '
@@ -245,6 +253,8 @@ async def _async_main() -> None:
             )
             gen.add_command(f'bash {gen.data_root / script_name}')
 
+    if args.split_size > 0:
+        gen.split_size = args.split_size * 1000
     if args.split:
         sources = list(gen.ordered_sources())
         await Requests.instance.upgrade_to_sha256(sources)

node/flatpak_node_generator/manifest.py

@@ -27,6 +27,7 @@ def __init__(self) -> None:
         # That way, we ensure uniqueness.
         self._sources: Set[Tuple[Tuple[str, Any], ...]] = set()
         self._commands: List[str] = []
+        self._split_size = -1
 
     def __exit__(
         self,
@@ -36,6 +37,16 @@ def __exit__(
     ) -> None:
         self._finalize()
 
+    @property
+    def split_size(self) -> int:
+        if self._split_size == -1:
+            return ManifestGenerator.MAX_GITHUB_SIZE
+        else:
+            return self._split_size
+    @split_size.setter
+    def split_size(self, val):
+        self._split_size = val
+
     @property
     def data_root(self) -> Path:
         return Path('flatpak-node')
@@ -66,7 +77,7 @@ def split_sources(self) -> Iterator[List[Dict[Any, Any]]]:
             # opening brackets.
             source_json = json.dumps([source], indent=ManifestGenerator.JSON_INDENT)
             source_json_len = len('\n'.join(source_json.splitlines()[1:-1]))
-            if current_size + source_json_len >= ManifestGenerator.MAX_GITHUB_SIZE:
+            if current_size + source_json_len >= self.split_size:
                 yield current
                 current = []
                 current_size = BASE_CURRENT_SIZE