129 lines
4.7 KiB
Java
129 lines
4.7 KiB
Java
/*
|
|
* Copyright © 2018 Samuel Holland <samuel@sholland.org>
|
|
* Copyright © 2018 Jason A. Donenfeld <Jason@zx2c4.com>. All Rights Reserved.
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*/
|
|
|
|
package com.wireguard.android.fragment;
|
|
|
|
import android.content.Context;
|
|
import android.content.Intent;
|
|
import android.databinding.DataBindingUtil;
|
|
import android.databinding.ViewDataBinding;
|
|
import android.support.annotation.Nullable;
|
|
import android.support.design.widget.Snackbar;
|
|
import android.support.v4.app.Fragment;
|
|
import android.util.Log;
|
|
import android.view.View;
|
|
import android.widget.Toast;
|
|
|
|
import com.wireguard.android.Application;
|
|
import com.wireguard.android.R;
|
|
import com.wireguard.android.activity.BaseActivity;
|
|
import com.wireguard.android.activity.BaseActivity.OnSelectedTunnelChangedListener;
|
|
import com.wireguard.android.backend.GoBackend;
|
|
import com.wireguard.android.databinding.TunnelDetailFragmentBinding;
|
|
import com.wireguard.android.databinding.TunnelListItemBinding;
|
|
import com.wireguard.android.model.Tunnel;
|
|
import com.wireguard.android.model.Tunnel.State;
|
|
import com.wireguard.android.util.ExceptionLoggers;
|
|
|
|
/**
|
|
* Base class for fragments that need to know the currently-selected tunnel. Only does anything when
|
|
* attached to a {@code BaseActivity}.
|
|
*/
|
|
|
|
public abstract class BaseFragment extends Fragment implements OnSelectedTunnelChangedListener {
|
|
private static final String TAG = "WireGuard/" + BaseFragment.class.getSimpleName();
|
|
private static final int REQUEST_CODE_VPN_PERMISSION = 23491;
|
|
|
|
@Nullable private BaseActivity activity;
|
|
@Nullable private Tunnel pendingTunnel;
|
|
@Nullable private Boolean pendingTunnelUp;
|
|
|
|
@Nullable
|
|
protected Tunnel getSelectedTunnel() {
|
|
return activity != null ? activity.getSelectedTunnel() : null;
|
|
}
|
|
|
|
@Override
|
|
public void onAttach(final Context context) {
|
|
super.onAttach(context);
|
|
if (context instanceof BaseActivity) {
|
|
activity = (BaseActivity) context;
|
|
activity.addOnSelectedTunnelChangedListener(this);
|
|
} else {
|
|
activity = null;
|
|
}
|
|
}
|
|
|
|
@Override
|
|
public void onDetach() {
|
|
if (activity != null)
|
|
activity.removeOnSelectedTunnelChangedListener(this);
|
|
activity = null;
|
|
super.onDetach();
|
|
}
|
|
|
|
@Override
|
|
public void onActivityResult(final int requestCode, final int resultCode, @Nullable final Intent data) {
|
|
super.onActivityResult(requestCode, resultCode, data);
|
|
|
|
if (requestCode == REQUEST_CODE_VPN_PERMISSION) {
|
|
if (pendingTunnel != null && pendingTunnelUp != null)
|
|
setTunnelStateWithPermissionsResult(pendingTunnel, pendingTunnelUp);
|
|
pendingTunnel = null;
|
|
pendingTunnelUp = null;
|
|
}
|
|
}
|
|
|
|
protected void setSelectedTunnel(@Nullable final Tunnel tunnel) {
|
|
if (activity != null)
|
|
activity.setSelectedTunnel(tunnel);
|
|
}
|
|
|
|
public void setTunnelState(final View view, final boolean checked) {
|
|
final ViewDataBinding binding = DataBindingUtil.findBinding(view);
|
|
final Tunnel tunnel;
|
|
if (binding instanceof TunnelDetailFragmentBinding)
|
|
tunnel = ((TunnelDetailFragmentBinding) binding).getTunnel();
|
|
else if (binding instanceof TunnelListItemBinding)
|
|
tunnel = ((TunnelListItemBinding) binding).getItem();
|
|
else
|
|
return;
|
|
if (tunnel == null)
|
|
return;
|
|
|
|
Application.onHaveBackend(backend -> {
|
|
if (backend instanceof GoBackend) {
|
|
final Intent intent = GoBackend.VpnService.prepare(view.getContext());
|
|
if (intent != null) {
|
|
pendingTunnel = tunnel;
|
|
pendingTunnelUp = checked;
|
|
startActivityForResult(intent, REQUEST_CODE_VPN_PERMISSION);
|
|
return;
|
|
}
|
|
}
|
|
|
|
setTunnelStateWithPermissionsResult(tunnel, checked);
|
|
});
|
|
}
|
|
|
|
private void setTunnelStateWithPermissionsResult(final Tunnel tunnel, final boolean checked) {
|
|
tunnel.setState(State.of(checked)).whenComplete((state, throwable) -> {
|
|
if (throwable == null)
|
|
return;
|
|
final String error = ExceptionLoggers.unwrapMessage(throwable);
|
|
final int messageResId = checked ? R.string.error_up : R.string.error_down;
|
|
final String message = getContext().getString(messageResId, error);
|
|
final View view = getView();
|
|
if (view != null)
|
|
Snackbar.make(view, message, Snackbar.LENGTH_LONG).show();
|
|
else
|
|
Toast.makeText(getContext(), message, Toast.LENGTH_LONG).show();
|
|
Log.e(TAG, message, throwable);
|
|
});
|
|
}
|
|
|
|
}
|