package dagger.hilt.android.internal.lifecycle;

import android.app.Activity;
import android.os.Bundle;
import androidx.lifecycle.an;
import androidx.lifecycle.at;
import dagger.Module;
import dagger.hilt.EntryPoint;
import dagger.hilt.EntryPoints;
import dagger.hilt.InstallIn;
import dagger.hilt.android.components.ActivityComponent;
import dagger.hilt.android.components.ViewModelComponent;
import dagger.hilt.android.internal.builders.ViewModelComponentBuilder;
import dagger.hilt.android.internal.lifecycle.HiltViewModelMap;
import dagger.multibindings.Multibinds;
import defpackage.kp6;
import defpackage.rq6;
import defpackage.sq3;
import defpackage.vo6;
import defpackage.wa8;
import defpackage.wm1;
import defpackage.xja;
import java.io.Closeable;
import java.util.Map;
import javax.inject.Provider;
import kotlin.reflect.KClass;

/* JADX INFO: loaded from: classes3.dex */
public final class HiltViewModelFactory implements at.c {
    public static final wm1.b<sq3<Object, xja>> CREATION_CALLBACK_KEY = new wm1.b<sq3<Object, xja>>() { // from class: dagger.hilt.android.internal.lifecycle.HiltViewModelFactory.1
    };
    private final at.c delegateFactory;
    private final at.c hiltViewModelFactory;
    private final Map<Class<?>, Boolean> hiltViewModelKeys;

    @EntryPoint
    @InstallIn({ActivityComponent.class})
    public interface ActivityCreatorEntryPoint {
        ViewModelComponentBuilder getViewModelComponentBuilder();

        @HiltViewModelMap.KeySet
        Map<Class<?>, Boolean> getViewModelKeys();
    }

    @EntryPoint
    @InstallIn({ViewModelComponent.class})
    public interface ViewModelFactoriesEntryPoint {
        @HiltViewModelAssistedMap
        Map<Class<?>, Object> getHiltViewModelAssistedMap();

        @HiltViewModelMap
        Map<Class<?>, Provider<xja>> getHiltViewModelMap();
    }

    @Module
    @InstallIn({ViewModelComponent.class})
    public interface ViewModelModule {
        @HiltViewModelAssistedMap
        @Multibinds
        Map<Class<?>, Object> hiltViewModelAssistedMap();

        @HiltViewModelMap
        @Multibinds
        Map<Class<?>, xja> hiltViewModelMap();
    }

    public HiltViewModelFactory(@vo6 Map<Class<?>, Boolean> map, @vo6 at.c cVar, @vo6 final ViewModelComponentBuilder viewModelComponentBuilder) {
        this.hiltViewModelKeys = map;
        this.delegateFactory = cVar;
        this.hiltViewModelFactory = new at.c() { // from class: dagger.hilt.android.internal.lifecycle.HiltViewModelFactory.2
            private <T extends xja> T createViewModel(@vo6 ViewModelComponent viewModelComponent, @vo6 Class<T> cls, @vo6 wm1 wm1Var) {
                Provider<xja> provider = ((ViewModelFactoriesEntryPoint) EntryPoints.get(viewModelComponent, ViewModelFactoriesEntryPoint.class)).getHiltViewModelMap().get(cls);
                sq3 sq3Var = (sq3) wm1Var.a(HiltViewModelFactory.CREATION_CALLBACK_KEY);
                Object obj = ((ViewModelFactoriesEntryPoint) EntryPoints.get(viewModelComponent, ViewModelFactoriesEntryPoint.class)).getHiltViewModelAssistedMap().get(cls);
                if (obj == null) {
                    if (sq3Var != null) {
                        throw new IllegalStateException("Found creation callback but class " + cls.getName() + " does not have an assisted factory specified in @HiltViewModel.");
                    }
                    if (provider != null) {
                        return (T) provider.get();
                    }
                    throw new IllegalStateException("Expected the @HiltViewModel-annotated class " + cls.getName() + " to be available in the multi-binding of @HiltViewModelMap but none was found.");
                }
                if (provider != null) {
                    throw new AssertionError("Found the @HiltViewModel-annotated class " + cls.getName() + " in both the multi-bindings of @HiltViewModelMap and @HiltViewModelAssistedMap.");
                }
                if (sq3Var != null) {
                    return (T) sq3Var.invoke(obj);
                }
                throw new IllegalStateException("Found @HiltViewModel-annotated class " + cls.getName() + " using @AssistedInject but no creation callback was provided in CreationExtras.");
            }

            @Override // androidx.lifecycle.at.c
            @kp6
            public /* bridge */ /* synthetic */ xja create(@kp6 Class cls) {
                super.create(cls);
                throw null;
            }

            @Override // androidx.lifecycle.at.c
            @kp6
            public /* bridge */ /* synthetic */ xja create(@kp6 KClass kClass, @kp6 wm1 wm1Var) {
                return super.create(kClass, wm1Var);
            }

            @Override // androidx.lifecycle.at.c
            @vo6
            public <T extends xja> T create(@vo6 Class<T> cls, @vo6 wm1 wm1Var) {
                final RetainedLifecycleImpl retainedLifecycleImpl = new RetainedLifecycleImpl();
                T t = (T) createViewModel(viewModelComponentBuilder.savedStateHandle(an.a(wm1Var)).viewModelLifecycle(retainedLifecycleImpl).build(), cls, wm1Var);
                t.addCloseable(new Closeable() { // from class: x34
                    @Override // java.io.Closeable, java.lang.AutoCloseable
                    public final void close() {
                        retainedLifecycleImpl.dispatchOnCleared();
                    }
                });
                return t;
            }
        };
    }

    public static at.c createInternal(@vo6 Activity activity, @vo6 wa8 wa8Var, @rq6 Bundle bundle, @vo6 at.c cVar) {
        return createInternal(activity, cVar);
    }

    @Override // androidx.lifecycle.at.c
    @kp6
    public /* bridge */ /* synthetic */ xja create(@kp6 KClass kClass, @kp6 wm1 wm1Var) {
        return super.create(kClass, wm1Var);
    }

    public static at.c createInternal(@vo6 Activity activity, @vo6 at.c cVar) {
        ActivityCreatorEntryPoint activityCreatorEntryPoint = (ActivityCreatorEntryPoint) EntryPoints.get(activity, ActivityCreatorEntryPoint.class);
        return new HiltViewModelFactory(activityCreatorEntryPoint.getViewModelKeys(), cVar, activityCreatorEntryPoint.getViewModelComponentBuilder());
    }

    @Override // androidx.lifecycle.at.c
    @vo6
    public <T extends xja> T create(@vo6 Class<T> cls, @vo6 wm1 wm1Var) {
        return this.hiltViewModelKeys.containsKey(cls) ? (T) this.hiltViewModelFactory.create(cls, wm1Var) : (T) this.delegateFactory.create(cls, wm1Var);
    }

    @Override // androidx.lifecycle.at.c
    @vo6
    public <T extends xja> T create(@vo6 Class<T> cls) {
        if (this.hiltViewModelKeys.containsKey(cls)) {
            return (T) this.hiltViewModelFactory.create(cls);
        }
        return (T) this.delegateFactory.create(cls);
    }
}