/*

 * Copyright (c) 2012, Oracle and/or its affiliates. All rights reserved.

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.  Oracle designates this

 * particular file as subject to the "Classpath" exception as provided

 * by Oracle in the LICENSE file that accompanied this code.

 *

 * This code is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

package com.sun.tools.jdeps;

import com.sun.tools.classfile.ClassFile;

import com.sun.tools.classfile.ConstantPoolException;

import com.sun.tools.classfile.Dependencies.ClassFileError;

import java.io.*;

import java.nio.file.FileVisitResult;

import java.nio.file.Files;

import java.nio.file.Path;

import java.nio.file.SimpleFileVisitor;

import java.nio.file.attribute.BasicFileAttributes;

import java.util.*;

import java.util.jar.JarEntry;

import java.util.jar.JarFile;

/**

 * ClassFileReader reads ClassFile(s) of a given path that can be

 * a .class file, a directory, or a JAR file.

 */

public class ClassFileReader {

    /**

     * Returns a ClassFileReader instance of a given path.

     */

    public static ClassFileReader newInstance(Path path) throws IOException {

        if (!Files.exists(path)) {

            throw new FileNotFoundException(path.toString());

        }

        if (Files.isDirectory(path)) {

            return new DirectoryReader(path);

        } else if (path.getFileName().toString().endsWith(".jar")) {

            return new JarFileReader(path);

        } else {

            return new ClassFileReader(path);

        }

    }

    /**

     * Returns a ClassFileReader instance of a given JarFile.

     */

    public static ClassFileReader newInstance(Path path, JarFile jf) throws IOException {

        return new JarFileReader(path, jf);

    }

    protected final Path path;

    protected final String baseFileName;

    private ClassFileReader(Path path) {

        this.path = path;

        this.baseFileName = path.getFileName() != null

                                ? path.getFileName().toString()

                                : path.toString();

    }

    public String getFileName() {

        return baseFileName;

    }

    /**

     * Returns the ClassFile matching the given binary name

     * or a fully-qualified class name.

     */

    public ClassFile getClassFile(String name) throws IOException {

        if (name.indexOf('.') > 0) {

            int i = name.lastIndexOf('.');

            String pathname = name.replace('.', File.separatorChar) + ".class";

            if (baseFileName.equals(pathname) ||

                    baseFileName.equals(pathname.substring(0, i) + "$" +

                                        pathname.substring(i+1, pathname.length()))) {

                return readClassFile(path);

            }

        } else {

            if (baseFileName.equals(name.replace('/', File.separatorChar) + ".class")) {

                return readClassFile(path);

            }

        }

        return null;

    }

    public Iterable<ClassFile> getClassFiles() throws IOException {

        return new Iterable<ClassFile>() {

            public Iterator<ClassFile> iterator() {

                return new FileIterator();

            }

        };

    }

    protected ClassFile readClassFile(Path p) throws IOException {

        InputStream is = null;

        try {

            is = Files.newInputStream(p);

            return ClassFile.read(is);

        } catch (ConstantPoolException e) {

            throw new ClassFileError(e);

        } finally {

            if (is != null) {

                is.close();

            }

        }

    }

    class FileIterator implements Iterator<ClassFile> {

        int count;

        FileIterator() {

            this.count = 0;

        }

        public boolean hasNext() {

            return count == 0 && baseFileName.endsWith(".class");

        }

        public ClassFile next() {

            if (!hasNext()) {

                throw new NoSuchElementException();

            }

            try {

                ClassFile cf = readClassFile(path);

                count++;

                return cf;

            } catch (IOException e) {

                throw new ClassFileError(e);

            }

        }

        public void remove() {

            throw new UnsupportedOperationException("Not supported yet.");

        }

    }

    public String toString() {

        return path.toString();

    }

    private static class DirectoryReader extends ClassFileReader {

        DirectoryReader(Path path) throws IOException {

            super(path);

        }

        public ClassFile getClassFile(String name) throws IOException {

            if (name.indexOf('.') > 0) {

                int i = name.lastIndexOf('.');

                String pathname = name.replace('.', File.separatorChar) + ".class";

                Path p = path.resolve(pathname);

                if (!Files.exists(p)) {

                    p = path.resolve(pathname.substring(0, i) + "$" +

                                     pathname.substring(i+1, pathname.length()));

                }

                if (Files.exists(p)) {

                    return readClassFile(p);

                }

            } else {

                Path p = path.resolve(name + ".class");

                if (Files.exists(p)) {

                    return readClassFile(p);

                }

            }

            return null;

        }

        public Iterable<ClassFile> getClassFiles() throws IOException {

            final Iterator<ClassFile> iter = new DirectoryIterator();

            return new Iterable<ClassFile>() {

                public Iterator<ClassFile> iterator() {

                    return iter;

                }

            };

        }

        private List<Path> walkTree(Path dir) throws IOException {

            Files.walkFileTree(dir, new SimpleFileVisitor<Path>() {

                public FileVisitResult visitFile(Path file, BasicFileAttributes attrs)

                        throws IOException {

                    if (file.getFileName().toString().endsWith(".class")) {

                        files.add(file);

                    }

                    return FileVisitResult.CONTINUE;

                }

            });

            return files;

        }

        class DirectoryIterator implements Iterator<ClassFile> {

            private List<Path> entries;

            private int index = 0;

            DirectoryIterator() throws IOException {

                entries = walkTree(path);

                index = 0;

            }

            public boolean hasNext() {

                return index != entries.size();

            }

            public ClassFile next() {

                if (!hasNext()) {

                    throw new NoSuchElementException();

                }

                Path path = entries.get(index++);

                try {

                    return readClassFile(path);

                } catch (IOException e) {

                    throw new ClassFileError(e);

                }

            }

            public void remove() {

                throw new UnsupportedOperationException("Not supported yet.");

            }

        }

    }

    private static class JarFileReader extends ClassFileReader {

        final JarFile jarfile;

        JarFileReader(Path path) throws IOException {

            this(path, new JarFile(path.toFile()));

        }

        JarFileReader(Path path, JarFile jf) throws IOException {

            super(path);

            this.jarfile = jf;

        }

        public ClassFile getClassFile(String name) throws IOException {

            if (name.indexOf('.') > 0) {

                int i = name.lastIndexOf('.');

                String entryName = name.replace('.', '/') + ".class";

                JarEntry e = jarfile.getJarEntry(entryName);

                if (e == null) {

                    e = jarfile.getJarEntry(entryName.substring(0, i) + "$"

                            + entryName.substring(i + 1, entryName.length()));

                }

                if (e != null) {

                    return readClassFile(e);

                }

            } else {

                JarEntry e = jarfile.getJarEntry(name + ".class");

                if (e != null) {

                    return readClassFile(e);

                }

            }

            return null;

        }

        private ClassFile readClassFile(JarEntry e) throws IOException {

            InputStream is = null;

            try {

                is = jarfile.getInputStream(e);

                return ClassFile.read(is);

            } catch (ConstantPoolException ex) {

                throw new ClassFileError(ex);

            } finally {

                if (is != null)

                    is.close();

            }

        }

        public Iterable<ClassFile> getClassFiles() throws IOException {

            final Iterator<ClassFile> iter = new JarFileIterator();

            return new Iterable<ClassFile>() {

                public Iterator<ClassFile> iterator() {

                    return iter;

                }

            };

        }

        class JarFileIterator implements Iterator<ClassFile> {

            private Enumeration<JarEntry> entries;

            private JarEntry nextEntry;

            JarFileIterator() {

                this.entries = jarfile.entries();

                while (entries.hasMoreElements()) {

                    JarEntry e = entries.nextElement();

                    String name = e.getName();

                    if (name.endsWith(".class")) {

                        this.nextEntry = e;

                        break;

                    }

                }

            }

            public boolean hasNext() {

                return nextEntry != null;

            }

            public ClassFile next() {

                if (!hasNext()) {

                    throw new NoSuchElementException();

                }

                ClassFile cf;

                try {

                    cf = readClassFile(nextEntry);

                } catch (IOException ex) {

                    throw new ClassFileError(ex);

                }

                JarEntry entry = nextEntry;

                nextEntry = null;

                while (entries.hasMoreElements()) {

                    JarEntry e = entries.nextElement();

                    String name = e.getName();

                    if (name.endsWith(".class")) {

                        nextEntry = e;

                        break;

                    }

                }

                return cf;

            }

            public void remove() {

                throw new UnsupportedOperationException("Not supported yet.");

            }

        }

    }

}