session-android/src/org/thoughtcrime/securesms/components/EmojiDrawer.java

package org.thoughtcrime.securesms.components;

import android.content.Context;
import android.graphics.drawable.Drawable;
import android.support.v4.view.PagerAdapter;
import android.support.v4.view.PagerTabStrip;
import android.support.v4.view.ViewPager;
import android.text.SpannableString;
import android.text.Spanned;
import android.text.style.ImageSpan;
import android.util.AttributeSet;
import android.view.LayoutInflater;
import android.view.View;
import android.view.ViewGroup;
import android.widget.AdapterView;
import android.widget.BaseAdapter;
import android.widget.EditText;
import android.widget.FrameLayout;
import android.widget.GridView;
import android.widget.ImageView;
import android.widget.TextView;

import org.thoughtcrime.securesms.R;
import org.thoughtcrime.securesms.util.Emoji;

public class EmojiDrawer extends FrameLayout {

  private static final int RECENT_TYPE = 0;
  private static final int ALL_TYPE    = 1;

  private FrameLayout   emojiGridLayout;
  private FrameLayout   recentEmojiGridLayout;
  private EditText      composeText;
  private Emoji         emoji;
  private GridView      emojiGrid;
  private GridView      recentEmojiGrid;
  private ViewPager     pager;
  private PagerTabStrip pagerTabStrip;

  public EmojiDrawer(Context context) {
    super(context);
    initialize();
  }

  public EmojiDrawer(Context context, AttributeSet attrs) {
    super(context, attrs);
    initialize();
  }

  public EmojiDrawer(Context context, AttributeSet attrs, int defStyle) {
    super(context, attrs, defStyle);
    initialize();
  }

  public void setComposeEditText(EditText composeText) {
    this.composeText = composeText;
  }

  public boolean isOpen() {
    return getVisibility() == View.VISIBLE;
  }

  private void initialize() {
    LayoutInflater inflater = (LayoutInflater)getContext().getSystemService(Context.LAYOUT_INFLATER_SERVICE);
    inflater.inflate(R.layout.emoji_drawer, this, true);

    initializeResources();
    initializeEmojiGrid();
  }

  private void initializeResources() {
    LayoutInflater inflater    = (LayoutInflater) getContext().getSystemService(Context.LAYOUT_INFLATER_SERVICE);
    this.pager                 = (ViewPager    )  findViewById(R.id.emoji_pager    );
    this.pagerTabStrip         = (PagerTabStrip)  findViewById(R.id.emoji_tab_strip);
    this.emojiGridLayout       = (FrameLayout  )  inflater.inflate(R.layout.emoji_grid_layout, null);
    this.recentEmojiGridLayout = (FrameLayout  )  inflater.inflate(R.layout.emoji_grid_layout, null);
    this.emojiGrid             = (GridView     )  emojiGridLayout.findViewById(R.id.emoji);
    this.recentEmojiGrid       = (GridView     )  recentEmojiGridLayout.findViewById(R.id.emoji);
    this.emoji                 = Emoji.getInstance(getContext());
  }

  private void initializeEmojiGrid() {
    emojiGrid.setAdapter(new EmojiGridAdapter(ALL_TYPE));
    emojiGrid.setOnItemClickListener(new EmojiClickListener(ALL_TYPE));
    recentEmojiGrid.setAdapter(new EmojiGridAdapter(RECENT_TYPE));
    recentEmojiGrid.setOnItemClickListener(new EmojiClickListener(RECENT_TYPE));

    pager.setAdapter(new EmojiPagerAdapter());

    if (emoji.getRecentlyUsedAssetCount() <= 0) {
      pager.setCurrentItem(1);
    }
  }

  private class EmojiClickListener implements AdapterView.OnItemClickListener {

    private final int type;

    public EmojiClickListener(int type) {
      this.type = type;
    }

    @Override
    public void onItemClick(AdapterView<?> parent, View view, int position, long id) {
      String characters;

      if (type == ALL_TYPE ) characters = emoji.getEmojiUnicode(position);
      else                   characters = emoji.getRecentEmojiUnicode(position);

      int start = composeText.getSelectionStart();
      int end   = composeText.getSelectionEnd  ();

      composeText.getText().replace(Math.min(start, end), Math.max(start, end),
                                    characters, 0, characters.length());

      composeText.setText(emoji.emojify(composeText.getText().toString()),
                          TextView.BufferType.SPANNABLE);

      composeText.setSelection(end+2);

      if (type != RECENT_TYPE) {
        emoji.setRecentlyUsed(position);
        ((BaseAdapter)recentEmojiGrid.getAdapter()).notifyDataSetChanged();
      }
    }
  }

  private class EmojiGridAdapter extends BaseAdapter {

    private final int type;

    public EmojiGridAdapter(int type) {
      this.type = type;
    }

    @Override
    public int getCount() {
      if (type == RECENT_TYPE) return emoji.getRecentlyUsedAssetCount();
      else                     return emoji.getEmojiAssetCount();

    }

    @Override
    public Object getItem(int position) {
      return null;
    }

    @Override
    public long getItemId(int position) {
      return 0;
    }

    @Override
    public View getView(int position, View convertView, ViewGroup parent) {
      Drawable drawable;

      if (type == RECENT_TYPE) drawable = emoji.getRecentlyUsed(position);
      else                     drawable = emoji.getEmojiDrawable(position);

      if (convertView != null && convertView instanceof ImageView) {
        ((ImageView)convertView).setImageDrawable(drawable);
        return convertView;
      } else {
        ImageView imageView = new ImageView(getContext());
        imageView.setImageDrawable(drawable);
        return imageView;
      }
    }
  }

  private class EmojiPagerAdapter extends PagerAdapter {

    @Override
    public int getCount() {
      return 2;
    }

    @Override
    public CharSequence getPageTitle(int position) {
      switch (position) {
        case 0:
          SpannableString recent      = new SpannableString(" Recent ");
          ImageSpan       recentImage = new ImageSpan(getContext(), R.drawable.ic_emoji_recent_light,
                                                      ImageSpan.ALIGN_BASELINE);

          recent.setSpan(recentImage, 1, recent.length()-1, Spanned.SPAN_EXCLUSIVE_EXCLUSIVE);

          return recent;
        case 1:
          SpannableString emoji      = new SpannableString(" Emoji ");
          ImageSpan       emojiImage = new ImageSpan(getContext(), R.drawable.ic_emoji_light,
                                                     ImageSpan.ALIGN_BASELINE);

          emoji.setSpan(emojiImage, 1, emoji.length()-1, Spanned.SPAN_EXCLUSIVE_EXCLUSIVE);

          return emoji;
        default:
          throw new AssertionError("Bad position!");
      }
    }

    @Override
    public boolean isViewFromObject(View view, Object o) {
      return view == o;
    }

    public Object instantiateItem(ViewGroup container, int position) {
      View view;

      switch (position) {
        case 0:  view = recentEmojiGridLayout; break;
        case 1:  view = emojiGridLayout;       break;
        default: throw new AssertionError("Too many positions!");
      }

      container.addView(view, 0);

      return view;
    }
  }
}