1 files changed, 13 insertions, 0 deletions
diff --git a/app/src/main/java/se/leap/bitmaskclient/base/views/MainButton.java b/app/src/main/java/se/leap/bitmaskclient/base/views/MainButton.java
index c7273613..e322e6c6 100644
--- a/app/src/main/java/se/leap/bitmaskclient/base/views/MainButton.java
+++ b/app/src/main/java/se/leap/bitmaskclient/base/views/MainButton.java
@@ -2,10 +2,13 @@ package se.leap.bitmaskclient.base.views;
 
 import android.annotation.TargetApi;
 import android.content.Context;
+import android.graphics.PorterDuff;
+import android.graphics.drawable.Drawable;
 import android.util.AttributeSet;
 import android.view.LayoutInflater;
 import android.widget.RelativeLayout;
 
+import androidx.annotation.DrawableRes;
 import androidx.appcompat.widget.AppCompatImageView;
 import androidx.core.content.ContextCompat;
 
@@ -56,4 +59,14 @@ public class MainButton extends RelativeLayout {
             button.setTag(isOn ? "button_circle_stop" : "button_circle_start");
         }
     }
+
+    public void setCustomDrawable(@DrawableRes int drawableResource) {
+        Drawable drawable = ContextCompat.getDrawable(getContext(), drawableResource);
+        if (drawable == null) {
+            return;
+        }
+
+        button.setImageDrawable(drawable);
+        button.setTag("button_setup_circle_custom");
+    }
 }