You've already forked adareducer
mirror of
https://github.com/AdaCore/adareducer.git
synced 2026-02-12 13:10:07 -08:00
ec89ff0cc7
For the benefit of the development environment, or any users who only "git clone" this repository and work from it. Fix the testsuite using this. Rename the "adareducer" source dir to "ada_reducer"
47 lines
1.3 KiB
Python
47 lines
1.3 KiB
Python
import libadalang as lal
|
|
from ada_reducer.types import Buffer
|
|
from ada_reducer.interfaces import ChunkInterface, StrategyInterface
|
|
from ada_reducer.dichotomy import dichotomize
|
|
|
|
|
|
class RemoveClause(ChunkInterface):
|
|
def __init__(self, buffer, node):
|
|
self.buffer = buffer
|
|
self.node = node
|
|
|
|
def do(self):
|
|
num_lines = self.node.sloc_range.end.line - self.node.sloc_range.start.line + 1
|
|
new_text = [""] * num_lines
|
|
self.range, self.lines = self.buffer.replace(self.node.sloc_range, new_text)
|
|
|
|
def undo(self):
|
|
self.buffer.replace(self.range, self.lines)
|
|
|
|
|
|
class RemoveImports(StrategyInterface):
|
|
""" Remove subprograms """
|
|
|
|
def save(self):
|
|
for file in self.buffers:
|
|
self.buffers[file].save()
|
|
|
|
def run_on_file(self, context, file, predicate):
|
|
self.buffers = {file: Buffer(file)}
|
|
unit = context.get_from_file(file)
|
|
|
|
if unit.root is None:
|
|
return
|
|
|
|
chunks = []
|
|
|
|
# First remove all the use clauses that we can, then
|
|
# try with clauses
|
|
|
|
for type in (lal.UsePackageClause, lal.WithClause):
|
|
|
|
for node in unit.root.findall(lambda x: x.is_a(type)):
|
|
# Create a chunk for each clause
|
|
chunks.append(RemoveClause(self.buffers[file], node))
|
|
|
|
dichotomize(chunks, predicate, self.save)
|