refactor: change repo layout

Assets/CodeGen/EnumsMigration/CommentsFinder.java

@@ -0,0 +1,51 @@
+import java.util.List;
+
+import org.antlr.v4.runtime.*;
+
+public class CommentsFinder {
+
+	enum CommentsPosition {
+		Unknown, Precomment, Postcomment
+	}
+
+	public CommentsFinder(BufferedTokenStream tokenStream) {
+		mTokenStream = tokenStream;
+		mCommentsPos = CommentsPosition.Unknown;
+	}
+
+	private BufferedTokenStream mTokenStream;
+	private CommentsPosition mCommentsPos;
+
+	public String getComment(Token preToken, Token postToken) {
+		switch (mCommentsPos) {
+		case Unknown: {
+			// if we don't know where is our token,
+			// we should assume it is from precomment to postcomment
+			// and check it.
+			List<Token> precomment = CommonHelper.getPreChannelTokens(mTokenStream, preToken, CKGeneralLexer.COMMENTS);
+			if (precomment != null) {
+				mCommentsPos = CommentsPosition.Precomment;
+				return CommonHelper.cutComments(precomment);
+			}
+			
+			List<Token> postcomment = CommonHelper.getPostChannelTokens(mTokenStream, postToken, CKGeneralLexer.COMMENTS);
+			if (postcomment != null) {
+				mCommentsPos = CommentsPosition.Postcomment;
+				return CommonHelper.cutComments(postcomment);
+			}
+			
+			// really do not have comment, return empty and keep comment pos
+			return null;
+		}
+		case Precomment: {
+			return CommonHelper.cutComments(CommonHelper.getPreChannelTokens(mTokenStream, preToken, CKGeneralLexer.COMMENTS));
+		}
+		case Postcomment: {
+			return CommonHelper.cutComments(CommonHelper.getPostChannelTokens(mTokenStream, postToken, CKGeneralLexer.COMMENTS));
+		}
+		default:
+			return null;
+		}
+	}
+
+}