001    /**
002     * Copyright (c) 2000-present Liferay, Inc. All rights reserved.
003     *
004     * This library is free software; you can redistribute it and/or modify it under
005     * the terms of the GNU Lesser General Public License as published by the Free
006     * Software Foundation; either version 2.1 of the License, or (at your option)
007     * any later version.
008     *
009     * This library is distributed in the hope that it will be useful, but WITHOUT
010     * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
011     * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
012     * details.
013     */
014    
015    package com.liferay.portal.tools;
016    
017    import com.liferay.portal.kernel.io.unsync.UnsyncStringReader;
018    import com.liferay.portal.kernel.util.StringPool;
019    import com.liferay.portal.kernel.util.StringUtil;
020    import com.liferay.portal.kernel.util.Validator;
021    import com.liferay.portal.kernel.xml.Document;
022    import com.liferay.portal.kernel.xml.Element;
023    import com.liferay.portal.util.FileImpl;
024    import com.liferay.portal.xml.SAXReaderImpl;
025    import com.liferay.util.xml.DocUtil;
026    
027    import com.thoughtworks.qdox.JavaDocBuilder;
028    import com.thoughtworks.qdox.model.AbstractJavaEntity;
029    import com.thoughtworks.qdox.model.DocletTag;
030    import com.thoughtworks.qdox.model.JavaClass;
031    import com.thoughtworks.qdox.model.JavaField;
032    import com.thoughtworks.qdox.model.JavaMethod;
033    import com.thoughtworks.qdox.model.JavaParameter;
034    import com.thoughtworks.qdox.model.Type;
035    
036    import jargs.gnu.CmdLineParser;
037    
038    import java.io.File;
039    import java.io.Reader;
040    
041    import java.util.ArrayList;
042    import java.util.HashMap;
043    import java.util.HashSet;
044    import java.util.List;
045    import java.util.Map;
046    import java.util.Set;
047    import java.util.TreeMap;
048    
049    import org.apache.tools.ant.DirectoryScanner;
050    
051    /**
052     * @author Brian Wing Shun Chan
053     */
054    public class JavadocBuilder {
055    
056            public static void main(String[] args) {
057                    try {
058                            new JavadocBuilder(args);
059                    }
060                    catch (Exception e) {
061                            e.printStackTrace();
062                    }
063            }
064    
065            public JavadocBuilder(String[] args) throws Exception {
066                    CmdLineParser cmdLineParser = new CmdLineParser();
067    
068                    cmdLineParser.parse(args);
069    
070                    CmdLineParser.Option commandOption = cmdLineParser.addStringOption(
071                            "command");
072    
073                    String command = (String)cmdLineParser.getOptionValue(commandOption);
074    
075                    CmdLineParser.Option limitOption = cmdLineParser.addStringOption(
076                            "limit");
077    
078                    String limit = (String)cmdLineParser.getOptionValue(limitOption);
079    
080                    CmdLineParser.Option ignoreAutogeneratedOption =
081                            cmdLineParser.addBooleanOption("ignoreAutogenerated");
082    
083                    Boolean ignoreAutogenerated = (Boolean)cmdLineParser.getOptionValue(
084                            ignoreAutogeneratedOption);
085    
086                    _process(command, limit, ignoreAutogenerated);
087            }
088    
089            private void _addClassCommentElement(
090                    Element rootElement, JavaClass javaClass) {
091    
092                    Element commentElement = rootElement.addElement("comment");
093    
094                    String comment = _getCDATA(javaClass);
095    
096                    if (comment.startsWith("Copyright (c) 2000-present Liferay, Inc.")) {
097                            comment = StringPool.BLANK;
098                    }
099    
100                    if (comment.startsWith(
101                                    "<a href=\"" + javaClass.getName() + ".java.html\">")) {
102    
103                            int pos = comment.indexOf("</a>");
104    
105                            comment = comment.substring(pos + 4).trim();
106                    }
107    
108                    commentElement.addCDATA(comment);
109            }
110    
111            private void _addDocletElements(
112                    Element parentElement, AbstractJavaEntity abstractJavaEntity,
113                    String name) {
114    
115                    DocletTag[] docletTags = abstractJavaEntity.getTagsByName(name);
116    
117                    for (DocletTag docletTag : docletTags) {
118                            String value = docletTag.getValue();
119    
120                            if (name.equals("author") || name.equals("see") ||
121                                    name.equals("since") || name.equals("version")) {
122    
123                                    /*if (value.startsWith("Raymond Aug")) {
124                                            value = "Raymond Aug\u00c3\u00a9";
125                                    }*/
126    
127                                    DocUtil.add(parentElement, name, value);
128                            }
129                            else {
130                                    Element element = parentElement.addElement(name);
131    
132                                    element.addCDATA(value);
133                            }
134                    }
135            }
136    
137            private void _addDocletTags(
138                    Element parentElement, String name, String indent, StringBuilder sb) {
139    
140                    List<Element> elements = parentElement.elements(name);
141    
142                    for (Element element : elements) {
143                            sb.append(indent);
144                            sb.append(" * @");
145                            sb.append(name);
146                            sb.append(" ");
147    
148                            Element commentElement = element.element("comment");
149    
150                            if (commentElement != null) {
151                                    sb.append(element.elementText("name"));
152                                    sb.append(" ");
153                                    sb.append(_getCDATA(element.elementText("comment")));
154                            }
155                            else {
156                                    sb.append(_getCDATA(element.getText()));
157                            }
158    
159                            sb.append("\n");
160                    }
161            }
162    
163            private void _addFieldElement(Element rootElement, JavaField javaField) {
164                    Element fieldElement = rootElement.addElement("field");
165    
166                    DocUtil.add(fieldElement, "name", javaField.getName());
167    
168                    Element commentElement = fieldElement.addElement("comment");
169    
170                    commentElement.addCDATA(_getCDATA(javaField));
171    
172                    _addDocletElements(fieldElement, javaField, "deprecated");
173                    _addDocletElements(fieldElement, javaField, "see");
174                    _addDocletElements(fieldElement, javaField, "since");
175                    _addDocletElements(fieldElement, javaField, "version");
176            }
177    
178            private void _addMethodElement(Element rootElement, JavaMethod javaMethod) {
179                    Element methodElement = rootElement.addElement("method");
180    
181                    DocUtil.add(methodElement, "name", javaMethod.getName());
182    
183                    Element commentElement = methodElement.addElement("comment");
184    
185                    commentElement.addCDATA(_getCDATA(javaMethod));
186    
187                    _addDocletElements(methodElement, javaMethod, "deprecated");
188                    _addParamElements(methodElement, javaMethod);
189                    _addReturnElement(methodElement, javaMethod);
190                    _addDocletElements(methodElement, javaMethod, "see");
191                    _addDocletElements(methodElement, javaMethod, "since");
192                    _addThrowsElements(methodElement, javaMethod);
193                    _addDocletElements(methodElement, javaMethod, "version");
194            }
195    
196            private void _addParamElement(
197                    Element methodElement, JavaParameter javaParameter,
198                    DocletTag[] paramDocletTags) {
199    
200                    String name = javaParameter.getName();
201                    String type = javaParameter.getType().getValue();
202                    String value = null;
203    
204                    for (DocletTag paramDocletTag : paramDocletTags) {
205                            String curValue = paramDocletTag.getValue();
206    
207                            if (!curValue.startsWith(name)) {
208                                    continue;
209                            }
210                            else {
211                                    curValue = value;
212    
213                                    break;
214                            }
215                    }
216    
217                    Element paramElement = methodElement.addElement("param");
218    
219                    DocUtil.add(paramElement, "name", name);
220                    DocUtil.add(paramElement, "type", type);
221    
222                    if (value != null) {
223                            value = value.substring(name.length());
224                    }
225    
226                    Element commentElement = paramElement.addElement("comment");
227    
228                    commentElement.addCDATA(_getCDATA(value));
229            }
230    
231            private void _addParamElements(
232                    Element methodElement, JavaMethod javaMethod) {
233    
234                    JavaParameter[] javaParameters = javaMethod.getParameters();
235    
236                    DocletTag[] paramDocletTags = javaMethod.getTagsByName("param");
237    
238                    for (JavaParameter javaParameter : javaParameters) {
239                            _addParamElement(methodElement, javaParameter, paramDocletTags);
240                    }
241            }
242    
243            private void _addReturnElement(
244                    Element methodElement, JavaMethod javaMethod) {
245    
246                    Type returns = javaMethod.getReturns();
247    
248                    if ((returns == null) || returns.getValue().equals("void")) {
249                            return;
250                    }
251    
252                    _addDocletElements(methodElement, javaMethod, "return");
253            }
254    
255            private void _addThrowsElement(
256                    Element methodElement, Type exception, DocletTag[] throwsDocletTags) {
257    
258                    String name = exception.getJavaClass().getName();
259                    String value = null;
260    
261                    for (DocletTag throwsDocletTag : throwsDocletTags) {
262                            String curValue = throwsDocletTag.getValue();
263    
264                            if (!curValue.startsWith(name)) {
265                                    continue;
266                            }
267                            else {
268                                    curValue = value;
269    
270                                    break;
271                            }
272                    }
273    
274                    Element throwsElement = methodElement.addElement("throws");
275    
276                    DocUtil.add(throwsElement, "name", name);
277                    DocUtil.add(throwsElement, "type", exception.getValue());
278    
279                    if (value != null) {
280                            value = value.substring(name.length());
281                    }
282    
283                    Element commentElement = throwsElement.addElement("comment");
284    
285                    commentElement.addCDATA(_getCDATA(value));
286            }
287    
288            private void _addThrowsElements(
289                    Element methodElement, JavaMethod javaMethod) {
290    
291                    Type[] exceptions = javaMethod.getExceptions();
292    
293                    DocletTag[] throwsDocletTags = javaMethod.getTagsByName("throws");
294    
295                    for (Type exception : exceptions) {
296                            _addThrowsElement(methodElement, exception, throwsDocletTags);
297                    }
298            }
299    
300            private String _getCDATA(AbstractJavaEntity abstractJavaEntity) {
301                    return _getCDATA(abstractJavaEntity.getComment());
302            }
303    
304            private String _getCDATA(String cdata) {
305                    if (cdata == null) {
306                            return StringPool.BLANK;
307                    }
308    
309                    cdata = StringUtil.replace(
310                            cdata, new String[] {"\n", "<p>", "</p>"},
311                            new String[] {" ", " <p> ", " </p> "});
312    
313                    while (cdata.contains("  ")) {
314                            cdata = StringUtil.replace(cdata, "  ", " ");
315                    }
316    
317                    return cdata.trim();
318            }
319    
320            private String _getFieldKey(Element fieldElement) {
321                    return fieldElement.elementText("name");
322            }
323    
324            private String _getFieldKey(JavaField javaField) {
325                    return javaField.getName();
326            }
327    
328            private JavaClass _getJavaClass(String fileName) throws Exception {
329                    return _getJavaClass(fileName, null);
330            }
331    
332            private JavaClass _getJavaClass(String fileName, Reader reader)
333                    throws Exception {
334    
335                    int pos = fileName.indexOf("src/");
336    
337                    if (pos == -1) {
338                            pos = fileName.indexOf("test/");
339                    }
340    
341                    if (pos == -1) {
342                            throw new RuntimeException(fileName);
343                    }
344    
345                    pos = fileName.indexOf("/", pos);
346    
347                    String srcFile = fileName.substring(pos + 1);
348                    String className = StringUtil.replace(
349                            srcFile.substring(0, srcFile.length() - 5), "/", ".");
350    
351                    JavaDocBuilder builder = new JavaDocBuilder();
352    
353                    if (reader == null) {
354                            File file = new File(fileName);
355    
356                            if (!file.exists()) {
357                                    return null;
358                            }
359    
360                            builder.addSource(file);
361                    }
362                    else {
363                            builder.addSource(reader);
364                    }
365    
366                    return builder.getClassByName(className);
367            }
368    
369            private String _getJavaClassComment(Element rootElement) {
370                    StringBuilder sb = new StringBuilder();
371    
372                    sb.append("/**\n");
373                    sb.append(" * ");
374                    sb.append(_getCDATA(rootElement.elementText("comment")));
375                    sb.append("\n");
376                    sb.append(" *\n");
377    
378                    String indent = StringPool.BLANK;
379    
380                    _addDocletTags(rootElement, "author", indent, sb);
381                    _addDocletTags(rootElement, "deprecated", indent, sb);
382                    _addDocletTags(rootElement, "see", indent, sb);
383                    _addDocletTags(rootElement, "serial", indent, sb);
384                    _addDocletTags(rootElement, "since", indent, sb);
385                    _addDocletTags(rootElement, "version", indent, sb);
386    
387                    sb.append(" */\n");
388    
389                    return sb.toString();
390            }
391    
392            private String _getJavadocXml(JavaClass javaClass) throws Exception {
393                    Element rootElement = _saxReaderUtil.createElement("javadoc");
394    
395                    Document document = _saxReaderUtil.createDocument(rootElement);
396    
397                    DocUtil.add(rootElement, "name", javaClass.getName());
398                    DocUtil.add(rootElement, "type", javaClass.getFullyQualifiedName());
399    
400                    _addClassCommentElement(rootElement, javaClass);
401                    _addDocletElements(rootElement, javaClass, "author");
402                    _addDocletElements(rootElement, javaClass, "deprecated");
403                    _addDocletElements(rootElement, javaClass, "see");
404                    _addDocletElements(rootElement, javaClass, "serial");
405                    _addDocletElements(rootElement, javaClass, "since");
406                    _addDocletElements(rootElement, javaClass, "version");
407    
408                    JavaMethod[] javaMethods = javaClass.getMethods();
409    
410                    for (JavaMethod javaMethod : javaMethods) {
411                            _addMethodElement(rootElement, javaMethod);
412                    }
413    
414                    JavaField[] javaFields = javaClass.getFields();
415    
416                    for (JavaField javaField : javaFields) {
417                            _addFieldElement(rootElement, javaField);
418                    }
419    
420                    return document.formattedString();
421            }
422    
423            private String _getJavaFieldComment(
424                    String[] lines, Map<String, Element> fieldElementsMap,
425                    JavaField javaField) {
426    
427                    String fieldKey = _getFieldKey(javaField);
428    
429                    Element fieldElement = fieldElementsMap.get(fieldKey);
430    
431                    if (fieldElement == null) {
432                            return null;
433                    }
434    
435                    String line = lines[javaField.getLineNumber() - 1];
436    
437                    String indent = StringPool.BLANK;
438    
439                    for (char c : line.toCharArray()) {
440                            if (Character.isWhitespace(c)) {
441                                    indent += c;
442                            }
443                            else {
444                                    break;
445                            }
446                    }
447    
448                    StringBuilder sb = new StringBuilder();
449    
450                    sb.append(indent);
451                    sb.append("/**\n");
452                    sb.append(indent);
453                    sb.append(" * ");
454                    sb.append(fieldElement.elementText("comment"));
455                    sb.append("\n");
456                    sb.append(indent);
457                    sb.append(" *\n");
458    
459                    _addDocletTags(fieldElement, "deprecated", indent, sb);
460                    _addDocletTags(fieldElement, "see", indent, sb);
461                    _addDocletTags(fieldElement, "since", indent, sb);
462                    _addDocletTags(fieldElement, "version", indent, sb);
463    
464                    sb.append(indent);
465                    sb.append(" */\n");
466    
467                    return sb.toString();
468            }
469    
470            private String _getJavaMethodComment(
471                    String[] lines, Map<String, Element> methodElementsMap,
472                    JavaMethod javaMethod) {
473    
474                    String methodKey = _getMethodKey(javaMethod);
475    
476                    Element methodElement = methodElementsMap.get(methodKey);
477    
478                    if (methodElement == null) {
479                            return null;
480                    }
481    
482                    String line = lines[javaMethod.getLineNumber() - 1];
483    
484                    String indent = StringPool.BLANK;
485    
486                    for (char c : line.toCharArray()) {
487                            if (Character.isWhitespace(c)) {
488                                    indent += c;
489                            }
490                            else {
491                                    break;
492                            }
493                    }
494    
495                    StringBuilder sb = new StringBuilder();
496    
497                    sb.append(indent);
498                    sb.append("/**\n");
499                    sb.append(indent);
500                    sb.append(" * ");
501                    sb.append(methodElement.elementText("comment"));
502                    sb.append("\n");
503                    sb.append(indent);
504                    sb.append(" *\n");
505    
506                    _addDocletTags(methodElement, "deprecated", indent, sb);
507                    _addDocletTags(methodElement, "param", indent, sb);
508                    _addDocletTags(methodElement, "return", indent, sb);
509                    _addDocletTags(methodElement, "see", indent, sb);
510                    _addDocletTags(methodElement, "since", indent, sb);
511                    _addDocletTags(methodElement, "throws", indent, sb);
512                    _addDocletTags(methodElement, "version", indent, sb);
513    
514                    sb.append(indent);
515                    sb.append(" */\n");
516    
517                    return sb.toString();
518            }
519    
520            private String _getMethodKey(Element methodElement) {
521                    StringBuilder sb = new StringBuilder();
522    
523                    sb.append(methodElement.elementText("name"));
524                    sb.append(StringPool.OPEN_PARENTHESIS);
525    
526                    List<Element> paramElements = methodElement.elements("param");
527    
528                    for (Element paramElement : paramElements) {
529                            sb.append(paramElement.elementText("name"));
530                            sb.append("|");
531                            sb.append(paramElement.elementText("type"));
532                            sb.append(",");
533                    }
534    
535                    sb.append(StringPool.CLOSE_PARENTHESIS);
536    
537                    return sb.toString();
538            }
539    
540            private String _getMethodKey(JavaMethod javaMethod) {
541                    StringBuilder sb = new StringBuilder();
542    
543                    sb.append(javaMethod.getName());
544                    sb.append(StringPool.OPEN_PARENTHESIS);
545    
546                    JavaParameter[] javaParameters = javaMethod.getParameters();
547    
548                    for (JavaParameter javaParameter : javaParameters) {
549                            sb.append(javaParameter.getName());
550                            sb.append("|");
551                            sb.append(javaParameter.getType().getValue());
552                            sb.append(",");
553                    }
554    
555                    sb.append(StringPool.CLOSE_PARENTHESIS);
556    
557                    return sb.toString();
558            }
559    
560            private boolean _isGenerated(String content) {
561                    if (content.contains("<javadoc autogenerated=\"true\">")) {
562                            return true;
563                    }
564                    else {
565                            return false;
566                    }
567            }
568    
569            private void _process(
570                            String command, String limit, Boolean ignoreAutogenerated)
571                    throws Exception {
572    
573                    DirectoryScanner ds = new DirectoryScanner();
574    
575                    ds.setBasedir(_basedir);
576                    ds.setExcludes(
577                            new String[] {
578                                    "**\\classes\\**", "**\\portal-client\\**", "**\\portal-web\\**"
579                            });
580    
581                    List<String> includes = new ArrayList<String>();
582    
583                    if (Validator.isNotNull(limit) && !limit.startsWith("$")) {
584                            String[] limitArray = StringUtil.split(limit, '/');
585    
586                            for (String curLimit : limitArray) {
587                                    includes.add(
588                                            "**\\" + StringUtil.replace(curLimit, ".", "\\") +
589                                                    "\\**\\*.java");
590                                    includes.add("**\\" + curLimit + ".java");
591                            }
592                    }
593                    else {
594                            includes.add("**\\*.java");
595                    }
596    
597                    ds.setIncludes(includes.toArray(new String[includes.size()]));
598    
599                    ds.scan();
600    
601                    String[] fileNames = ds.getIncludedFiles();
602    
603                    for (String fileName : fileNames) {
604                            fileName = StringUtil.replace(fileName, "\\", "/");
605    
606                            /*if (!fileName.endsWith("Isolation.java")) {
607                                    continue;
608                            }*/
609    
610                            if ((ignoreAutogenerated != null) &&
611                                    ignoreAutogenerated.booleanValue()) {
612    
613                                    File file = new File(_basedir + fileName);
614    
615                                    if (file.exists()) {
616                                            String oldContent = _fileUtil.read(
617                                                    _basedir + fileName + "doc");
618    
619                                            if (_isGenerated(oldContent)) {
620                                                    continue;
621                                            }
622                                    }
623                            }
624    
625                            if (command.equals("cleanup")) {
626                                    _processGet(fileName);
627                                    _processSave(fileName);
628                                    _processDelete(fileName);
629                            }
630                            else if (command.equals("commit")) {
631                                    _processSave(fileName);
632                                    _processDelete(fileName);
633                            }
634                            else if (command.equals("delete")) {
635                                    _processDelete(fileName);
636                            }
637                            else if (command.equals("get")) {
638                                    _processGet(fileName);
639                            }
640                            else if (command.equals("save")) {
641                                    _processSave(fileName);
642                            }
643                    }
644            }
645    
646            private void _processDelete(String fileName) throws Exception {
647                    _removeJavadocFromJava(fileName, true);
648            }
649    
650            private void _processGet(String fileName) throws Exception {
651                    File javadocFile = new File(_basedir + fileName + "doc");
652    
653                    if (!javadocFile.exists()) {
654                            _updateJavadocFromJava(fileName);
655                    }
656    
657                    String javaWithoutJavadoc = _removeJavadocFromJava(fileName, false);
658    
659                    _updateJavaFromJavadoc(fileName, javaWithoutJavadoc);
660            }
661    
662            private void _processSave(String fileName) throws Exception {
663                    _updateJavadocFromJava(fileName);
664            }
665    
666            private String _removeJavadocFromJava(String fileName, boolean log)
667                    throws Exception {
668    
669                    File file = new File(_basedir + fileName);
670    
671                    String oldContent = _fileUtil.read(file);
672    
673                    String[] lines = StringUtil.splitLines(oldContent);
674    
675                    JavaClass javaClass = _getJavaClass(
676                            fileName, new UnsyncStringReader(oldContent));
677    
678                    Set<Integer> lineNumbers = new HashSet<Integer>();
679    
680                    lineNumbers.add(javaClass.getLineNumber());
681    
682                    JavaMethod[] javaMethods = javaClass.getMethods();
683    
684                    for (JavaMethod javaMethod : javaMethods) {
685                            lineNumbers.add(javaMethod.getLineNumber());
686                    }
687    
688                    JavaField[] javaFields = javaClass.getFields();
689    
690                    for (JavaField javaField : javaFields) {
691                            lineNumbers.add(javaField.getLineNumber());
692                    }
693    
694                    for (int lineNumber : lineNumbers) {
695                            int pos = lineNumber - 2;
696    
697                            String line = lines[pos].trim();
698    
699                            if (line.endsWith("*/")) {
700                                    while (true) {
701                                            lines[pos] = null;
702    
703                                            if (line.startsWith("/**")) {
704                                                    break;
705                                            }
706    
707                                            line = lines[--pos].trim();
708                                    }
709                            }
710                    }
711    
712                    StringBuilder sb = new StringBuilder(oldContent.length());
713    
714                    for (String line : lines) {
715                            if (line != null) {
716                                    sb.append(line);
717                                    sb.append("\n");
718                            }
719                    }
720    
721                    String newContent = sb.toString().trim();
722    
723                    if ((oldContent == null) || !oldContent.equals(newContent)) {
724                            _fileUtil.write(file, newContent);
725    
726                            if (log) {
727                                    System.out.println("Writing " + file);
728                            }
729                    }
730    
731                    return newContent;
732            }
733    
734            private void _updateJavadocFromJava(String fileName) throws Exception {
735                    File file = new File(_basedir + fileName + "doc");
736    
737                    String oldContent = null;
738    
739                    if (file.exists()) {
740                            oldContent = _fileUtil.read(file);
741    
742                            if (_isGenerated(oldContent)) {
743                                    return;
744                            }
745                    }
746    
747                    JavaClass javaClass = _getJavaClass(fileName);
748    
749                    String newContent = _getJavadocXml(javaClass);
750    
751                    if ((oldContent == null) || !oldContent.equals(newContent)) {
752                            _fileUtil.write(file, newContent.getBytes());
753    
754                            System.out.println("Writing " + file);
755                    }
756            }
757    
758            private void _updateJavaFromJavadoc(String fileName, String oldContent)
759                    throws Exception {
760    
761                    File javadocFile = new File(_basedir + fileName + "doc");
762    
763                    if (!javadocFile.exists()) {
764                            return;
765                    }
766    
767                    File file = new File(_basedir + fileName);
768    
769                    if (oldContent == null) {
770                            oldContent = _fileUtil.read(file);
771                    }
772    
773                    String[] lines = StringUtil.splitLines(oldContent);
774    
775                    JavaClass javaClass = _getJavaClass(
776                            fileName, new UnsyncStringReader(oldContent));
777    
778                    Document document = _saxReaderUtil.read(javadocFile);
779    
780                    Element rootElement = document.getRootElement();
781    
782                    Map<Integer, String> commentsMap = new TreeMap<Integer, String>();
783    
784                    commentsMap.put(
785                            javaClass.getLineNumber(), _getJavaClassComment(rootElement));
786    
787                    Map<String, Element> methodElementsMap = new HashMap<String, Element>();
788    
789                    List<Element> methodElements = rootElement.elements("method");
790    
791                    for (Element methodElement : methodElements) {
792                            String methodKey = _getMethodKey(methodElement);
793    
794                            methodElementsMap.put(methodKey, methodElement);
795                    }
796    
797                    JavaMethod[] javaMethods = javaClass.getMethods();
798    
799                    for (JavaMethod javaMethod : javaMethods) {
800                            if (commentsMap.containsKey(javaMethod.getLineNumber())) {
801                                    continue;
802                            }
803    
804                            commentsMap.put(
805                                    javaMethod.getLineNumber(),
806                                    _getJavaMethodComment(lines, methodElementsMap, javaMethod));
807                    }
808    
809                    Map<String, Element> fieldElementsMap = new HashMap<String, Element>();
810    
811                    List<Element> fieldElements = rootElement.elements("field");
812    
813                    for (Element fieldElement : fieldElements) {
814                            String fieldKey = _getFieldKey(fieldElement);
815    
816                            fieldElementsMap.put(fieldKey, fieldElement);
817                    }
818    
819                    JavaField[] javaFields = javaClass.getFields();
820    
821                    for (JavaField javaField : javaFields) {
822                            if (commentsMap.containsKey(javaField.getLineNumber())) {
823                                    continue;
824                            }
825    
826                            commentsMap.put(
827                                    javaField.getLineNumber(),
828                                    _getJavaFieldComment(lines, fieldElementsMap, javaField));
829                    }
830    
831                    StringBuilder sb = new StringBuilder(oldContent.length());
832    
833                    for (int lineNumber = 1; lineNumber <= lines.length; lineNumber++) {
834                            String line = lines[lineNumber - 1];
835    
836                            String comments = commentsMap.get(lineNumber);
837    
838                            if (comments != null) {
839                                    sb.append(comments);
840                            }
841    
842                            sb.append(line);
843                            sb.append("\n");
844                    }
845    
846                    String newContent = sb.toString().trim();
847    
848                    if ((oldContent == null) || !oldContent.equals(newContent)) {
849                            _fileUtil.write(file, newContent);
850    
851                            System.out.println("Writing " + file);
852                    }
853            }
854    
855            private static FileImpl _fileUtil = FileImpl.getInstance();
856            private static SAXReaderImpl _saxReaderUtil = SAXReaderImpl.getInstance();
857    
858            private String _basedir = "./";
859    
860    }