diff src/com/five_ten_sg/connectbot/PortForwardListActivity.java @ 0:0ce5cc452d02

initial version
author Carl Byington <carl@five-ten-sg.com>
date Thu, 22 May 2014 10:41:19 -0700
parents
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/src/com/five_ten_sg/connectbot/PortForwardListActivity.java	Thu May 22 10:41:19 2014 -0700
@@ -0,0 +1,406 @@
+/*
+ * ConnectBot: simple, powerful, open-source SSH client for Android
+ * Copyright 2007 Kenny Root, Jeffrey Sharkey
+ *
+ * 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.five_ten_sg.connectbot;
+
+import java.util.List;
+
+import com.five_ten_sg.connectbot.bean.HostBean;
+import com.five_ten_sg.connectbot.bean.PortForwardBean;
+import com.five_ten_sg.connectbot.service.TerminalBridge;
+import com.five_ten_sg.connectbot.service.TerminalManager;
+import com.five_ten_sg.connectbot.util.HostDatabase;
+import android.app.AlertDialog;
+import android.app.ListActivity;
+import android.content.ComponentName;
+import android.content.Context;
+import android.content.DialogInterface;
+import android.content.Intent;
+import android.content.ServiceConnection;
+import android.content.res.Resources;
+import android.database.SQLException;
+import android.graphics.Paint;
+import android.os.Bundle;
+import android.os.Handler;
+import android.os.IBinder;
+import android.os.Message;
+import android.util.Log;
+import android.view.ContextMenu;
+import android.view.LayoutInflater;
+import android.view.Menu;
+import android.view.MenuItem;
+import android.view.MenuItem.OnMenuItemClickListener;
+import android.view.View;
+import android.view.ViewGroup;
+import android.widget.AdapterView;
+import android.widget.AdapterView.OnItemClickListener;
+import android.widget.AdapterView.OnItemSelectedListener;
+import android.widget.ArrayAdapter;
+import android.widget.EditText;
+import android.widget.ListView;
+import android.widget.Spinner;
+import android.widget.TextView;
+import android.widget.Toast;
+
+/**
+ * List all portForwards for a particular host and provide a way for users to add more portForwards,
+ * edit existing portForwards, and delete portForwards.
+ *
+ * @author Kenny Root
+ */
+public class PortForwardListActivity extends ListActivity {
+    public final static String TAG = "ConnectBot.PortForwardListActivity";
+
+    private static final int LISTENER_CYCLE_TIME = 500;
+
+    protected HostDatabase hostdb;
+
+    private List<PortForwardBean> portForwards;
+
+    private ServiceConnection connection = null;
+    protected TerminalBridge hostBridge = null;
+    protected LayoutInflater inflater = null;
+
+    private HostBean host;
+
+    @Override
+    public void onStart() {
+        super.onStart();
+        this.bindService(new Intent(this, TerminalManager.class), connection, Context.BIND_AUTO_CREATE);
+
+        if (this.hostdb == null)
+            this.hostdb = new HostDatabase(this);
+    }
+
+    @Override
+    public void onStop() {
+        super.onStop();
+        this.unbindService(connection);
+
+        if (this.hostdb != null) {
+            this.hostdb.close();
+            this.hostdb = null;
+        }
+    }
+
+    @Override
+    public void onCreate(Bundle icicle) {
+        super.onCreate(icicle);
+        long hostId = this.getIntent().getLongExtra(Intent.EXTRA_TITLE, -1);
+        setContentView(R.layout.act_portforwardlist);
+        // connect with hosts database and populate list
+        this.hostdb = new HostDatabase(this);
+        host = hostdb.findHostById(hostId);
+        {
+            final String nickname = host != null ? host.getNickname() : null;
+            final Resources resources = getResources();
+
+            if (nickname != null) {
+                this.setTitle(String.format("%s: %s (%s)",
+                                            resources.getText(R.string.app_name),
+                                            resources.getText(R.string.title_port_forwards_list),
+                                            nickname));
+            }
+            else {
+                this.setTitle(String.format("%s: %s",
+                                            resources.getText(R.string.app_name),
+                                            resources.getText(R.string.title_port_forwards_list)));
+            }
+        }
+        connection = new ServiceConnection() {
+            public void onServiceConnected(ComponentName className, IBinder service) {
+                TerminalManager bound = ((TerminalManager.TerminalBinder) service).getService();
+                hostBridge = bound.getConnectedBridge(host);
+                updateHandler.sendEmptyMessage(-1);
+            }
+            public void onServiceDisconnected(ComponentName name) {
+                hostBridge = null;
+            }
+        };
+        this.updateList();
+        this.registerForContextMenu(this.getListView());
+        this.getListView().setOnItemClickListener(new OnItemClickListener() {
+            public void onItemClick(AdapterView<?> adapter, View view, int position, long id) {
+                ListView lv = PortForwardListActivity.this.getListView();
+                PortForwardBean pfb = (PortForwardBean) lv.getItemAtPosition(position);
+
+                if (hostBridge != null) {
+                    if (pfb.isEnabled())
+                        hostBridge.disablePortForward(pfb);
+                    else {
+                        if (!hostBridge.enablePortForward(pfb))
+                            Toast.makeText(PortForwardListActivity.this, getString(R.string.portforward_problem), Toast.LENGTH_LONG).show();
+                    }
+
+                    updateHandler.sendEmptyMessage(-1);
+                }
+            }
+        });
+        this.inflater = LayoutInflater.from(this);
+    }
+
+    @Override
+    public boolean onCreateOptionsMenu(Menu menu) {
+        super.onCreateOptionsMenu(menu);
+        MenuItem add = menu.add(R.string.portforward_menu_add);
+        add.setIcon(android.R.drawable.ic_menu_add);
+        add.setOnMenuItemClickListener(new OnMenuItemClickListener() {
+            public boolean onMenuItemClick(MenuItem item) {
+                // build dialog to prompt user about updating
+                final View portForwardView = inflater.inflate(R.layout.dia_portforward, null, false);
+                final EditText destEdit = (EditText) portForwardView.findViewById(R.id.portforward_destination);
+                final Spinner typeSpinner = (Spinner)portForwardView.findViewById(R.id.portforward_type);
+                typeSpinner.setOnItemSelectedListener(new OnItemSelectedListener() {
+                    public void onItemSelected(AdapterView<?> value, View view,
+                                               int position, long id) {
+                        destEdit.setEnabled(position != 2);
+                    }
+                    public void onNothingSelected(AdapterView<?> arg0) {
+                    }
+                });
+                new AlertDialog.Builder(PortForwardListActivity.this)
+                .setView(portForwardView)
+                .setPositiveButton(R.string.portforward_pos, new DialogInterface.OnClickListener() {
+                    public void onClick(DialogInterface dialog, int which) {
+                        try {
+                            final EditText nicknameEdit = (EditText) portForwardView.findViewById(R.id.nickname);
+                            final EditText sourcePortEdit = (EditText) portForwardView.findViewById(R.id.portforward_source);
+                            String type = HostDatabase.PORTFORWARD_LOCAL;
+
+                            switch (typeSpinner.getSelectedItemPosition()) {
+                                case 0:
+                                    type = HostDatabase.PORTFORWARD_LOCAL;
+                                    break;
+
+                                case 1:
+                                    type = HostDatabase.PORTFORWARD_REMOTE;
+                                    break;
+
+                                case 2:
+                                    type = HostDatabase.PORTFORWARD_DYNAMIC5;
+                                    break;
+                            }
+
+                            PortForwardBean pfb = new PortForwardBean(
+                                host != null ? host.getId() : -1,
+                                nicknameEdit.getText().toString(), type,
+                                sourcePortEdit.getText().toString(),
+                                destEdit.getText().toString());
+
+                            if (hostBridge != null) {
+                                hostBridge.addPortForward(pfb);
+                                hostBridge.enablePortForward(pfb);
+                            }
+
+                            if (host != null && !hostdb.savePortForward(pfb))
+                                throw new SQLException("Could not save port forward");
+
+                            updateHandler.sendEmptyMessage(-1);
+                        }
+                        catch (Exception e) {
+                            Log.e(TAG, "Could not update port forward", e);
+                            // TODO Show failure dialog.
+                        }
+                    }
+                })
+                .setNegativeButton(R.string.delete_neg, null).create().show();
+                return true;
+            }
+        });
+        return true;
+    }
+
+    @Override
+    public void onCreateContextMenu(ContextMenu menu, View v, ContextMenu.ContextMenuInfo menuInfo) {
+        // Create menu to handle deleting and editing port forward
+        AdapterView.AdapterContextMenuInfo info = (AdapterView.AdapterContextMenuInfo) menuInfo;
+        final PortForwardBean pfb = (PortForwardBean) this.getListView().getItemAtPosition(info.position);
+        menu.setHeaderTitle(pfb.getNickname());
+        MenuItem edit = menu.add(R.string.portforward_edit);
+        edit.setOnMenuItemClickListener(new OnMenuItemClickListener() {
+            public boolean onMenuItemClick(MenuItem item) {
+                final View editTunnelView = inflater.inflate(R.layout.dia_portforward, null, false);
+                final Spinner typeSpinner = (Spinner) editTunnelView.findViewById(R.id.portforward_type);
+
+                if (HostDatabase.PORTFORWARD_LOCAL.equals(pfb.getType()))
+                    typeSpinner.setSelection(0);
+                else if (HostDatabase.PORTFORWARD_REMOTE.equals(pfb.getType()))
+                    typeSpinner.setSelection(1);
+                else
+                    typeSpinner.setSelection(2);
+
+                final EditText nicknameEdit = (EditText) editTunnelView.findViewById(R.id.nickname);
+                nicknameEdit.setText(pfb.getNickname());
+                final EditText sourcePortEdit = (EditText) editTunnelView.findViewById(R.id.portforward_source);
+                sourcePortEdit.setText(String.valueOf(pfb.getSourcePort()));
+                final EditText destEdit = (EditText) editTunnelView.findViewById(R.id.portforward_destination);
+
+                if (HostDatabase.PORTFORWARD_DYNAMIC5.equals(pfb.getType())) {
+                    destEdit.setEnabled(false);
+                }
+                else {
+                    destEdit.setText(String.format("%s:%d", pfb.getDestAddr(), pfb.getDestPort()));
+                }
+
+                typeSpinner.setOnItemSelectedListener(new OnItemSelectedListener() {
+                    public void onItemSelected(AdapterView<?> value, View view,
+                                               int position, long id) {
+                        destEdit.setEnabled(position != 2);
+                    }
+                    public void onNothingSelected(AdapterView<?> arg0) {
+                    }
+                });
+                new AlertDialog.Builder(PortForwardListActivity.this)
+                .setView(editTunnelView)
+                .setPositiveButton(R.string.button_change, new DialogInterface.OnClickListener() {
+                    public void onClick(DialogInterface dialog, int which) {
+                        try {
+                            if (hostBridge != null)
+                                hostBridge.disablePortForward(pfb);
+
+                            pfb.setNickname(nicknameEdit.getText().toString());
+
+                            switch (typeSpinner.getSelectedItemPosition()) {
+                                case 0:
+                                    pfb.setType(HostDatabase.PORTFORWARD_LOCAL);
+                                    break;
+
+                                case 1:
+                                    pfb.setType(HostDatabase.PORTFORWARD_REMOTE);
+                                    break;
+
+                                case 2:
+                                    pfb.setType(HostDatabase.PORTFORWARD_DYNAMIC5);
+                                    break;
+                            }
+
+                            pfb.setSourcePort(Integer.parseInt(sourcePortEdit.getText().toString()));
+                            pfb.setDest(destEdit.getText().toString());
+
+                            // Use the new settings for the existing connection.
+                            if (hostBridge != null)
+                                updateHandler.postDelayed(new Runnable() {
+                                public void run() {
+                                    hostBridge.enablePortForward(pfb);
+                                    updateHandler.sendEmptyMessage(-1);
+                                }
+                            }, LISTENER_CYCLE_TIME);
+
+                            if (!hostdb.savePortForward(pfb))
+                                throw new SQLException("Could not save port forward");
+
+                            updateHandler.sendEmptyMessage(-1);
+                        }
+                        catch (Exception e) {
+                            Log.e(TAG, "Could not update port forward", e);
+                            // TODO Show failure dialog.
+                        }
+                    }
+                })
+                .setNegativeButton(android.R.string.cancel, null).create().show();
+                return true;
+            }
+        });
+        MenuItem delete = menu.add(R.string.portforward_delete);
+        delete.setOnMenuItemClickListener(new OnMenuItemClickListener() {
+            public boolean onMenuItemClick(MenuItem item) {
+                // prompt user to make sure they really want this
+                new AlertDialog.Builder(PortForwardListActivity.this)
+                .setMessage(getString(R.string.delete_message, pfb.getNickname()))
+                .setPositiveButton(R.string.delete_pos, new DialogInterface.OnClickListener() {
+                    public void onClick(DialogInterface dialog, int which) {
+                        try {
+                            // Delete the port forward from the host if needed.
+                            if (hostBridge != null)
+                                hostBridge.removePortForward(pfb);
+
+                            hostdb.deletePortForward(pfb);
+                        }
+                        catch (Exception e) {
+                            Log.e(TAG, "Could not delete port forward", e);
+                        }
+
+                        updateHandler.sendEmptyMessage(-1);
+                    }
+                })
+                .setNegativeButton(R.string.delete_neg, null).create().show();
+                return true;
+            }
+        });
+    }
+
+    protected Handler updateHandler = new Handler() {
+        @Override
+        public void handleMessage(Message msg) {
+            PortForwardListActivity.this.updateList();
+        }
+    };
+
+    protected void updateList() {
+        if (hostBridge != null) {
+            this.portForwards = hostBridge.getPortForwards();
+        }
+        else {
+            if (this.hostdb == null) return;
+
+            this.portForwards = this.hostdb.getPortForwardsForHost(host);
+        }
+
+        PortForwardAdapter adapter = new PortForwardAdapter(this, portForwards);
+        this.setListAdapter(adapter);
+    }
+
+    class PortForwardAdapter extends ArrayAdapter<PortForwardBean> {
+        class ViewHolder {
+            public TextView nickname;
+            public TextView caption;
+        }
+
+        private List<PortForwardBean> portForwards;
+
+        public PortForwardAdapter(Context context, List<PortForwardBean> portForwards) {
+            super(context, R.layout.item_portforward, portForwards);
+            this.portForwards = portForwards;
+        }
+
+        @Override
+        public View getView(int position, View convertView, ViewGroup parent) {
+            ViewHolder holder;
+
+            if (convertView == null) {
+                convertView = inflater.inflate(R.layout.item_portforward, null, false);
+                holder = new ViewHolder();
+                holder.nickname = (TextView)convertView.findViewById(android.R.id.text1);
+                holder.caption = (TextView)convertView.findViewById(android.R.id.text2);
+                convertView.setTag(holder);
+            }
+            else
+                holder = (ViewHolder) convertView.getTag();
+
+            PortForwardBean pfb = portForwards.get(position);
+            holder.nickname.setText(pfb.getNickname());
+            holder.caption.setText(pfb.getDescription());
+
+            if (hostBridge != null && !pfb.isEnabled()) {
+                holder.nickname.setPaintFlags(holder.nickname.getPaintFlags() | Paint.STRIKE_THRU_TEXT_FLAG);
+                holder.caption.setPaintFlags(holder.caption.getPaintFlags() | Paint.STRIKE_THRU_TEXT_FLAG);
+            }
+
+            return convertView;
+        }
+    }
+}