Processing import Statements

[thron7]

Processing import Statements

[Luro02]

Spoon doesn't visit CtImport, don't ask me why, but I think it was explained in #5168.

This is the code I am using to visit all imports in a model:

    private static void visitCtCompilationUnit(CtModel ctModel, Consumer<? super CtCompilationUnit> lambda) {
        // it is not possible to visit CtCompilationUnit through the processor API.
        //
        // in https://github.com/INRIA/spoon/issues/5168 the below code is mentioned as a workaround:
        ctModel
            .getAllTypes()
            .stream()
            .map(CtType::getPosition)
            .filter(SourcePosition::isValidPosition)
            .map(SourcePosition::getCompilationUnit)
            // visit each compilation unit only once
            .distinct()
            .forEach(lambda);
    }

    public void visitAllImports(CtModel ctModel) {
        visitCtCompilationUnit(ctModel, ctCompilationUnit -> {
            for (CtImport ctImport : ctCompilationUnit.getImports()) {
                this.checkImport(ctImport, ctCompilationUnit);
            }
        });
    }

    private void checkImport(CtImport ctImport, CtCompilationUnit ctCompilationUnit) {
        // write your code for imports here
    }

[thron7]

Great, thanks for the help.