Pileus Git - ~andy/freeotp/blob - src/org/fedorahosted/freeotp/TokenAdapter.java

   1 /*
   2  * FreeOTP
   3  *
   4  * Authors: Nathaniel McCallum <npmccallum@redhat.com>
   5  *
   6  * Copyright (C) 2013  Nathaniel McCallum, Red Hat
   7  * see file 'COPYING' for use and warranty information
   8  *
   9  * This program is free software you can redistribute it and/or modify
  10  * it under the terms of the GNU General Public License as published by
  11  * the Free Software Foundation, either version 3 of the License, or
  12  * (at your option) any later version.
  13  *
  14  * This program is distributed in the hope that it will be useful,
  15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17  * GNU General Public License for more details.
  18  *
  19  * You should have received a copy of the GNU General Public License
  20  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  21  */
  22
  23 package org.fedorahosted.freeotp;
  24
  25 import java.security.NoSuchAlgorithmException;
  26 import java.util.ArrayList;
  27 import java.util.Collections;
  28 import java.util.Comparator;
  29 import java.util.HashMap;
  30 import java.util.List;
  31 import java.util.Map;
  32
  33 import org.fedorahosted.freeotp.Token.TokenUriInvalidException;
  34
  35 import android.app.AlertDialog;
  36 import android.content.Context;
  37 import android.content.DialogInterface;
  38 import android.os.Handler;
  39 import android.os.Message;
  40 import android.view.View;
  41 import android.view.View.OnClickListener;
  42 import android.view.ViewGroup;
  43 import android.widget.BaseAdapter;
  44 import android.widget.ImageButton;
  45 import android.widget.ProgressBar;
  46 import android.widget.TextView;
  47
  48 public class TokenAdapter extends BaseAdapter {
  49         private static class Ticker extends Handler {
  50                 private static interface OnTickListener {
  51                         public void tick(ProgressBar pb);
  52                 }
  53
  54                 private final Map<ProgressBar, OnTickListener> map = new HashMap<ProgressBar, OnTickListener>();
  55
  56                 @Override
  57                 public void handleMessage(Message msg) {
  58                         for (ProgressBar pb : map.keySet())
  59                                 map.get(pb).tick(pb);
  60
  61                         sendEmptyMessageDelayed(0, 200);
  62                 }
  63
  64                 public void set(ProgressBar pb, OnTickListener otl) {
  65                         map.put(pb, otl);
  66                 }
  67         }
  68
  69         private final List<Token> tokens = new ArrayList<Token>();
  70         private final Ticker ticker = new Ticker();
  71
  72         private void sort() {
  73                 Collections.sort(tokens, new Comparator<Token>() {
  74                         @Override
  75                         public int compare(Token lhs, Token rhs) {
  76                                 return lhs.getTitle().compareTo(rhs.getTitle());
  77                         }
  78                 });
  79         }
  80
  81         public TokenAdapter(Context ctx) {
  82                 tokens.addAll(Token.getTokens(ctx));
  83                 ticker.sendEmptyMessageDelayed(0, 200);
  84                 sort();
  85         }
  86
  87         @Override
  88         public int getCount() {
  89                 return tokens.size();
  90         }
  91
  92         @Override
  93         public Token getItem(int position) {
  94                 return tokens.get(position);
  95         }
  96
  97         @Override
  98         public long getItemId(int position) {
  99                 return position;
 100         }
 101
 102         @Override
 103         public View getView(int position, View convertView, ViewGroup parent) {
 104                 final Context ctx = parent.getContext();
 105
 106                 if (convertView == null) {
 107                         switch (getItem(position).getType()) {
 108                         case HOTP:
 109                                 convertView = View.inflate(ctx, R.layout.hotp, null);
 110                                 break;
 111
 112                         case TOTP:
 113                                 convertView = View.inflate(ctx, R.layout.totp, null);
 114                                 break;
 115                         }
 116                 }
 117
 118                 final Token item = getItem(position);
 119                 final TextView code = (TextView) convertView.findViewById(R.id.code);
 120                 final TextView title = (TextView) convertView.findViewById(R.id.title);
 121                 final ImageButton ib = (ImageButton) convertView.findViewById(R.id.button);
 122
 123                 code.setText(item.getCurrentTokenValue(ctx, false));
 124                 title.setText(item.getTitle());
 125
 126                 ib.setOnClickListener(new OnClickListener() {
 127                         @Override
 128                         public void onClick(View v) {
 129                                 String delmsg = ctx.getString(R.string.delete_message);
 130
 131                                 AlertDialog ad = new AlertDialog.Builder(ctx)
 132                                 .setTitle("Delete")
 133                                 .setMessage(delmsg + item.getTitle())
 134                                 .setIcon(android.R.drawable.ic_delete)
 135                                 .setPositiveButton(R.string.delete,
 136                                                 new DialogInterface.OnClickListener() {
 137                                                         @Override
 138                                                         public void onClick(DialogInterface dialog, int which) {
 139                                                                 tokens.remove(tokens.indexOf(item));
 140                                                                 item.remove(ctx);
 141                                                                 notifyDataSetChanged();
 142                                                                 dialog.dismiss();
 143                                                         }
 144
 145                                                 })
 146                                 .setNegativeButton(android.R.string.cancel,
 147                                                 new DialogInterface.OnClickListener() {
 148                                                         @Override
 149                                                         public void onClick(DialogInterface dialog, int which) {
 150                                                                 dialog.cancel();
 151                                                         }
 152                                                 }).create();
 153                                 ad.show();
 154                         }
 155                 });
 156
 157                 switch (getItem(position).getType()) {
 158                 case HOTP:
 159                         ImageButton hotp = (ImageButton) convertView.findViewById(R.id.hotpButton);
 160                         hotp.setOnClickListener(new OnClickListener() {
 161                                 @Override
 162                                 public void onClick(View v) {
 163                                         code.setText(item.getCurrentTokenValue(ctx, true));
 164                                 }
 165                         });
 166                         break;
 167
 168                 case TOTP:
 169                         ProgressBar pb = (ProgressBar) convertView.findViewById(R.id.totpProgressBar);
 170                         ticker.set(pb, new Ticker.OnTickListener() {
 171                                 @Override
 172                                 public void tick(ProgressBar pb) {
 173                                         int max = pb.getMax();
 174                                         int pro = item.getProgress();
 175                                         pb.setProgress(max - pro);
 176                                         if (pro < max / 20 || pro > max / 20 * 19)
 177                                                 code.setText(item.getCurrentTokenValue(ctx, false));
 178                                 }
 179                         });
 180                         break;
 181                 }
 182
 183                 return convertView;
 184         }
 185
 186         @Override
 187         public int getViewTypeCount() {
 188                 return 2;
 189         }
 190
 191         @Override
 192         public int getItemViewType(int position) {
 193                 switch (getItem(position).getType()) {
 194                 case HOTP:
 195                         return 0;
 196                 case TOTP:
 197                         return 1;
 198                 default:
 199                         return -1;
 200                 }
 201         }
 202
 203         public void add(Context ctx, String uri) throws NoSuchAlgorithmException, TokenUriInvalidException {
 204                 Token t = new Token(uri);
 205                 t.save(ctx);
 206                 tokens.add(t);
 207                 sort();
 208                 notifyDataSetChanged();
 209         }
 210 }