webkit/integration-tests/testapp/src/main/java/com/example/androidx/webkit/AssetLoaderInternalStorageActivity.java

/*
 * Copyright 2019 The Android Open Source Project
 *
 * 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.example.androidx.webkit;

import static java.nio.charset.StandardCharsets.UTF_8;

import android.app.Activity;
import android.net.Uri;
import android.os.Bundle;
import android.webkit.WebResourceRequest;
import android.webkit.WebResourceResponse;
import android.webkit.WebView;
import android.webkit.WebViewClient;

import androidx.annotation.NonNull;
import androidx.annotation.Nullable;
import androidx.annotation.RequiresApi;
import androidx.appcompat.app.AppCompatActivity;
import androidx.webkit.WebViewAssetLoader;
import androidx.webkit.WebViewAssetLoader.InternalStoragePathHandler;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;

/**
 * An {@link Activity} to show case a use case of using {@link InternalStoragePathHandler}.
 */
public class AssetLoaderInternalStorageActivity extends AppCompatActivity {
    private static final String DEMO_HTML_CONTENT =
            "<h3 id=\"data_success_msg\">Successfully loaded html from app files dir!</h3>";

    @NonNull private File mPublicDir;
    @NonNull private File mDemoFile;

    @NonNull private WebViewAssetLoader mAssetLoader;
    @NonNull private WebView mWebView;

    private class MyWebViewClient extends WebViewClient {
        @Override
        @SuppressWarnings("deprecation") // use the old one for compatibility with all API levels.
        public boolean shouldOverrideUrlLoading(WebView view, String url) {
            return false;
        }

        @Override
        @RequiresApi(21)
        public WebResourceResponse shouldInterceptRequest(WebView view,
                                            WebResourceRequest request) {
            return mAssetLoader.shouldInterceptRequest(request.getUrl());
        }

        @Override
        @SuppressWarnings("deprecation") // use the old one for compatibility with all API levels.
        public WebResourceResponse shouldInterceptRequest(WebView view, String url) {
            return mAssetLoader.shouldInterceptRequest(Uri.parse(url));
        }
    }

    @Override
    @SuppressWarnings("deprecation") /* AsyncTask */
    protected void onCreate(@Nullable Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);

        setContentView(R.layout.activity_asset_loader);
        setTitle(R.string.asset_loader_internal_storage_activity_title);
        WebkitHelpers.appendWebViewVersionToTitle(this);

        mWebView = findViewById(R.id.webview_asset_loader_webview);
        mWebView.setWebViewClient(new MyWebViewClient());

        mPublicDir = new File(getFilesDir(), "public");
        mDemoFile = new File(mPublicDir, "some_text.html");

        // Host "files/public/" in app's data directory under:
        // http://appassets.androidplatform.net/public_data/...
        mAssetLoader = new WebViewAssetLoader.Builder()
                .addPathHandler("/public_data/", new InternalStoragePathHandler(this, mPublicDir))
                .build();

        // Write the demo file asynchronously and then load the file after it's written.
        new WriteFileTask(mDemoFile, DEMO_HTML_CONTENT) {
            @Override
            protected void onPostExecute(Void result) {
                Uri path = new Uri.Builder()
                        .scheme("https")
                        .authority(WebViewAssetLoader.DEFAULT_DOMAIN)
                        .appendPath("public_data")
                        .appendPath("some_text.html")
                        .build();

                mWebView.loadUrl(path.toString());
            }
        }.execute();
    }

    @Override
    protected void onDestroy() {
        super.onDestroy();
        // Clean the test/demo file for tests.
        mDemoFile.delete();
        mPublicDir.delete();
    }

    // Writes to file asynchronously in the background thread.
    private static class WriteFileTask extends android.os.AsyncTask<Void, Void, Void> {
        @NonNull private final File mFile;
        @NonNull private final String mContent;

        WriteFileTask(@NonNull File file, @NonNull String content) {
            mFile = file;
            mContent = content;
        }

        @Override
        protected Void doInBackground(Void... params) {
            mFile.getParentFile().mkdirs();
            try (FileOutputStream fos = new FileOutputStream(mFile)) {
                fos.write(mContent.getBytes(UTF_8));
            } catch (IOException e) {
                throw new RuntimeException(e);
            }
            return null;
        }

    }
}