/*

 * Copyright (c) 2015, 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.

 *

 * 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.

 */

import java.io.ByteArrayInputStream;

import java.io.ByteArrayOutputStream;

import java.io.IOException;

import java.time.Instant;

import java.time.ZoneId;

import java.time.ZonedDateTime;

import java.time.format.DateTimeFormatter;

import java.time.temporal.ChronoUnit;

import java.util.Arrays;

import java.util.Collections;

import java.util.List;

import java.util.Locale;

import java.util.Objects;

import java.util.Properties;

import java.util.logging.Level;

import java.util.logging.LogManager;

import java.util.logging.LogRecord;

import java.util.logging.XMLFormatter;

/**

 * @test

 * @bug 8072645

 * @summary tests that XmlFormatter will print out dates with the new

 *                nanosecond precision.

 * @run main/othervm XmlFormatterNanos

 * @author danielfuchs

 */

public class XmlFormatterNanos {

    static final int MILLIS_IN_SECOND = 1000;

    static final int NANOS_IN_MILLI = 1000_000;

    static final int NANOS_IN_MICRO = 1000;

    static final int NANOS_IN_SECOND = 1000_000_000;

    static final boolean verbose = true;

    static final class TestAssertException extends RuntimeException {

        TestAssertException(String msg) { super(msg); }

    }

    private static void assertEquals(long expected, long received, String msg) {

        if (expected != received) {

            throw new TestAssertException("Unexpected result for " + msg

                    + ".\n\texpected: " + expected

                    +  "\n\tactual:   " + received);

        } else if (verbose) {

            System.out.println("Got expected " + msg + ": " + received);

        }

    }

    private static void assertEquals(Object expected, Object received, String msg) {

        if (!Objects.equals(expected, received)) {

            throw new TestAssertException("Unexpected result for " + msg

                    + ".\n\texpected: " + expected

                    +  "\n\tactual:   " + received);

        } else if (verbose) {

            System.out.println("Got expected " + msg + ": " + received);

        }

    }

    static class CustomXMLFormatter extends XMLFormatter {}

    static class Configuration {

        final Properties propertyFile;

        private Configuration(Properties properties) {

            propertyFile = properties;

        }

        public Configuration apply() {

            try {

                ByteArrayOutputStream bytes = new ByteArrayOutputStream();

                propertyFile.store(bytes, testName());

                ByteArrayInputStream bais = new ByteArrayInputStream(bytes.toByteArray());

                LogManager.getLogManager().readConfiguration(bais);

            } catch (IOException ex) {

                throw new RuntimeException(ex);

            }

            return this;

        }

        public static String useInstantProperty(Class<?> type) {

            return type.getName()+".useInstant";

        }

        public boolean useInstant(XMLFormatter formatter) {

            return Boolean.parseBoolean(propertyFile.getProperty(

                    formatter.getClass().getName()+".useInstant", "true"));

        }

        public String testName() {

            return propertyFile.getProperty("test.name", "????");

        }

        public static Configuration of(Properties props) {

            return new Configuration(props);

        }

        public static Configuration apply(Properties props) {

            return of(props).apply();

        }

    }

    final static List<Properties> properties;

    static {

        Properties props1 = new Properties();

        props1.setProperty("test.name", "with XML nano element (default)");

        Properties props2 = new Properties();

        props2.setProperty("test.name", "with XML nano element (standard=true, custom=false)");

        props2.setProperty(Configuration.useInstantProperty(XMLFormatter.class), "true");

        props2.setProperty(Configuration.useInstantProperty(CustomXMLFormatter.class), "false");

        Properties props3 = new Properties();

        props3.setProperty("test.name", "with XML nano element (standard=false, custom=true)");

        props3.setProperty(Configuration.useInstantProperty(XMLFormatter.class), "false");

        props3.setProperty(Configuration.useInstantProperty(CustomXMLFormatter.class), "true");

        properties = Collections.unmodifiableList(Arrays.asList(

                    props1,

                    props2));

    }

    public static void main(String[] args) throws Exception {

        Locale.setDefault(Locale.ENGLISH);

        properties.stream().forEach(XmlFormatterNanos::test);

    }

    static int getNanoAdjustment(LogRecord record) {

        return record.getInstant().getNano() % NANOS_IN_MILLI;

    }

    static void setNanoAdjustment(LogRecord record, int nanos) {

        record.setInstant(Instant.ofEpochSecond(record.getInstant().getEpochSecond(),

                (record.getInstant().getNano() / NANOS_IN_MILLI) * NANOS_IN_MILLI + nanos));

    }

    public static void test(Properties props) {

        Configuration conf = Configuration.apply(props);

        LogRecord record = new LogRecord(Level.INFO, "Test Name: {0}");

        record.setLoggerName("test");

        record.setParameters(new Object[] {conf.testName()});

        int nanos = record.getInstant().getNano() % NANOS_IN_MILLI;

        long millis = record.getMillis();

        // make sure we don't have leading zeros when printing below

        // the second precision

        if (millis % MILLIS_IN_SECOND < 100) millis = millis + 100;

        // make sure we some nanos - and make sure we don't have

        // trailing zeros

        if (nanos % 10 == 0) nanos = nanos + 7;

        record.setMillis(millis);

        setNanoAdjustment(record, nanos);

        final Instant instant = record.getInstant();

        if (nanos < 0) {

            throw new RuntimeException("Unexpected negative nano adjustment: "

                    + getNanoAdjustment(record));

        }

        if (nanos >= NANOS_IN_MILLI) {

            throw new RuntimeException("Nano adjustment exceeds 1ms: "

                    + getNanoAdjustment(record));

        }

        if (millis != record.getMillis()) {

            throw new RuntimeException("Unexpected millis: " + millis + " != "

                    + record.getMillis());

        }

        if (millis != record.getInstant().toEpochMilli()) {

            throw new RuntimeException("Unexpected millis: "

                    + record.getInstant().toEpochMilli());

        }

        long expectedNanos = (millis % MILLIS_IN_SECOND) * NANOS_IN_MILLI + nanos;

        assertEquals(expectedNanos, instant.getNano(), "Instant.getNano()");

        XMLFormatter formatter = new XMLFormatter();

        testMatching(formatter, record, instant, expectedNanos, conf.useInstant(formatter));

        XMLFormatter custom = new CustomXMLFormatter();

        testMatching(custom, record, instant, expectedNanos, conf.useInstant(custom));

    }

    private static void testMatching(XMLFormatter formatter,

            LogRecord record,  Instant instant, long expectedNanos,

            boolean useInstant) {

        ZonedDateTime zdt = ZonedDateTime.ofInstant(instant, ZoneId.systemDefault());

        int zdtNanos = zdt.getNano();

        assertEquals(expectedNanos, zdtNanos, "ZonedDateTime.getNano()");

        String str = formatter.format(record);

        String match = "."+expectedNanos;

        if (str.contains(match) != useInstant) {

            throw new RuntimeException(formatter.getClass().getSimpleName()

                    + ".format()"

                    + " string does not contain expected nanos: "

                    + "\n\texpected match for: '" + match + "'"

                    + "\n\tin: \n" + str);

        }

        System.out.println("Found expected match for '"+match+"' in \n"+str);

        match = "<millis>"+instant.toEpochMilli()+"</millis>";

        if (!str.contains(match)) {

            throw new RuntimeException(formatter.getClass().getSimpleName()

                    + ".format()"

                    + " string does not contain expected millis: "

                    + "\n\texpected match for: '" + match + "'"

                    + "\n\tin: \n" + str);

        }

        System.out.println("Found expected match for '"+match+"' in \n"+str);

        match = "<nanos>";

        if (str.contains(match) != useInstant) {

            throw new RuntimeException(formatter.getClass().getSimpleName()

                    + ".format()"

                    + " string "

                    + (useInstant

                            ? "does not contain expected nanos: "

                            : "contains unexpected nanos: ")

                    + "\n\t" + (useInstant ? "expected" : "unexpected")

                    + " match for: '" + match + "'"

                    + "\n\tin: \n" + str);

        }

        match = "<nanos>"+getNanoAdjustment(record)+"</nanos>";

        if (str.contains(match) != useInstant) {

            throw new RuntimeException(formatter.getClass().getSimpleName()

                    + ".format()"

                    + " string "

                    + (useInstant

                            ? "does not contain expected nanos: "

                            : "contains unexpected nanos: ")

                    + "\n\t" + (useInstant ? "expected" : "unexpected")

                    + " match for: '" + match + "'"

                    + "\n\tin: \n" + str);

        }

        if (useInstant) {

            System.out.println("Found expected match for '"+match+"' in \n"+str);

        } else {

            System.out.println("As expected '"+match+"' is not present in \n"+str);

        }

        match = useInstant ? DateTimeFormatter.ISO_INSTANT.format(instant)

                : zdt.truncatedTo(ChronoUnit.SECONDS)

                        .format(DateTimeFormatter.ISO_LOCAL_DATE_TIME);

        match = "<date>"+match+"</date>";

        if (!str.contains(match)) {

            throw new RuntimeException(formatter.getClass().getSimpleName()

                    + ".format()"

                    + " string does not contain expected date: "

                    + "\n\texpected match for: '" + match + "'"

                    + "\n\tin: \n" + str);

        }

        System.out.println("Found expected match for '"+match+"' in \n"+str);

    }

}