UsbStorageActivity.java

/* * Copyright (C) 2007 Google Inc. * * 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.android.server.status;import com.android.internal.R;import android.app.Activity;import android.app.ActivityManager;import android.app.AlertDialog;import android.app.Dialog;import android.content.BroadcastReceiver;import android.content.Context;import android.content.DialogInterface;import android.content.Intent;import android.content.IntentFilter;import android.content.DialogInterface.OnCancelListener;import android.content.pm.ApplicationInfo;import android.content.pm.PackageManager;import android.content.pm.PackageManager.NameNotFoundException;import android.os.Bundle;import android.os.Environment;import android.os.Handler;import android.os.IBinder;import android.os.storage.IMountService;import android.os.storage.StorageManager;import android.os.storage.StorageEventListener;import android.os.RemoteException;import android.os.ServiceManager;import android.widget.ImageView;import android.widget.Button;import android.widget.ProgressBar;import android.widget.TextView;import android.view.View;import android.view.Window;import android.util.Log;import java.util.List;/** * This activity is shown to the user for him/her to enable USB mass storage * on-demand (that is, when the USB cable is connected). It uses the alert * dialog style. It will be launched from a notification. */public class UsbStorageActivity extends Activity        implements View.OnClickListener, OnCancelListener {    private static final String TAG = "UsbStorageActivity";    private Button mMountButton;    private Button mUnmountButton;    private ProgressBar mProgressBar;    private TextView mBanner;    private TextView mMessage;    private ImageView mIcon;    private StorageManager mStorageManager = null;    private static final int DLG_CONFIRM_KILL_STORAGE_USERS = 1;    private static final int DLG_ERROR_SHARING = 2;    static final boolean localLOGV = false;    /** Used to detect when the USB cable is unplugged, so we can call finish() */    private BroadcastReceiver mBatteryReceiver = new BroadcastReceiver() {        @Override        public void onReceive(Context context, Intent intent) {            if (intent.getAction() == Intent.ACTION_BATTERY_CHANGED) {                handleBatteryChanged(intent);            }        }    };    private StorageEventListener mStorageListener = new StorageEventListener() {        @Override        public void onStorageStateChanged(String path, String oldState, String newState) {            final boolean on = newState.equals(Environment.MEDIA_SHARED);            switchDisplay(on);        }    };        @Override    protected void onCreate(Bundle savedInstanceState) {        super.onCreate(savedInstanceState);        if (mStorageManager == null) {            mStorageManager = (StorageManager) getSystemService(Context.STORAGE_SERVICE);            if (mStorageManager == null) {                Log.w(TAG, "Failed to get StorageManager");            }        }        requestWindowFeature(Window.FEATURE_INDETERMINATE_PROGRESS);        setProgressBarIndeterminateVisibility(true);        setTitle(getString(com.android.internal.R.string.usb_storage_activity_title));        setContentView(com.android.internal.R.layout.usb_storage_activity);        mIcon = (ImageView) findViewById(com.android.internal.R.id.icon);        mBanner = (TextView) findViewById(com.android.internal.R.id.banner);        mMessage = (TextView) findViewById(com.android.internal.R.id.message);        mMountButton = (Button) findViewById(com.android.internal.R.id.mount_button);        mMountButton.setOnClickListener(this);        mUnmountButton = (Button) findViewById(com.android.internal.R.id.unmount_button);        mUnmountButton.setOnClickListener(this);        mProgressBar = (ProgressBar) findViewById(com.android.internal.R.id.progress);    }    private void switchDisplay(boolean usbStorageInUse) {        if (usbStorageInUse) {            mProgressBar.setVisibility(View.GONE);            mUnmountButton.setVisibility(View.VISIBLE);            mMountButton.setVisibility(View.GONE);            mIcon.setImageResource(com.android.internal.R.drawable.usb_android_connected);            mBanner.setText(com.android.internal.R.string.usb_storage_stop_title);            mMessage.setText(com.android.internal.R.string.usb_storage_stop_message);        } else {            mProgressBar.setVisibility(View.GONE);            mUnmountButton.setVisibility(View.GONE);            mMountButton.setVisibility(View.VISIBLE);            mIcon.setImageResource(com.android.internal.R.drawable.usb_android);            mBanner.setText(com.android.internal.R.string.usb_storage_title);            mMessage.setText(com.android.internal.R.string.usb_storage_message);        }    }    @Override    protected void onResume() {        super.onResume();        mStorageManager.registerListener(mStorageListener);        registerReceiver(mBatteryReceiver, new IntentFilter(Intent.ACTION_BATTERY_CHANGED));        try {            switchDisplay(mStorageManager.isUsbMassStorageEnabled());        } catch (Exception ex) {            Log.e(TAG, "Failed to read UMS enable state", ex);        }    }    @Override    protected void onPause() {        super.onPause();                unregisterReceiver(mBatteryReceiver);        if (mStorageManager == null && mStorageListener != null) {            mStorageManager.unregisterListener(mStorageListener);        }    }    private void handleBatteryChanged(Intent intent) {        int pluggedType = intent.getIntExtra("plugged", 0);        if (pluggedType == 0) {            // It was disconnected from the plug, so finish            finish();        }    }    private IMountService getMountService() {        IBinder service = ServiceManager.getService("mount");        if (service != null) {            return IMountService.Stub.asInterface(service);        }        return null;    }    @Override    public Dialog onCreateDialog(int id, Bundle args) {        switch (id) {        case DLG_CONFIRM_KILL_STORAGE_USERS:            return new AlertDialog.Builder(this)                    .setTitle(R.string.dlg_confirm_kill_storage_users_title)                    .setPositiveButton(R.string.dlg_ok, new DialogInterface.OnClickListener() {                        public void onClick(DialogInterface dialog, int which) {                            switchUsbMassStorageAsync(true);                        }})                    .setNegativeButton(R.string.cancel, null)                    .setMessage(R.string.dlg_confirm_kill_storage_users_text)                    .setOnCancelListener(this)                    .create();        case DLG_ERROR_SHARING:            return new AlertDialog.Builder(this)                    .setTitle(R.string.dlg_error_title)                    .setNeutralButton(R.string.dlg_ok, null)                    .setMessage(R.string.usb_storage_error_message)                    .setOnCancelListener(this)                    .create();        }        return null;    }    private void showDialogInner(int id) {        removeDialog(id);        showDialog(id);    }    private void switchUsbMassStorageAsync(boolean on) {        mUnmountButton.setVisibility(View.GONE);        mMountButton.setVisibility(View.GONE);        mProgressBar.setVisibility(View.VISIBLE);        // will be hidden once USB mass storage kicks in (or fails)                final boolean _on = on;        new Thread() {            public void run() {                if (_on) {                    mStorageManager.enableUsbMassStorage();                } else {                    mStorageManager.disableUsbMassStorage();                }            }        }.start();    }    private void checkStorageUsers() {        IMountService ims = getMountService();        if (ims == null) {            // Display error dialog            showDialogInner(DLG_ERROR_SHARING);        }        String extStoragePath = Environment.getExternalStorageDirectory().toString();        boolean showDialog = false;        try {            int[] stUsers = ims.getStorageUsers(extStoragePath);            if (stUsers != null && stUsers.length > 0) {                showDialog = true;            } else {                // List of applications on sdcard.                ActivityManager am = (ActivityManager)getSystemService(Context.ACTIVITY_SERVICE);                List<ApplicationInfo> infoList = am.getRunningExternalApplications();                if (infoList != null && infoList.size() > 0) {                    showDialog = true;                }            }        } catch (RemoteException e) {            // Display error dialog            showDialogInner(DLG_ERROR_SHARING);        }        if (showDialog) {            // Display dialog to user            showDialogInner(DLG_CONFIRM_KILL_STORAGE_USERS);        } else {            if (localLOGV) Log.i(TAG, "Enabling UMS");            switchUsbMassStorageAsync(true);        }    }    public void onClick(View v) {        if (v == mMountButton) {           // Check for list of storage users and display dialog if needed.            checkStorageUsers();        } else if (v == mUnmountButton) {            if (localLOGV) Log.i(TAG, "Disabling UMS");            switchUsbMassStorageAsync(false);        }    }    public void onCancel(DialogInterface dialog) {        finish();    }}