/*
 * Copyright (C) 2015-2016 Willi Ye <[email protected]>
 *
 * This file is part of Kernel Adiutor.
 *
 * Kernel Adiutor 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.
 *
 * Kernel Adiutor 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 Kernel Adiutor.  If not, see <http://www.gnu.org/licenses/>.
 *
 */
package com.smartpack.kernelmanager.utils;

import android.os.Build;

import com.smartpack.kernelmanager.utils.root.RootUtils;

import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Objects;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

/**
 * Created by willi on 31.12.15.
 */
public class Device {

    public static class Input {

        private static Input sInstance;

        public static Input getInstance() {
            if (sInstance == null) {
                sInstance = new Input();
            }
            return sInstance;
        }

        private static final String BUS_INPUT = "/proc/bus/input/devices";

        private List<Item> mItems = new ArrayList<>();

        private Input() {
            String value = Utils.readFile(BUS_INPUT);
            List<String> input = new ArrayList<>();
            for (String line : value.split("\\r?\\n")) {
                if (line.isEmpty()) {
                    mItems.add(new Item(input));
                    input = new ArrayList<>();
                } else {
                    input.add(line);
                }
            }
        }

        public List<Item> getItems() {
            return mItems;
        }

        public static class Item {

            private String mBus;
            private String mVendor;
            private String mProduct;
            private String mVersion;
            private String mName;
            private String mSysfs;
            private String mHandlers;

            private Item(List<String> input) {
                for (String line : input) {
                    if (line.startsWith("I:")) {
                        line = line.replace("I:", "").trim();
                        try {
                            mBus = line.split("Bus=")[1].split(" ")[0];
                        } catch (Exception ignored) {
                        }
                        try {
                            mVendor = line.split("Vendor=")[1].split(" ")[0];
                        } catch (Exception ignored) {
                        }
                        try {
                            mProduct = line.split("Product=")[1].split(" ")[0];
                        } catch (Exception ignored) {
                        }
                        try {
                            mVersion = line.split("Version=")[1].split(" ")[0];
                        } catch (Exception ignored) {
                        }
                    } else if (line.startsWith("N:")) {
                        mName = line.replace("N:", "").trim().replace("Name=", "").replace("\"", "");
                    } else if (line.startsWith("S:")) {
                        mSysfs = line.replace("S:", "").trim().replace("Sysfs=", "").replace("\"", "");
                    } else if (line.startsWith("H:")) {
                        mHandlers = line.replace("H:", "").trim().replace("Handlers=", "").replace("\"", "");
                    }
                }
            }

            public String getBus() {
                return mBus;
            }

            public String getVendor() {
                return mVendor;
            }

            public String getProduct() {
                return mProduct;
            }

            public String getVersion() {
                return mVersion;
            }

            public String getName() {
                return mName;
            }

            public String getSysfs() {
                return mSysfs;
            }

            public String getHandlers() {
                return mHandlers;
            }

        }

    }

    public static class ROMInfo {

        private static ROMInfo sInstance;

        public static ROMInfo getInstance() {
            if (sInstance == null) {
                sInstance = new ROMInfo();
            }
            return sInstance;
        }

        private static String[] sProps = {
                "ro.cm.version",
                "ro.pa.version",
                "ro.pac.version",
                "ro.carbon.version",
                "ro.slim.version",
                "ro.mod.version",
                "ro.lineage.version",
                "ro.rr.version",
                "ro.oxygen.version"
        };

        private String mROMVersion;

        private ROMInfo() {
            for (String prop : sProps) {
                mROMVersion = RootUtils.getProp(prop);
                if (!mROMVersion.isEmpty()) {
                    break;
                }
            }
        }

        public String getVersion() {
            return mROMVersion;
        }

    }

    public static class MemInfo {

        private static MemInfo sInstance;

        public static MemInfo getInstance() {
            if (sInstance == null) {
                sInstance = new MemInfo();
            }
            return sInstance;
        }

        private static final String MEMINFO_PROC = "/proc/meminfo";
        private String MEMINFO;

        private MemInfo() {
            MEMINFO = Utils.readFile(MEMINFO_PROC);
        }

        public long getTotalMem() {
            try {
                return Long.parseLong(getItem("MemTotal").replaceAll("[^\\d]", "")) / 1024L;
            } catch (NumberFormatException ignored) {
                return 0;
            }
        }

        public long getItemMb( String prefix) {
            try {
                return Long.parseLong(getItem(prefix).replaceAll("[^\\d]", "")) / 1024L;
            } catch (NumberFormatException ignored) {
                return 0;
            }
        }

        public List<String> getItems() {
            List<String> list = new ArrayList<>();
            try {
                for (String line : MEMINFO.split("\\r?\\n")) {
                    list.add(line.split(":")[0]);
                }
            } catch (Exception ignored) {
            }
            return list;
        }

        public String getItem(String prefix) {
            try {
                for (String line : MEMINFO.split("\\r?\\n")) {
                    if (line.startsWith(prefix)) {
                        return line.split(":")[1].trim();
                    }
                }
            } catch (Exception ignored) {
            }
            return "";
        }

    }

    public static class CPUInfo {

        private static CPUInfo sInstance;

        public static CPUInfo getInstance() {
            if (sInstance == null) {
                sInstance = new CPUInfo();
            }
            return sInstance;
        }

        private static final String CPUINFO_PROC = "/proc/cpuinfo";

        private final String mCPUInfo;

        private CPUInfo() {
            mCPUInfo = Utils.readFile(CPUINFO_PROC, false);
        }

        public String getFeatures() {
            String features = getString("Features");
            if (!features.isEmpty()) return features;
            return getString("flags");
        }

        public String getProcessor() {
            String pro = getString("Processor");
            if (!pro.isEmpty()) return pro;
            return getString("model name");
        }

        public String getVendor() {
            String vendor = getString("Hardware");
            if (!vendor.isEmpty()) return vendor;
            return getString("vendor_id");
        }

        public String getCpuInfo() {
            return mCPUInfo;
        }

        private String getString(String prefix) {
            try {
                for (String line : mCPUInfo.split("\\r?\\n")) {
                    if (line.startsWith(prefix)) {
                        return line.split(":")[1].trim();
                    }
                }
            } catch (Exception ignored) {
            }
            return "";
        }

    }

    public static class TrustZone {

        private static TrustZone sInstance;

        public static TrustZone getInstance() {
            if (sInstance == null) {
                sInstance = new TrustZone();
            }
            return sInstance;
        }

        private static HashMap<String, String> PARTITIONS = new HashMap<>();

        static {
            PARTITIONS.put("/dev/block/platform/msm_sdcc.1/by-name/tz", "QC_IMAGE_VERSION_STRING=");
            PARTITIONS.put("/dev/block/bootdevice/by-name/tz", "QC_IMAGE_VERSION_STRING=");
        }

        private String mVersion = "";

        private TrustZone() {
            String partition = null;

            for (String p : PARTITIONS.keySet()) {
                if (Utils.existFile(p)) {
                    partition = p;
                    break;
                }
            }

            if (partition == null) return;
            String prefix = PARTITIONS.get(partition);
            String raw = RootUtils.runAndGetOutput("strings " + partition + " | grep " + prefix);
            for (String line : raw.split("\\r?\\n")) {
                assert prefix != null;
                if (line.startsWith(prefix)) {
                    mVersion = line.replace(prefix, "");
                    break;
                }
            }
        }

        public String getVersion() {
            return mVersion;
        }
    }

    public static String getKernelVersion(boolean extended) {
        return getKernelVersion(extended, true);
    }

    private static String getKernelVersion(boolean extended, boolean root) {
        return extended ? Utils.readFile("/proc/version", root) : RootUtils.runAndGetOutput("uname -r");
    }

    public static String getArchitecture() {
        return RootUtils.runAndGetOutput("uname -m");
    }

    public static String getHardware() {
        return Build.HARDWARE;
    }

    public static String getBootloader() {
        return Build.BOOTLOADER;
    }

    public static String getBaseBand() {
        return Build.getRadioVersion();
    }

    public static String getCodename() {
        String codeName = "";
        Field[] fields = Build.VERSION_CODES.class.getFields();
        for (Field field : fields) {
            String fieldName = field.getName();
            int fieldValue = -1;

            try {
                fieldValue = field.getInt(new Object());
            } catch (IllegalArgumentException | IllegalAccessException | NullPointerException ignored) {
            }

            if (fieldValue == Build.VERSION.SDK_INT) {
                codeName = fieldName;
                break;
            }
        }
        return codeName;
    }

    public static int getSDK() {
        return Build.VERSION.SDK_INT;
    }

    private interface BoardFormatter {
        String format(String board);
    }

    private static HashMap<String, BoardFormatter> sBoardFormatters = new HashMap<>();
    private static HashMap<String, String> sBoardAliases = new HashMap<>();

    static {
        sBoardFormatters.put(".*msm.+.\\d+.*", board -> "msm" + board.split("msm")[1].trim().split(" ")[0]);

        sBoardFormatters.put("mt\\d*.", board -> "mt" + board.split("mt")[1].trim().split(" ")[0]);

        sBoardFormatters.put(".*apq.+.\\d+.*", board -> "apq" + board.split("apq")[1].trim().split(" ")[0]);

        sBoardFormatters.put(".*omap+\\d.*", board -> {
            Matcher matcher = Pattern.compile("omap+\\d").matcher(board);
            if (matcher.find()) {
                return matcher.group();
            }
            return null;
        });

        sBoardFormatters.put("sun+\\d.", board -> board);

        sBoardFormatters.put("spyder", board -> "omap4");
        sBoardFormatters.put("tuna", board -> "omap4");

        sBoardAliases.put("msm8994v2.1", "msm8994");
        sBoardAliases.put("msm8974pro.*", "msm8974pro");
    }

    public static String getBoard() {
        String hardware = CPUInfo.getInstance().getVendor().toLowerCase();
        String ret = null;
        for (String boardregex : sBoardFormatters.keySet()) {
            if (hardware.matches(boardregex)) {
                ret = Objects.requireNonNull(sBoardFormatters.get(boardregex)).format(hardware);
            }
        }
        if (ret != null) {
            for (String alias : sBoardAliases.keySet()) {
                assert ret != null;
                if (ret.matches(alias)) {
                    ret = sBoardAliases.get(alias);
                }
            }
        }
        return ret != null ? ret : Build.BOARD.toLowerCase();
    }

    public static String getBuildDisplayId() {
        return Build.DISPLAY;
    }

    public static String getFingerprint() {
        return Build.FINGERPRINT;
    }

    public static String getVersion() {
        return Build.VERSION.RELEASE;
    }

    public static String getVendor() {
        return Build.MANUFACTURER;
    }

    public static String getDeviceName() {
        return Build.DEVICE;
    }

    public static String getModel() {
        return Build.MODEL;
    }

    public static String getTrebleStatus() {
        return RootUtils.getProp("ro.treble.enabled");
    }

    public static String getGPULibVersion() {
        return RootUtils.runAndGetOutput("dumpsys SurfaceFlinger | grep GLES | head -n 1 | cut -f 3,4,5 -d ','");
    }

}