PsiElement root = PsiTreeUtil.getChildOfType(psiFile, GherkinFeature.class);
if (root == null) {
root = psiFile;
}
return
new StructureViewModelBase(psiFile, editor, new GherkinStructureViewElement(root))
.withSuitableClasses(GherkinFile.class, GherkinFeature.class, GherkinStepsHolder.class, GherkinStep.class);
}
};
}