Update dirStream.py

src/ms_ovba/Models/Entities/module_base.py

@@ -4,7 +4,7 @@
 )
 from ms_ovba.Models.Fields.packed_data import PackedData
 from ms_ovba.Models.Fields.idSizeField import IdSizeField
-from typing import TypeVar
+from typing import Literal, TypeVar
 
 
 T = TypeVar('T', bound='ModuleBase')
@@ -24,7 +24,7 @@ def __init__(self: T, name: str) -> None:
         # self.readonly = SimpleRecord(0x001E, 4, helpContext)
         # self.private = SimpleRecord(0x001E, 4, helpContext)
         self._cache = b''
-        self.workspace = [0, 0, 0, 0, 'C']
+        self.workspace: tuple[int, int, int, int, Literal['C', 'I', 'Z']]
         self.type = ''
         self.created = 0
         self.modified = 0
@@ -76,8 +76,9 @@ def bin_path(self: T) -> str:
         return self._file_path + ".bin"
 
     def add_workspace(self: T, val1: int, val2: int,
-                      val3: int, val4: int, val5: int) -> None:
-        self.workspace = [val1, val2, val3, val4, val5]
+                      val3: int, val4: int,
+                      val5: Literal['C', 'I', 'Z']) -> None:
+        self.workspace = (val1, val2, val3, val4, val5)
 
     def pack(self: T, endien: str, cp_name: str) -> bytes:
         """

src/ms_ovba/Views/dirStream.py

@@ -21,19 +21,19 @@ class DirStream():
     """
 
     def __init__(self: T, project: VbaProject) -> None:
-        self.project = project
+        self._project = project
         self._include_compat = project.compat
 
     def to_bytes(self: T) -> bytes:
         information = self._load_information()
-        endien = self.project.endien
-        cp_name = self.project.codepage_name
+        endien = self._project.endien
+        cp_name = self._project.codepage_name
         pack_symbol = '<' if endien == 'little' else '>'
         # should be 0xFFFF
-        cookie_value = self.project.project_cookie
+        cookie_value = self._project.project_cookie
         self.project_cookie = IdSizeField(19, 2, cookie_value)
-        references = self.project.references
-        modules = self.project.modules
+        references = self._project.references
+        modules = self._project.modules
         output = b''
         for record in information:
             output += record.pack(endien, cp_name)
@@ -70,7 +70,7 @@ def _load_information(self: T) -> List:
         project_name = IdSizeField(4, 10, "VBAProject")
         docstring = DoubleEncodedString([5, 0x0040], "")
         helpfile = DoubleEncodedString([6, 0x003D], "")
-        help_context = IdSizeField(7, 4, 0)
+        help_context = IdSizeField(7, 4, self._project.help_context_id)
         lib_flags = IdSizeField(8, 4, 0)
         version = IdSizeField(9, 4, 0x65BE0257)
         minor_version = PackedData("H", 17)