io.moderne.ai.SpellCheckCommentsInFrench Maven / Gradle / Ivy
/*
* Copyright 2021 the original author or authors.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* https://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package io.moderne.ai;
import lombok.EqualsAndHashCode;
import lombok.Value;
import org.openrewrite.ExecutionContext;
import org.openrewrite.Recipe;
import org.openrewrite.TreeVisitor;
import org.openrewrite.internal.ListUtils;
import org.openrewrite.java.JavaIsoVisitor;
import org.openrewrite.java.JavadocVisitor;
import org.openrewrite.java.tree.Javadoc;
import org.openrewrite.java.tree.Space;
import org.openrewrite.java.tree.TextComment;
@Value
@EqualsAndHashCode(callSuper = false)
public class SpellCheckCommentsInFrench extends Recipe {
@Override
public String getDisplayName() {
return "Fix mis-encoded comments in French";
}
@Override
public String getDescription() {
return "Use spellchecker to fix mis-encoded French comments. Mis-encoded comments will contain either '?' or '�'.";
}
@Override
public TreeVisitor getVisitor() {
return new JavaIsoVisitor() {
@Override
protected JavadocVisitor getJavadocVisitor() {
return new JavadocVisitor(this) {
@Override
public Javadoc visitDocComment(Javadoc.DocComment javadoc, ExecutionContext ctx) {
Javadoc.DocComment dc = (Javadoc.DocComment) super.visitDocComment(javadoc, ctx);
dc = dc.withBody(ListUtils.map(dc.getBody(), docLine -> {
if (docLine instanceof Javadoc.Text) {
String commentText = ((Javadoc.Text) docLine).getText();
if (!commentText.trim().isEmpty() && "fr".equals(LanguageDetectorModelClient.getInstance()
.getLanguage(commentText).getLanguage())) {
String fixedComment = SpellCheckerClient.getInstance().getCommentGradio(commentText);
if (!fixedComment.equals(commentText)) {
docLine = ((Javadoc.Text) docLine).withText(fixedComment);
}
}
}
return docLine;
}));
return dc;
}
};
}
@Override
public Space visitSpace(Space space, Space.Location loc, ExecutionContext ctx) {
Space s = super.visitSpace(space, loc, ctx);
return s.withComments(ListUtils.map(s.getComments(), c -> {
if (c instanceof TextComment) {
TextComment tc = (TextComment) c;
String commentText = tc.getText();
if (!commentText.isEmpty() && "fr".equals(LanguageDetectorModelClient.getInstance()
.getLanguage(commentText).getLanguage())) {
String fixedComment = SpellCheckerClient.getInstance().getCommentGradio(commentText);
if (!fixedComment.equals(commentText)) {
return tc.withText(fixedComment);
}
}
}
return c;
}));
}
};
}
}