Adding sources to libraries via a plugin

Hi,

I am trying to add sources to libraries added from file system by gradle via  a plugin in IntelliJ. I am using the following code in a startup activity. I can see that the the sources are attached to the library when I print the library on console but I don't see the sources in Project structure in the IDE. Is there anyway to do this automatically via a plugin?

Module @NotNull [] modules = ModuleManager.getInstance(this.project).getModules();
VirtualFileManager vfManager = VirtualFileManager.getInstance();
VirtualFile vf = vfManager.findFileByUrl("file:///path/to/library/sources");
for (var module : modules) {
    ModuleRootManager moduleRootManager = ModuleRootManager.getInstance(module);
    OrderEntry[] entries = moduleRootManager.getOrderEntries();

    for (OrderEntry orderEntry : entries) {
        if (orderEntry instanceof LibraryOrderEntry) {
            LibraryOrderEntry libraryOrderEntry = (LibraryOrderEntry) orderEntry;
            Library library = libraryOrderEntry.getLibrary();
            if (library != null) {
                var libModel = library.getModifiableModel();
                var libTable = library.getTable();

                libModel.addRoot(vf, OrderRootType.SOURCES);
                libModel.commit();
                if (libTable != null) {
                    libTable.getModifiableModel().commit();
                }
            } 
        }
    }
}

Update: This code works and adds the sources but with a few caveats:

1. The added library sources only show up after closing and reopening the project. Is there some way to trigger whatever processing happens on reopening the project from the plugin itself?

2. This works for project level libraries but not for module level libraries. I suspect this is because the library table for module level libraries is null and without committing the library table, the changes to libraries are not reflected. Is there any workaround for this?