Aegis/app/src/main/java/me/impy/aegis/KeyProfile.java

package me.impy.aegis;

import com.amulyakhare.textdrawable.TextDrawable;
import com.amulyakhare.textdrawable.util.ColorGenerator;

import java.io.Serializable;
import java.lang.reflect.UndeclaredThrowableException;

import me.impy.aegis.crypto.otp.OTP;
import me.impy.aegis.db.DatabaseEntry;

public class KeyProfile implements Serializable {
    private String _code;
    private DatabaseEntry _entry;

    public KeyProfile(DatabaseEntry entry) {
        _entry = entry;
    }

    public DatabaseEntry getEntry() {
        return _entry;
    }
    public String getCode() {
        return _code;
    }

    public String refreshCode() {
        try {
            _code = OTP.generateOTP(_entry.getInfo());
        } catch (Exception e) {
            throw new UndeclaredThrowableException(e);
        }
        return _code;
    }

    public TextDrawable getDrawable() {
        String name = _entry.getName();
        if (name == null) {
            return null;
        }

        ColorGenerator generator = ColorGenerator.MATERIAL;
        int color = generator.getColor(name);

        return TextDrawable.builder().buildRound(name.substring(0, 1).toUpperCase(), color);
    }
}