app/src/main/java/com/wireguard/android/widget/fab/LabeledFloatingActionButton.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58

/*
 * Copyright © 2014 Jerzy Chalupski
 * Copyright © 2018 Jason A. Donenfeld <Jason@zx2c4.com>. All Rights Reserved.
 * SPDX-License-Identifier: Apache-2.0
 */

package com.wireguard.android.widget.fab;

import android.content.Context;
import android.content.res.TypedArray;
import android.support.annotation.Nullable;
import android.support.design.widget.FloatingActionButton;
import android.util.AttributeSet;
import android.widget.TextView;

import com.wireguard.android.R;

public class LabeledFloatingActionButton extends FloatingActionButton {

    @Nullable private final String title;

    public LabeledFloatingActionButton(final Context context) {
        this(context, null);
    }

    public LabeledFloatingActionButton(final Context context, @Nullable final AttributeSet attrs) {
        this(context, attrs, 0);
    }

    public LabeledFloatingActionButton(final Context context, @Nullable final AttributeSet attrs, final int defStyle) {
        super(context, attrs, defStyle);

        final TypedArray attr = context.obtainStyledAttributes(attrs, R.styleable.LabeledFloatingActionButton, 0, 0);
        title = attr.getString(R.styleable.LabeledFloatingActionButton_fab_title);
        attr.recycle();
    }

    @Nullable
    TextView getLabelView() {
        return (TextView) getTag(R.id.fab_label);
    }

    @Nullable
    public String getTitle() {
        return title;
    }

    @Override
    public void setVisibility(final int visibility) {
        final TextView label = getLabelView();
        if (label != null) {
            label.setVisibility(visibility);
        }

        super.setVisibility(visibility);
    }

}