/**
* Create a new method that will override the parent.
*
* @param function The method that you want ot override.
*/
private void addNewMethod(Function function) {
String fnNameFormat = "%s.prototype." + function.getName();
String methodPrototype = String.format(fnNameFormat, finder.getCurrentNamespace());
String parentMethodPrototype = String.format(fnNameFormat, finder.getParentNamespace());
String jsDoc = function.getExtendedJsDoc();
String arguments = function.getArguments();
String callArguments = arguments;
if (callArguments.trim().length() > 0) {
callArguments = ", " + callArguments;
}
final String methodTemplate =
String.format(
"%s\n%s = function(%s) {\n" + " %s.call(this%s);\n" + "};\n",
jsDoc, methodPrototype, arguments, parentMethodPrototype, callArguments);
runWriteActionInsideCommand(
project,
"Override method",
new Runnable() {
@Override
public void run() {
int offset = editor.getCaretModel().getOffset();
document.replaceString(offset, offset, methodTemplate);
}
});
}
@Override
public void actionPerformed(AnActionEvent actionEvent) {
if (!canEnableAction(actionEvent)) {
return;
}
project = actionEvent.getData(PlatformDataKeys.PROJECT);
editor = actionEvent.getData(PlatformDataKeys.EDITOR);
if (project == null || editor == null) {
return;
}
document = editor.getDocument();
finder = new ParentNamespaceFinder(document, project);
finder.addResultsReadyListener(
new ChangeListener() {
@Override
public void stateChanged(ChangeEvent changeEvent) {
if (changeEvent.getSource().equals("ParentNamespaceFound")) {
// TODO: show not found.
showDialog();
}
}
});
finder.findParentClass();
}
/** Show the dialog to select the method to override. */
private void showDialog() {
List<Function> functionNames = finder.getFunctionNames();
// Select the closest element found from the current position.
final JBList jbList = new JBList(functionNames.toArray());
// Open a pop-up to select which describe() or it() you want to change.
JBPopupFactory.getInstance()
.createListPopupBuilder(jbList)
.setTitle("Select the method to override")
.setFilteringEnabled(
new com.intellij.util.Function<Object, String>() {
@Override
public String fun(Object o) {
return o == null ? "" : o.toString();
}
})
.setItemChoosenCallback(
new Runnable() {
public void run() {
if (jbList.getSelectedValue() != null) {
addNewMethod((Function) jbList.getSelectedValue());
}
}
})
.createPopup()
.showCenteredInCurrentWindow(project);
}
