From 3d83b07ca7090817192bcd8f250effc0d0817b2a Mon Sep 17 00:00:00 2001
From: Robin Krahl <robin.krahl@ireas.org>
Date: Fri, 25 Sep 2020 02:09:18 +0200
Subject: Support Librem Key

This patch adds the librem device model for the Librem Key.
---
 src/args.rs      | 4 ++++
 src/tests/mod.rs | 1 +
 2 files changed, 5 insertions(+)

(limited to 'src')

diff --git a/src/args.rs b/src/args.rs
index e4bc77d..3bb1225 100644
--- a/src/args.rs
+++ b/src/args.rs
@@ -38,6 +38,7 @@ pub struct Args {
 Enum! {
   /// The available Nitrokey models.
   DeviceModel, [
+    Librem => "librem",
     Pro => "pro",
     Storage => "storage",
   ]
@@ -46,6 +47,7 @@ Enum! {
 impl DeviceModel {
   pub fn as_user_facing_str(&self) -> &str {
     match self {
+      DeviceModel::Librem => "Librem",
       DeviceModel::Pro => "Pro",
       DeviceModel::Storage => "Storage",
     }
@@ -55,6 +57,7 @@ impl DeviceModel {
 impl From<DeviceModel> for nitrokey::Model {
   fn from(model: DeviceModel) -> nitrokey::Model {
     match model {
+      DeviceModel::Librem => nitrokey::Model::Librem,
       DeviceModel::Pro => nitrokey::Model::Pro,
       DeviceModel::Storage => nitrokey::Model::Storage,
     }
@@ -66,6 +69,7 @@ impl convert::TryFrom<nitrokey::Model> for DeviceModel {
 
   fn try_from(model: nitrokey::Model) -> Result<DeviceModel, anyhow::Error> {
     match model {
+      nitrokey::Model::Librem => Ok(DeviceModel::Librem),
       nitrokey::Model::Pro => Ok(DeviceModel::Pro),
       nitrokey::Model::Storage => Ok(DeviceModel::Storage),
       _ => Err(anyhow::anyhow!("Unsupported device model: {}", model)),
diff --git a/src/tests/mod.rs b/src/tests/mod.rs
index 23eecc5..65983bb 100644
--- a/src/tests/mod.rs
+++ b/src/tests/mod.rs
@@ -76,6 +76,7 @@ impl Nitrocli {
 
   fn model_to_arg(model: nitrokey::Model) -> &'static str {
     match model {
+      nitrokey::Model::Librem => "--model=librem",
       nitrokey::Model::Pro => "--model=pro",
       nitrokey::Model::Storage => "--model=storage",
       _ => panic!("Unexpected model in test suite: {}", model),
-- 
cgit v1.2.3