Home | History | Annotate | Download | only in setup 
      1 /*
      2  * Copyright (C) 2017 The Android Open Source Project
      3  *
      4  * Licensed under the Apache License, Version 2.0 (the "License");
      5  * you may not use this file except in compliance with the License.
      6  * You may obtain a copy of the License at
      7  *
      8  *      http://www.apache.org/licenses/LICENSE-2.0
      9  *
     10  * Unless required by applicable law or agreed to in writing, software
     11  * distributed under the License is distributed on an "AS IS" BASIS,
     12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     13  * See the License for the specific language governing permissions and
     14  * limitations under the License.
     15  */
     16 
     17 package com.android.tv.settings.connectivity.setup;
     18 
     19 import android.arch.lifecycle.ViewModelProviders;
     20 import android.content.Context;
     21 import android.os.Bundle;
     22 import android.support.annotation.VisibleForTesting;
     23 import android.support.v17.leanback.widget.GuidanceStylist;
     24 import android.support.v17.leanback.widget.GuidedAction;
     25 import android.support.v4.app.Fragment;
     26 import android.support.v4.app.FragmentActivity;
     27 
     28 import com.android.settingslib.wifi.AccessPoint;
     29 import com.android.tv.settings.R;
     30 import com.android.tv.settings.connectivity.WifiConfigHelper;
     31 import com.android.tv.settings.connectivity.util.State;
     32 import com.android.tv.settings.connectivity.util.StateMachine;
     33 
     34 import java.util.List;
     35 
     36 /**
     37  * State responsible for choosing security type.
     38  */
     39 public class ChooseSecurityState implements State {
     40     private final FragmentActivity mActivity;
     41     private Fragment mFragment;
     42 
     43     public ChooseSecurityState(FragmentActivity activity) {
     44         mActivity = activity;
     45     }
     46 
     47     @Override
     48     public void processForward() {
     49         mFragment = new ChooseSecurityFragment();
     50         FragmentChangeListener listener = (FragmentChangeListener) mActivity;
     51         if (listener != null) {
     52             listener.onFragmentChange(mFragment, true);
     53         }
     54     }
     55 
     56     @Override
     57     public void processBackward() {
     58         mFragment = new ChooseSecurityFragment();
     59         FragmentChangeListener listener = (FragmentChangeListener) mActivity;
     60         if (listener != null) {
     61             listener.onFragmentChange(mFragment, false);
     62         }
     63     }
     64 
     65     @Override
     66     public Fragment getFragment() {
     67         return mFragment;
     68     }
     69 
     70     /**
     71      * Fragment that needs user to select security type.
     72      */
     73     public static class ChooseSecurityFragment extends WifiConnectivityGuidedStepFragment {
     74         @VisibleForTesting
     75         static final long WIFI_SECURITY_TYPE_NONE = 100001;
     76         @VisibleForTesting
     77         static final long WIFI_SECURITY_TYPE_WEP = 100002;
     78         @VisibleForTesting
     79         static final long WIFI_SECURITY_TYPE_WPA = 100003;
     80         @VisibleForTesting
     81         static final long WIFI_SECURITY_TYPE_EAP = 100004;
     82         private StateMachine mStateMachine;
     83         private UserChoiceInfo mUserChoiceInfo;
     84 
     85         @Override
     86         public GuidanceStylist.Guidance onCreateGuidance(Bundle savedInstanceState) {
     87             return new GuidanceStylist.Guidance(
     88                     getString(R.string.security_type),
     89                     null,
     90                     null,
     91                     null);
     92         }
     93 
     94         @Override
     95         public void onCreate(Bundle savedInstanceState) {
     96             mUserChoiceInfo = ViewModelProviders
     97                     .of(getActivity())
     98                     .get(UserChoiceInfo.class);
     99             mStateMachine = ViewModelProviders
    100                     .of(getActivity())
    101                     .get(StateMachine.class);
    102             super.onCreate(savedInstanceState);
    103         }
    104 
    105         @Override
    106         public void onCreateActions(List<GuidedAction> actions, Bundle savedInstanceState) {
    107             Context context = getActivity();
    108             actions.add(new GuidedAction.Builder(context)
    109                     .title(R.string.wifi_security_type_none)
    110                     .id(WIFI_SECURITY_TYPE_NONE)
    111                     .build());
    112             actions.add(new GuidedAction.Builder(context)
    113                     .title(R.string.wifi_security_type_wep)
    114                     .id(WIFI_SECURITY_TYPE_WEP)
    115                     .build());
    116             actions.add(new GuidedAction.Builder(context)
    117                     .title(R.string.wifi_security_type_wpa)
    118                     .id(WIFI_SECURITY_TYPE_WPA)
    119                     .build());
    120             actions.add(new GuidedAction.Builder(context)
    121                     .title(R.string.wifi_security_type_eap)
    122                     .id(WIFI_SECURITY_TYPE_EAP)
    123                     .build());
    124         }
    125 
    126         @Override
    127         public void onGuidedActionClicked(GuidedAction action) {
    128             if (action.getId() == WIFI_SECURITY_TYPE_NONE) {
    129                 mUserChoiceInfo.setWifiSecurity(AccessPoint.SECURITY_NONE);
    130             } else if (action.getId() == WIFI_SECURITY_TYPE_WEP) {
    131                 mUserChoiceInfo.setWifiSecurity(AccessPoint.SECURITY_WEP);
    132             } else if (action.getId() == WIFI_SECURITY_TYPE_WPA) {
    133                 mUserChoiceInfo.setWifiSecurity(AccessPoint.SECURITY_PSK);
    134             } else if (action.getId() == WIFI_SECURITY_TYPE_EAP) {
    135                 mUserChoiceInfo.setWifiSecurity(AccessPoint.SECURITY_EAP);
    136             }
    137             WifiConfigHelper.setConfigKeyManagementBySecurity(
    138                     mUserChoiceInfo.getWifiConfiguration(),
    139                     mUserChoiceInfo.getWifiSecurity());
    140             if (action.getId() == WIFI_SECURITY_TYPE_NONE) {
    141                 mStateMachine.getListener().onComplete(StateMachine.OPTIONS_OR_CONNECT);
    142             } else {
    143                 mStateMachine.getListener().onComplete(StateMachine.PASSWORD);
    144             }
    145         }
    146     }
    147 }
    148