/*
 * QAuxiliary - An Xposed module for QQ/TIM
 * Copyright (C) 2019-2022 qwq233@qwq2333.top
 * https://github.com/cinit/QAuxiliary
 *
 * This software is non-free but opensource software: you can redistribute it
 * and/or modify it under the terms of the GNU Affero General Public License
 * as published by the Free Software Foundation; either
 * version 3 of the License, or any later version and our eula as published
 * by QAuxiliary contributors.
 *
 * This software 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
 * Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * and eula along with this software.  If not, see
 * <https://www.gnu.org/licenses/>
 * <https://github.com/cinit/QAuxiliary/blob/master/LICENSE.md>.
 */
package cc.ioctl.hook.chat;

import androidx.annotation.NonNull;
import androidx.annotation.Nullable;
import cc.ioctl.util.HookUtils;
import de.robv.android.xposed.XC_MethodHook;
import de.robv.android.xposed.XposedBridge;
import io.github.qauxv.base.annotation.FunctionHookEntry;
import io.github.qauxv.base.annotation.UiItemAgentEntry;
import io.github.qauxv.dsl.FunctionEntryRouter.Locations.Auxiliary;
import io.github.qauxv.hook.CommonSwitchFunctionHook;
import io.github.qauxv.util.Initiator;
import io.github.qauxv.util.dexkit.DexKit;
import io.github.qauxv.util.dexkit.DexKitTarget;
import io.github.qauxv.util.dexkit.DexMethodDescriptor;
import io.github.qauxv.util.dexkit.NVipUtils_getPrivilegeFlags;
import java.lang.reflect.Method;

@FunctionHookEntry
@UiItemAgentEntry
public class FakeVipHook extends CommonSwitchFunctionHook {

    public static final FakeVipHook INSTANCE = new FakeVipHook();

    private FakeVipHook() {
        super(new DexKitTarget[]{NVipUtils_getPrivilegeFlags.INSTANCE});
    }

    @Override
    public boolean initOnce() throws NoSuchMethodException {
        XC_MethodHook hook = HookUtils.afterIfEnabled(this, param -> {
            int ret;
            // null is self
            Object uin = param.args[param.args.length - 1];
            if (uin == null) {
                ret = (int) param.getResult();
                // vip + svip + 大会员
                param.setResult(2 | 4 | 8 | ret);
            }
        });
        Method getPrivilegeFlags0VipUtils;
        try {
            getPrivilegeFlags0VipUtils = new DexMethodDescriptor("Lcom/tencent/mobileqq/utils/VipUtils;->a(Lmqq/app/AppRuntime;Ljava/lang/String;)I")
                    .getMethodInstance(Initiator.getHostClassLoader());
        } catch (NoSuchMethodException e) {
            getPrivilegeFlags0VipUtils = DexKit.loadMethodFromCache(NVipUtils_getPrivilegeFlags.INSTANCE);
        }
        if (getPrivilegeFlags0VipUtils != null && getPrivilegeFlags0VipUtils.getReturnType() != int.class) {
            throw new IllegalStateException("VipUtils.getPrivilegeFlags(AppRuntime, String) return type is not int");
        }
        // com.tencent.mobileqq.vip.VipStatusManagerImpl#getPrivilegeFlags(String)I
        Method getPrivilegeFlags1VipStatusManagerImpl = null;
        Class<?> kVipStatusManagerImpl = Initiator.load("com.tencent.mobileqq.vip.VipStatusManagerImpl");
        if (kVipStatusManagerImpl != null) {
            getPrivilegeFlags1VipStatusManagerImpl = kVipStatusManagerImpl.getDeclaredMethod("getPrivilegeFlags", String.class);
            XposedBridge.hookMethod(getPrivilegeFlags1VipStatusManagerImpl, hook);
        }
        if (getPrivilegeFlags0VipUtils != null) {
            XposedBridge.hookMethod(getPrivilegeFlags0VipUtils, hook);
        }
        return true;
    }

    @NonNull
    @Override
    public String[] getUiItemLocation() {
        return Auxiliary.CHAT_CATEGORY;
    }

    @NonNull
    @Override
    public String getName() {
        return "非会员贴表情";
    }

    @Nullable
    @Override
    public String getDescription() {
        return "假装是 SVIP, 非会员也可以贴表情";
    }

    @Nullable
    @Override
    public String[] getExtraSearchKeywords() {
        return new String[]{"VIP", "SVIP", "贴表情", "贴收藏表情"};
    }
}
