fix add lto function

Author: Jason2866

builder/frameworks/component_manager.py

@@ -1328,58 +1328,77 @@ def add_lto_flags(self) -> bool:
 
         try:
             with open(build_py_path, 'r', encoding='utf-8') as f:
-                lines = f.readlines()
+                content = f.read()
 
+            original_content = content
             modified = False
-            new_lines = []
 
-            for line in lines:
-                # Add LTO flags to compiler flags
-                if 'CCFLAGS=' in line and '-flto' not in line:
-                    line = line.rstrip()
-                    if line.endswith(','):
-                        line = line[:-1] + ', "-flto=auto",\n'
-                    elif line.endswith(']'):
-                        line = line[:-1] + ', "-flto=auto"]\n'
-                    else:
-                        line = line + '\n'
-                    modified = True
-                    
-                elif 'CFLAGS=' in line and 'CCFLAGS' not in line and '-flto' not in line:
-                    line = line.rstrip()
-                    if line.endswith(','):
-                        line = line[:-1] + ', "-flto=auto",\n'
-                    elif line.endswith(']'):
-                        line = line[:-1] + ', "-flto=auto"]\n'
-                    else:
-                        line = line + '\n'
-                    modified = True
-                    
-                elif 'CXXFLAGS=' in line and '-flto' not in line:
-                    line = line.rstrip()
-                    if line.endswith(','):
-                        line = line[:-1] + ', "-flto=auto",\n'
-                    elif line.endswith(']'):
-                        line = line[:-1] + ', "-flto=auto"]\n'
+            # Pattern to find CCFLAGS list and add -flto=auto if not present
+            if 'CCFLAGS=[' in content and '"-flto' not in content[content.find('CCFLAGS=['):content.find('],', content.find('CCFLAGS=['))]:
+                # Find the closing bracket for CCFLAGS
+                start = content.find('CCFLAGS=[')
+                end = content.find('],', start) + 1
+                ccflags_section = content[start:end]
+                
+                # Add -flto=auto before the closing bracket
+                new_ccflags = ccflags_section.replace('],', ',\n        "-flto=auto"\n    ],')
+                content = content[:start] + new_ccflags + content[end:]
+                modified = True
+            
+            # Pattern to find CFLAGS list and add -flto=auto if not present
+            if 'CFLAGS=[' in content and '"-flto' not in content[content.find('CFLAGS=['):content.find('],', content.find('CFLAGS=['))]:
+                start = content.find('CFLAGS=[')
+                end = content.find('],', start) + 1
+                cflags_section = content[start:end]
+                
+                new_cflags = cflags_section.replace('],', ',\n        "-flto=auto"\n    ],')
+                content = content[:start] + new_cflags + content[end:]
+                modified = True
+            
+            # Pattern to find CXXFLAGS list and add -flto=auto if not present
+            if 'CXXFLAGS=[' in content and '"-flto' not in content[content.find('CXXFLAGS=['):content.find('],', content.find('CXXFLAGS=['))]:
+                start = content.find('CXXFLAGS=[')
+                end = content.find('],', start) + 1
+                cxxflags_section = content[start:end]
+                
+                new_cxxflags = cxxflags_section.replace('],', ',\n        "-flto=auto"\n    ],')
+                content = content[:start] + new_cxxflags + content[end:]
+                modified = True
+            
+            # Pattern to find LINKFLAGS list and add -flto if not present
+            # Note: remove_no_lto_flags() needs to be called before
+            if 'LINKFLAGS=[' in content:
+                start = content.find('LINKFLAGS=[')
+                # Find the end of LINKFLAGS - look for the matching bracket
+                bracket_count = 0
+                pos = start + len('LINKFLAGS=[') - 1
+                while pos < len(content):
+                    if content[pos] == '[':
+                        bracket_count += 1
+                    elif content[pos] == ']':
+                        bracket_count -= 1
+                        if bracket_count == 0:
+                            end = pos + 1
+                            break
+                    pos += 1
+                
+                linkflags_section = content[start:end]
+                
+                # Check if -flto is already present
+                if '"-flto"' not in linkflags_section:
+                    # Add -flto before the closing bracket
+                    new_linkflags = linkflags_section.rstrip()
+                    if new_linkflags.endswith(','):
+                        new_linkflags = new_linkflags + '\n        "-flto"'
                     else:
-                        line = line + '\n'
-                    modified = True
+                        new_linkflags = new_linkflags[:-1] + ',\n        "-flto"\n    ]'
 
-                elif 'LINKFLAGS=' in line and '-flto' not in line:
-                    line = line.rstrip()
-                    if line.endswith(','):
-                        line = line[:-1] + ', "-flto",\n'
-                    elif line.endswith(']'):
-                        line = line[:-1] + ', "-flto"]\n'
-                    else:
-                        line = line + '\n'
+                    content = content[:start] + new_linkflags + content[end:]
                     modified = True
-                
-                new_lines.append(line)
 
             if modified:
                 with open(build_py_path, 'w', encoding='utf-8') as f:
-                    f.writelines(new_lines)
+                    f.write(content)
 
                 print("*** Added LTO flags to pioarduino-build.py ***")
                 print("    CCFLAGS: -flto=auto")