cobol_testgen/grammar.lark

@@ -1,7 +1,8 @@
 start: data_div_content
 data_div_content: (file_section | working_storage | linkage)*
-file_section: "FILE" "SECTION" DOT fd+
+file_section: "FILE" "SECTION" DOT (fd | sd)+
 fd: "FD" NAME FD_SUFFIX data_item+
+sd: "SD" NAME FD_SUFFIX data_item*
 FD_SUFFIX: /(?:"[^"]*"|'[^']*'|[^.])*\./
 working_storage: "WORKING-STORAGE" "SECTION" DOT data_item*
 linkage: "LINKAGE" "SECTION" DOT data_item*
@@ -21,7 +22,9 @@ value_literal: INT | SIGNED_NUMBER | STRING | SQSTRING
              | "LOW-VALUE" | "LOW-VALUES"
 SQSTRING: /'[^']*'/
 redefines_clause: "REDEFINES" NAME
-occurs_clause: "OCCURS" INT "TIMES"? ("DEPENDING" "ON" NAME)?
+occurs_clause: "OCCURS" INT "TIMES"? ("DEPENDING" "ON" NAME)? key_clause? indexed_clause?
+key_clause: ("ASCENDING" | "DESCENDING") "KEY" "IS"? NAME (","? NAME)*
+indexed_clause: "INDEXED" "BY" NAME (","? NAME)*
 usage_clause: USAGE_VAL
 USAGE_VAL: "COMP" | "COMP-3" | "COMP-5" | "BINARY" | "PACKED-DECIMAL" | "DISPLAY"
 LEVEL: /0[1-9]|[1-4][0-9]|49|77|88/

cobol_testgen/read.py

@@ -425,14 +425,15 @@ def parse_file_section(source: str) -> dict:
         return {}
     fs = m.group(1)
     result = {}
-    # ? FD ?????? FD ?
-    fd_blocks = re.split(r'\n\s*(?=FD\s+)', fs.strip())
-    for block in fd_blocks:
-        m = re.match(r'FD\s+(\w[\w-]*)', block, re.IGNORECASE)
+    # FD 和 SD 条目
+    blocks = re.split(r'\n\s*(?=(?:FD|SD)\s+)', fs.strip())
+    for block in blocks:
+        m = re.match(r'(FD|SD)\s+(\w[\w-]*)', block, re.IGNORECASE)
         if not m:
             continue
-        name = m.group(1).upper()
-        # ???????? 01 ????
+        entry_type = m.group(1).upper()  # "FD" or "SD"
+        name = m.group(2).upper()
+        # 找 01 层记录
         recs = re.findall(r'^\s*0{0,1}1\s+(\w[\w-]*)', block, re.MULTILINE)
         result[name] = [r.upper() for r in recs]
     return result