package eu.faircode.email;

/*
    This file is part of FairEmail.

    FairEmail is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    FairEmail 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 for more details.

    You should have received a copy of the GNU General Public License
    along with FairEmail.  If not, see <http://www.gnu.org/licenses/>.

    Copyright 2018-2022 by Marcel Bokhorst (M66B)
*/

import android.text.TextUtils;
import android.util.Pair;

import java.nio.ByteBuffer;
import java.nio.charset.CharacterCodingException;
import java.nio.charset.Charset;
import java.nio.charset.CharsetDecoder;
import java.nio.charset.CodingErrorAction;
import java.nio.charset.StandardCharsets;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.Locale;

public class CharsetHelper {
    private static final int MAX_SAMPLE_SIZE = 8192;
    private static String CHINESE = new Locale("zh").getLanguage();
    private static final List<String> COMMON = Collections.unmodifiableList(Arrays.asList(
            "US-ASCII",
            "ISO-8859-1", "ISO-8859-2",
            "windows-1250", "windows-1252", "windows-1257",
            "UTF-7", "UTF-8"
    ));
    private static final int MIN_W1252 = 10;
    private static final Pair<byte[], byte[]>[] sUtf8W1252 = new Pair[128];

    static {
        System.loadLibrary("fairemail");

        // https://www.i18nqa.com/debug/utf8-debug.html
        Charset w1252 = Charset.forName("windows-1252");
        for (int c = 128; c < 256; c++) {
            String y = new String(new byte[]{(byte) c}, w1252);
            String x = new String(y.getBytes(), w1252);
            sUtf8W1252[c - 128] = new Pair<>(x.getBytes(), y.getBytes());
        }
    }

    private static native DetectResult jni_detect_charset(byte[] octets);

    static boolean isUTF8(String text) {
        // Get extended ASCII characters
        byte[] octets = text.getBytes(StandardCharsets.ISO_8859_1);
        return isUTF8(octets);
    }

    static boolean isUTF8(byte[] octets) {
        CharsetDecoder utf8Decoder = StandardCharsets.UTF_8.newDecoder()
                .onMalformedInput(CodingErrorAction.REPORT)
                .onUnmappableCharacter(CodingErrorAction.REPORT);
        try {
            utf8Decoder.decode(ByteBuffer.wrap(octets));
            return true;
        } catch (CharacterCodingException ex) {
            Log.w(ex);
            return false;
        }
    }

    static boolean isUTF8Alt(String text) {
        byte[] octets = text.getBytes(StandardCharsets.ISO_8859_1);

        int bytes;
        for (int i = 0; i < octets.length; i++) {
            if ((octets[i] & 0b10000000) == 0b00000000)
                bytes = 1;
            else if ((octets[i] & 0b11100000) == 0b11000000)
                bytes = 2;
            else if ((octets[i] & 0b11110000) == 0b11100000)
                bytes = 3;
            else if ((octets[i] & 0b11111000) == 0b11110000)
                bytes = 4;
            else if ((octets[i] & 0b11111100) == 0b11111000)
                bytes = 5;
            else if ((octets[i] & 0b11111110) == 0b11111100)
                bytes = 6;
            else
                return false;

            if (i + bytes > octets.length)
                return false;

            while (--bytes > 0)
                if ((octets[++i] & 0b11000000) != 0b10000000)
                    return false;
        }
        return true;
    }

    static String utf8toW1252(String text) {
        try {
            Charset w1252 = Charset.forName("windows-1252");

            //String result = new String(text.getBytes(StandardCharsets.ISO_8859_1), w1252);
            //for (int c = 0; c < 128; c++) {
            //    String y = new String(sUtf8W1252[c].second);
            //    String x = new String(sUtf8W1252[c].first);
            //    result = result.replace(x, y);
            //}
            //return result;

            byte[] t = new String(text.getBytes(StandardCharsets.ISO_8859_1), w1252).getBytes();
            byte[] result = new byte[t.length];

            int i = 0;
            int len = 0;
            int count = 0;
            while (i < t.length && (i < MAX_SAMPLE_SIZE || count >= MIN_W1252)) {
                boolean found = false;
                for (int c = 0; c < 128; c++) {
                    int sl = sUtf8W1252[c].first.length;
                    if (i + sl < t.length) {
                        found = true;
                        for (int a = 0; a < sl; a++)
                            if (t[i + a] != sUtf8W1252[c].first[a]) {
                                found = false;
                                break;
                            }
                        if (found) {
                            count++;
                            int tl = sUtf8W1252[c].second.length;
                            System.arraycopy(sUtf8W1252[c].second, 0, result, len, tl);
                            len += tl;
                            i += sl;
                            break;
                        }
                    }
                    if (found)
                        break;
                }
                if (!found)
                    result[len++] = t[i++];
            }
            return (count < MIN_W1252 ? text : new String(result, 0, len));
        } catch (Throwable ex) {
            Log.w(ex);
            return text;
        }
    }

    public static Charset detect(String text) {
        try {
            byte[] octets = text.getBytes(StandardCharsets.ISO_8859_1);

            byte[] sample;
            if (octets.length < MAX_SAMPLE_SIZE)
                sample = octets;
            else {
                sample = new byte[MAX_SAMPLE_SIZE];
                System.arraycopy(octets, 0, sample, 0, MAX_SAMPLE_SIZE);
            }

            Log.i("compact_enc_det sample=" + sample.length);
            DetectResult detected = jni_detect_charset(sample);

            if (TextUtils.isEmpty(detected.charset)) {
                Log.e("compact_enc_det result=" + detected);
                return null;
            } else if (COMMON.contains(detected.charset))
                Log.w("compact_enc_det result=" + detected);
            else if ("GB18030".equals(detected.charset)) {
                boolean chinese = Locale.getDefault().getLanguage().equals(CHINESE);
                // https://github.com/google/compact_enc_det/issues/8
                Log.e("compact_enc_det result=" + detected + " chinese=" + chinese);
                if (!chinese)
                    return null;
            } else // GBK, Big5, ISO-2022-JP, HZ-GB-2312, Shift_JIS
                Log.e("compact_enc_det result=" + detected);

            return Charset.forName(detected.charset);
        } catch (Throwable ex) {
            Log.w(ex);
            return null;
        }
    }

    private static class DetectResult {
        String charset;
        int sample_size;
        int bytes_consumed;
        boolean is_reliable;

        DetectResult(String charset, int sample_size, int bytes_consumed, boolean is_reliable) {
            this.charset = charset;
            this.sample_size = sample_size;
            this.bytes_consumed = bytes_consumed;
            this.is_reliable = is_reliable;
        }

        @Override
        public String toString() {
            return charset + " s=" + bytes_consumed + "/" + sample_size + " r=" + is_reliable;
        }
    }
}