add to cart

app/src/main/java/ch/pizzalink/android/activity/MainActivity.java

@@ -8,15 +8,18 @@ import android.support.v4.widget.DrawerLayout;
 import android.os.Bundle;
 import android.support.v7.widget.LinearLayoutManager;
 import android.support.v7.widget.RecyclerView;
+import android.view.Gravity;
 import android.view.MenuItem;
 import android.view.View;
 import android.widget.TextView;
+import android.widget.Toast;
 
 import com.ittianyu.bottomnavigationviewex.BottomNavigationViewEx;
 
 import java.util.ArrayList;
 import java.util.List;
 
+import butterknife.BindColor;
 import butterknife.BindView;
 import butterknife.ButterKnife;
 import ch.pizzalink.android.R;
@@ -33,6 +36,8 @@ import ch.pizzalink.android.view.PizzalinkToolbar;
 import io.github.luizgrp.sectionedrecyclerviewadapter.SectionParameters;
 import io.github.luizgrp.sectionedrecyclerviewadapter.SectionedRecyclerViewAdapter;
 import io.github.luizgrp.sectionedrecyclerviewadapter.StatelessSection;
+import q.rorbin.badgeview.Badge;
+import q.rorbin.badgeview.QBadgeView;
 
 public class MainActivity extends BaseActivity {
 
@@ -41,11 +46,15 @@ public class MainActivity extends BaseActivity {
     @BindView(R.id.pizzalinkToolbar) PizzalinkToolbar pizzalinkToolbar;
     @BindView(R.id.bottomNavigationView) BottomNavigationViewEx bottomNavigationView;
 
+    @BindColor(R.color.red) int redColor;
+    @BindColor(R.color.white) int whiteColor;
+
     private FragmentManager fragmentManager;
     private String currentFragmentName = "";
     private int currentCategoryId = -1;
     private ArrayList<CategoryModel> categoryList = new ArrayList<>();
     private NavigationMenuRecyclerAdapter navigationMenuRecyclerAdapter;
+    private Badge badge;
 
     private SectionedRecyclerViewAdapter sectionAdapter = new SectionedRecyclerViewAdapter();
 
@@ -61,6 +70,7 @@ public class MainActivity extends BaseActivity {
         initNavigationDrawer();
         initBottomNavigationView();
         showStartScreen();
+        setCartItemCount();
     }
 
     private void showStartScreen(){
@@ -274,4 +284,21 @@ public class MainActivity extends BaseActivity {
         }
 
     }
+
+    public void setCartItemCount(){
+        addBadgeAt(1, SharedPrefsHelper.getCartItemCount());
+    }
+
+    private void addBadgeAt(int position, int number) {
+        // add badge
+        badge = new QBadgeView(this)
+                .setBadgeNumber(number)
+                .setBadgeBackgroundColor(redColor)
+                .setBadgeTextColor(whiteColor)
+                .setBadgeGravity(Gravity.TOP | Gravity.END)
+                .bindTarget(bottomNavigationView.getBottomNavigationItemView(position));
+
+        if(number == 0)
+            badge.hide(true);
+    }
 }