package com.mesalab.ua.analyser.impl;

import com.mesalab.ua.analyser.enums.UserInfo;
import com.mesalab.ua.analyser.enums.RegexFeatureEnum;

import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.regex.Pattern;

/**
 * 终端设备分类
 * @author yjy
 * @version 1.0
 * @date 2020/11/15 12:20 下午
 */
public class TerminalClassify {

    public static void checkSys(UserInfo userInfo){
        List<String> uaList = userInfo.getUaList();
        List<String> catList = userInfo.getTerminalCatList();

        Pattern apple = Pattern.compile(RegexFeatureEnum.TERMINAL_CAT_APPLE_FEATURE.getCode());
        Pattern android = Pattern.compile(RegexFeatureEnum.TERMINAL_CAT_ANDROID_FEATURE.getCode());
        Pattern win = Pattern.compile(RegexFeatureEnum.TERMINAL_CAT_WINDOWS_FEATURE.getCode());

        Map<String, Integer> catCnt = new HashMap<>();
        catCnt.put(RegexFeatureEnum.TERMINAL_CAT_APPLE.getCode(), 0);
        catCnt.put(RegexFeatureEnum.TERMINAL_CAT_ANDROID.getCode(), 0);
        catCnt.put(RegexFeatureEnum.TERMINAL_CAT_WINDOWS.getCode(), 0);

        for (String s : uaList) {
            if (apple.matcher(s).find()) {
                catCnt.put(RegexFeatureEnum.TERMINAL_CAT_APPLE.getCode(),
                        catCnt.get(RegexFeatureEnum.TERMINAL_CAT_APPLE.getCode()) + 1);
                continue;
            }
            if (android.matcher(s).find()) {
                catCnt.put(RegexFeatureEnum.TERMINAL_CAT_ANDROID.getCode(),
                        catCnt.get(RegexFeatureEnum.TERMINAL_CAT_ANDROID.getCode()) + 1);
                continue;
            }
            if (win.matcher(s).find()) {
                catCnt.put(RegexFeatureEnum.TERMINAL_CAT_WINDOWS.getCode(),
                        catCnt.get(RegexFeatureEnum.TERMINAL_CAT_WINDOWS.getCode()) + 1);
            }
        }

        if (catCnt.get(RegexFeatureEnum.TERMINAL_CAT_WINDOWS.getCode())!=0) {
            catList.add(RegexFeatureEnum.TERMINAL_CAT_WINDOWS.getCode());
        }
        if (catCnt.get(RegexFeatureEnum.TERMINAL_CAT_ANDROID.getCode())!=0) {
            catList.add(RegexFeatureEnum.TERMINAL_CAT_ANDROID.getCode());
        }
        if (catCnt.get(RegexFeatureEnum.TERMINAL_CAT_APPLE.getCode())!=0) {
            catList.add(RegexFeatureEnum.TERMINAL_CAT_APPLE.getCode());
        }

        userInfo.setTerminalCatList(catList);
    }
}