package com.mobify.astro.plugins.counterbadgeplugin;

import android.content.Context;
import android.graphics.drawable.GradientDrawable;
import android.util.TypedValue;
import android.widget.TextView;

import com.mobify.astro.R;

public class BadgeView extends TextView {
    protected GradientDrawable badgeImage;

    public BadgeView(Context context) {
        super(context);

        badgeImage =  new GradientDrawable();
        badgeImage.setShape(GradientDrawable.RECTANGLE);

        int badgeRadius = context.getResources().getDimensionPixelSize(R.dimen.counter_badge_radius);
        badgeImage.setCornerRadius(badgeRadius);

        this.setBackground(badgeImage);

        float textSize = context.getResources().getDimension(R.dimen.counter_badge_text_size);
        this.setTextSize(TypedValue.COMPLEX_UNIT_PX, textSize);
    }

    public void setBackgroundColor(int color) {
        badgeImage.mutate();
        badgeImage.setColor(color);
    }

    public void setCount(String count) {
        this.setText(count);
    }
}