EhViewer/app/src/main/java/com/hippo/ehviewer/ui/UConfigActivity.java

/*
 * Copyright 2018 Hippo Seven
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.hippo.ehviewer.ui;

/*
 * Created by Hippo on 2018/2/9.
 */

import android.annotation.SuppressLint;
import android.content.Context;
import android.content.DialogInterface;
import android.graphics.Bitmap;
import android.os.Build;
import android.os.Bundle;
import android.support.annotation.Nullable;
import android.support.design.widget.Snackbar;
import android.support.v7.app.AlertDialog;
import android.view.Menu;
import android.view.MenuItem;
import android.view.View;
import android.webkit.CookieManager;
import android.webkit.CookieSyncManager;
import android.webkit.JsPromptResult;
import android.webkit.JsResult;
import android.webkit.WebChromeClient;
import android.webkit.WebView;
import android.webkit.WebViewClient;
import android.widget.EditText;
import android.widget.TextView;
import com.hippo.ehviewer.EhApplication;
import com.hippo.ehviewer.R;
import com.hippo.ehviewer.client.EhCookieStore;
import com.hippo.ehviewer.client.EhUrl;
import com.hippo.widget.ProgressView;
import okhttp3.Cookie;
import okhttp3.HttpUrl;

public class UConfigActivity extends ToolbarActivity {

  private WebView webView;
  private ProgressView progress;
  private String url;
  private boolean loaded;

  @SuppressLint("SetJavaScriptEnabled")
  @Override
  protected void onCreate(@Nullable Bundle savedInstanceState) {
    super.onCreate(savedInstanceState);

    // http://stackoverflow.com/questions/32284642/how-to-handle-an-uncatched-exception
    CookieManager cookieManager = CookieManager.getInstance();
    if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.LOLLIPOP) {
      cookieManager.flush();
      cookieManager.removeAllCookies(null);
      cookieManager.removeSessionCookies(null);
    } else {
      CookieSyncManager cookieSyncManager = CookieSyncManager.createInstance(this);
      cookieSyncManager.startSync();
      cookieManager.removeAllCookie();
      cookieManager.removeSessionCookie();
      cookieSyncManager.stopSync();
    }

    // Copy cookies from okhttp cookie store to CookieManager
    url = EhUrl.getUConfigUrl();
    EhCookieStore store = EhApplication.getEhCookieStore(this);
    for (Cookie cookie : store.getCookies(HttpUrl.parse(url))) {
      cookieManager.setCookie(url, cookie.toString());
    }

    setContentView(R.layout.activity_u_config);
    setNavigationIcon(R.drawable.v_arrow_left_dark_x24);
    webView = (WebView) findViewById(R.id.webview);
    webView.getSettings().setJavaScriptEnabled(true);
    webView.setWebViewClient(new UConfigWebViewClient());
    webView.setWebChromeClient(new UConfigWebChromeClient());
    webView.loadUrl(url);
    progress = (ProgressView) findViewById(R.id.progress);

    Snackbar.make(webView, R.string.apply_tip, Snackbar.LENGTH_LONG).show();
  }

  private void apply() {
    webView.loadUrl("javascript:"
        + "(function() {\n"
        + "    var apply = document.getElementById(\"apply\").children[0];\n"
        + "    apply.click();\n"
        + "})();");
  }

  @Override
  public boolean onCreateOptionsMenu(Menu menu) {
    super.onCreateOptionsMenu(menu);
    getMenuInflater().inflate(R.menu.activity_u_config, menu);
    return true;
  }

  @Override
  public boolean onOptionsItemSelected(MenuItem item) {
    switch (item.getItemId()) {
      case android.R.id.home:
        finish();
        return true;
      case R.id.action_apply:
        if (loaded) {
          apply();
        }
        return true;
      default:
        return super.onOptionsItemSelected(item);
    }
  }

  private Cookie longLive(Cookie cookie) {
    return new Cookie.Builder()
        .name(cookie.name())
        .value(cookie.value())
        .domain(cookie.domain())
        .path(cookie.path())
        .expiresAt(Long.MAX_VALUE)
        .build();
  }

  @Override
  protected void onDestroy() {
    super.onDestroy();
    webView.destroy();

    // Put cookies back to okhttp cookie store
    CookieManager cookieManager = CookieManager.getInstance();
    String cookiesString = cookieManager.getCookie(url);

    if (cookiesString != null && !cookiesString.isEmpty()) {
      EhCookieStore store = EhApplication.getEhCookieStore(this);
      HttpUrl eUrl = HttpUrl.parse(EhUrl.HOST_E);
      HttpUrl exUrl = HttpUrl.parse(EhUrl.HOST_EX);

      // The cookies saved in the uconfig page should be shared between e and ex
      for (String header : cookiesString.split(";")) {
        Cookie eCookie = Cookie.parse(eUrl, header);
        if (eCookie != null) {
          store.addCookie(longLive(eCookie));
        }

        Cookie exCookie = Cookie.parse(exUrl, header);
        if (exCookie != null) {
          store.addCookie(longLive(exCookie));
        }
      }
    }
  }

  private class UConfigWebViewClient extends WebViewClient {
    @Override
    public boolean shouldOverrideUrlLoading(WebView view, String url) {
      // Never load other urls
      return true;
    }

    @Override
    public void onPageStarted(WebView view, String url, Bitmap favicon) {
      progress.setVisibility(View.VISIBLE);
      loaded = false;
    }

    @Override
    public void onPageFinished(WebView view, String url) {
      progress.setVisibility(View.GONE);
      loaded = true;
    }
  }

  private class UConfigWebChromeClient extends WebChromeClient {

    private Context context = UConfigActivity.this;

    @Override
    public boolean onJsPrompt(WebView view, String url, String message, String defaultValue,
        final JsPromptResult result) {

      View promptView = UConfigActivity.this.getLayoutInflater()
          .inflate(R.layout.dialog_js_prompt, null, false);
      TextView messageView = (TextView) promptView.findViewById(R.id.message);
      messageView.setText(message);
      final EditText valueView = (EditText) promptView.findViewById(R.id.value);
      valueView.setText(defaultValue);

      new AlertDialog.Builder(context)
          .setView(promptView)
          .setPositiveButton(android.R.string.ok, new DialogInterface.OnClickListener() {
            @Override
            public void onClick(DialogInterface dialog, int which) {
              result.confirm(valueView.getText().toString());
            }
          })
          .setOnCancelListener(new DialogInterface.OnCancelListener() {
            @Override
            public void onCancel(DialogInterface dialog) {
              result.cancel();
            }
          })
          .show();

      return true;
    }

    private void showMessageDialog(String message, final JsResult result) {
      new AlertDialog.Builder(context)
          .setMessage(message)
          .setPositiveButton(android.R.string.ok, new DialogInterface.OnClickListener() {
            @Override
            public void onClick(DialogInterface dialog, int which) {
              result.confirm();
            }
          })
          .setOnCancelListener(new DialogInterface.OnCancelListener() {
            @Override
            public void onCancel(DialogInterface dialog) {
              result.cancel();
            }
          })
          .show();
    }

    @Override
    public boolean onJsAlert(WebView view, String url, String message, final JsResult result) {
      showMessageDialog(message, result);
      return true;
    }

    @Override
    public boolean onJsConfirm(WebView view, String url, String message, final JsResult result) {
      showMessageDialog(message, result);
      return true;
    }
  }
}