protected ScriptOperationContext wrapContext(ScriptOperationContext ctx) {
for (String entryId : ctx.keySet()) {
Object entry = ctx.get(entryId);
if (entry instanceof DocumentModel) {
ctx.put(entryId, new DocumentWrapper(ctx.getCoreSession(), (DocumentModel) entry));
}
if (entry instanceof DocumentModelList) {
List<DocumentWrapper> docs = new ArrayList<>();
for (DocumentModel doc : (DocumentModelList) entry) {
docs.add(new DocumentWrapper(ctx.getCoreSession(), doc));
}
ctx.put(entryId, docs);
}
}
return ctx;
}
