com.yworks.compiler.FileObjects Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of yguard Show documentation

The open-source Java obfuscation tool working with Ant and Gradle by yWorks - the diagramming experts

There is a newer version: 4.1.1

package com.yworks.compiler;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.Reader;
import java.net.URI;
import java.net.URL;

import javax.tools.JavaFileObject;
import javax.tools.SimpleJavaFileObject;

/**
 * Provides factory methods for {@link javax.tools.JavaFileObject} instances.
 * @author Thomas Behr
 */
class FileObjects {
  private FileObjects() {
  }

  static JavaFileObject newInMemoryFileObject(
          final String typeName, final String code
  ) {
    return new InMemoryFileObject(typeName, code);
  }

  static JavaFileObject newUrlFileObject(
          final String typeName, final URL url
  ) {
    return new UrlFileObject(typeName, url);
  }



  private abstract static class AbstractSourceObject extends SimpleJavaFileObject {
    AbstractSourceObject( final String typname) {
      super(asUri(typname), Kind.SOURCE);
    }

    static URI asUri( final String typeName ) {
      return URI.create("string:///" + typeName.replace('.', '/') + Kind.SOURCE.extension);
    }
  }

  private static final class InMemoryFileObject extends AbstractSourceObject {
    private final String code;

    InMemoryFileObject( final String typeName, final String code ) {
      super(typeName);
      this.code = code;
    }

    public CharSequence getCharContent(
            final boolean ignoreEncodingErrors
    ) throws IOException {
      return code;
    }
  }

  private static final class UrlFileObject extends AbstractSourceObject {
    private final URL url;

    UrlFileObject( final String typname, final URL url ) {
      super(typname);
      this.url = url;
    }

    public CharSequence getCharContent(
            final boolean ignoreEncodingErrors
    ) throws IOException {
      final StringBuffer sb = new StringBuffer();

      final int[] tail = new int[1];
      final BufferedReader br = new BufferedReader(
              Streams.newTail(openReader(ignoreEncodingErrors), tail));
      try {
        for (String line = br.readLine(); line != null; line = br.readLine()) {
          sb.append(line).append('\n');
        }
      } finally {
        br.close();
      }

      if (sb.length() > 0 && tail[0] != '\n') {
        sb.setLength(sb.length() - 1);
      }

      return sb.toString();
    }

    public Reader openReader(
            final boolean ignoreEncodingErrors
    ) throws IOException {
      return new InputStreamReader(url.openStream(), "UTF-8");
    }
  }
}