/**
 * Copyright 2007 Google Inc.
 *
 * 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
 *
 * http://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 com.tonicsystems.jarjar.strings;

import com.tonicsystems.jarjar.classpath.ClassPath;
import com.tonicsystems.jarjar.classpath.ClassPathArchive;
import com.tonicsystems.jarjar.classpath.ClassPathResource;
import com.tonicsystems.jarjar.util.IoUtil;
import com.tonicsystems.jarjar.util.RuntimeIOException;
import java.io.IOException;
import java.io.InputStream;
import javax.annotation.Nonnull;
import org.objectweb.asm.ClassReader;

public class StringDumper {

    public void run(Appendable out, ClassPath classPath) throws IOException {
        StringReader stringReader = new DumpStringReader(out);
        for (ClassPathArchive classPathArchive : classPath) {
            for (ClassPathResource classPathResource : classPathArchive) {
                InputStream in = classPathResource.openStream();
                try {
                    new ClassReader(in).accept(stringReader, 0);
                } catch (Exception e) {
                    System.err.println("Error reading " + classPathResource + ": " + e.getMessage());
                } finally {
                    in.close();
                }
                IoUtil.flush(out);
            }
        }
    }

    private static class DumpStringReader extends StringReader {

        private final Appendable out;
        private String className;

        public DumpStringReader(@Nonnull Appendable out) {
            this.out = out;
        }

        @Override
        public void visitString(String className, String value, int line) {
            if (value.length() > 0) {
                try {
                    if (!className.equals(this.className)) {
                        this.className = className;
                        out.append(className.replace('/', '.'));
                    }
                    out.append("\t");
                    if (line >= 0)
                        out.append(line + ": ");
                    out.append(escapeStringLiteral(value));
                    out.append("\n");
                } catch (IOException e) {
                    throw new RuntimeIOException(e);
                }
            }
        }
    };

    @Nonnull
    private static String escapeStringLiteral(@Nonnull String value) {
        StringBuilder sb = new StringBuilder();
        sb.append("\"");
        for (int i = 0, size = value.length(); i < size; i++) {
            char ch = value.charAt(i);
            switch (ch) {
                case '\n':
                    sb.append("\\n");
                    break;
                case '\r':
                    sb.append("\\r");
                    break;
                case '\b':
                    sb.append("\\b");
                    break;
                case '\f':
                    sb.append("\\f");
                    break;
                case '\t':
                    sb.append("\\t");
                    break;
                case '\"':
                    sb.append("\\\"");
                    break;
                case '\\':
                    sb.append("\\\\");
                    break;
                default:
                    sb.append(ch);
            }
        }
        sb.append("\"");
        return sb.toString();
    }
}