WaveshareUARTFingerprintSensor/WaveshareUARTFingerprintSensor.Sample/Views/TUIManager.cs

using NStack;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using Terminal.Gui;

namespace WaveshareUARTFingerprintSensor.Sample.Views
{
    public class TUIManager : Toplevel
    {
        public const string OutputFilePath = "out.txt";
        public const string SettingsFilePath = "settings.txt";

        private FingerprintSensor _fingerprintSensor;
        private Label _serialPortLabel;
        private Label _comparisonLevelLabel;
        private Label _userCountLabel;

        private object _outputFileLock = new object();

        public TUIManager()
        {
            Init();
        }

        private void Init()
        {
            ColorScheme = Colors.Error;

            // Creates the top-level window to show
            var win = new Window("TUIManager")
            {
                X = 0,
                Y = 1, // Leave one row for the toplevel menu

                // By using Dim.Fill(), it will automatically resize without manual intervention
                Width = Dim.Fill(),
                Height = Dim.Fill()
            };

            win.ColorScheme = Colors.ColorSchemes["Dialog"];

            Add(win);

            // Creates a menubar, the item "New" has a help menu.
            var menu = new MenuBar(new MenuBarItem[] {
                new MenuBarItem ("_Options", new MenuItem [] {
                    new MenuItem ("_Change Config", "", () => { ChangeConfig(); }),
                    new MenuItem ("_Quit", "", () => { Application.RequestStop(); })
                })
            });

            Add(menu);

            // Window Content
            _serialPortLabel = new Label("Serial Port:")
            {
                X = 2,
                Y = 1,
                Width = Dim.Fill()
            };
            _comparisonLevelLabel = new Label("Comparison Level: 0")
            {
                X = Pos.Left(_serialPortLabel),
                Y = Pos.Bottom(_serialPortLabel),
                Width = Dim.Fill()
            };
            _userCountLabel = new Label("Users: 0")
            {
                X = Pos.Left(_comparisonLevelLabel),
                Y = Pos.Bottom(_comparisonLevelLabel),
                Width = Dim.Fill()
            };

            var userCountButton = new Button("Query _User Count")
            {
                X = Pos.Center(),
                Y = 7
            };
            userCountButton.Clicked += UserCountButton_Clicked;

            var readFingerprintButton = new Button("_Read Fingerprint")
            {
                X = Pos.Center(),
                Y = Pos.Bottom(userCountButton) + 1
            };
            readFingerprintButton.Clicked += ReadFingerprintButton_Clicked;

            var readEigenvaluesButton = new Button("Read _Eigenvalues")
            {
                X = Pos.Center(),
                Y = Pos.Bottom(readFingerprintButton)
            };
            readEigenvaluesButton.Clicked += ReadEigenvaluesButton_Clicked;

            var readImageButton = new Button("Read _Image")
            {
                X = Pos.Center(),
                Y = Pos.Bottom(readEigenvaluesButton)
            };
            readImageButton.Clicked += ReadImageButton_Clicked;

            var addFingerprintButton = new Button("_Add Fingerprint")
            {
                X = Pos.Center(),
                Y = Pos.Bottom(readImageButton) + 1
            };
            addFingerprintButton.Clicked += AddFingerprintButton_Clicked;

            var deleteAFingerprint = new Button("_Delete a Fingerprint")
            {
                X = Pos.Center(),
                Y = Pos.Bottom(addFingerprintButton)
            };
            deleteAFingerprint.Clicked += DeleteAFingerprint_Clicked;

            var clearFingerprintsButton = new Button("_Clear Fingerprints")
            {
                X = Pos.Center(),
                Y = Pos.Bottom(deleteAFingerprint)
            };
            clearFingerprintsButton.Clicked += ClearFingerprintsButton_Clicked;

            var setComparisonLevelButton = new Button("Set Comparison _Level")
            {
                X = Pos.Center(),
                Y = Pos.Bottom(clearFingerprintsButton) + 1
            };
            setComparisonLevelButton.Clicked += SetComparisonLevelButton_Clicked;

            var sleepButton = new Button("_Sleep Mode")
            {
                X = Pos.Center(),
                Y = Pos.Bottom(setComparisonLevelButton) + 1
            };
            sleepButton.Clicked += SleepButton_Clicked;

            var quitButton = new Button("_Quit")
            {
                X = Pos.Center(),
                Y = Pos.Percent(95)
            };
            quitButton.Clicked += Quit_Clicked;

            win.Add(
                _serialPortLabel,
                _comparisonLevelLabel,
                _userCountLabel,
                userCountButton,
                readFingerprintButton,
                readEigenvaluesButton,
                readImageButton,
                addFingerprintButton,
                deleteAFingerprint,
                clearFingerprintsButton,
                setComparisonLevelButton,
                sleepButton,
                quitButton
            );

            // Init Sensor
            if (!File.Exists(SettingsFilePath))
            {
                Application.Run(new SettingsDisplay());
            }

            InitSensor();

            // Update gui
            UpdateSerialPort();
            UpdateUserCount();
            UpdateComparisonLevel();
        }

        private void UpdateSerialPort()
        {
            _serialPortLabel.Text = $"Serial Port: {_fingerprintSensor.PortName}";
        }

        private void InitSensor()
        {
            _fingerprintSensor = new FingerprintSensor(File.ReadAllText(SettingsFilePath));

            _fingerprintSensor.Start();
        }

        private void UpdateUserCount()
        {
            if (_fingerprintSensor.TryGetUserCount(out ushort count))
            {
                _userCountLabel.Text = $"Users: {count}";
            }
        }

        private void UpdateComparisonLevel()
        {
            if (_fingerprintSensor.TryGetComparisonLevel(out byte comparisonLevel))
            {
                _comparisonLevelLabel.Text = $"Comparison Level: {comparisonLevel}";
            }
        }

        private void ReadImageButton_Clicked()
        {
            var dialog = new FingerprintDialog("Acquire Image", "Can't acquire image, try to place your finger flat on the sensor");
            byte[] image = null;

            Task.Run(() =>
            {
                if (_fingerprintSensor.TryAcquireImage(out image))
                {
                    dialog.Cancel();
                }
                else
                {
                    dialog.CancelAndShowError();
                }
            });

            dialog.Show();

            if (image != null)
            {
                WriteOut($"Image:\n{Utils.ArrayDisplay(image)}\n\n\n");

                var window = new DataDisplay("Image", image.ToArray());

                Application.Run(window);
            }
        }

        private void ReadEigenvaluesButton_Clicked()
        {
            var dialog = new FingerprintDialog("Acquire Eigenvalues", "Can't acquire eigenvalues, try to place your finger flat on the sensor");
            byte[] eigenvalues = null;

            Task.Run(() =>
            {
                if (_fingerprintSensor.TryAcquireEigenvalues(out var values))
                {
                    eigenvalues = values.ToArray();

                    dialog.Cancel();
                }
                else
                {
                    dialog.CancelAndShowError();
                }
            });

            dialog.Show();

            if (eigenvalues != null)
            {
                var window = new DataDisplay("Eigenvalues", eigenvalues);

                 WriteOut($"Eigenvalues:\n{Utils.ArrayDisplay(eigenvalues)}\n\n\n");

                Application.Run(window);
            }
        }

        private void SetComparisonLevelButton_Clicked()
        {
            if (new EntryDialog("Comparison Level", i => int.TryParse(i.ToString(), out var n) && n > 0 && n < 10, "Need to be a valid number in 0-9 range").TryShow(out var input))
            {
                if (!_fingerprintSensor.TrySetComparisonLevel(byte.Parse(input.ToString())))
                {
                    MessageBox.ErrorQuery("Comparison Level", "Can't set comparison level", "Ok");
                }
            }

            UpdateComparisonLevel();
        }

        private void DeleteAFingerprint_Clicked()
        {
            if (new EntryDialog("User id", i => int.TryParse(i.ToString(), out var n), "Need to be a valid user id").TryShow(out var input))
            {
                if (!_fingerprintSensor.DeleteUser(ushort.Parse(input.ToString())))
                {
                    MessageBox.ErrorQuery("Delete User", "Can't delete user, check user id", "Ok");
                }
            }

            UpdateUserCount();
        }

        private void SleepButton_Clicked()
        {
            var window = new SleepDisplay(_fingerprintSensor);

            Application.Run(window);
        }

        private void ClearFingerprintsButton_Clicked()
        {
            if (!_fingerprintSensor.DeleteAllUsers())
            {
                MessageBox.ErrorQuery("Delete All Users", "Can't delete all user", "Ok");
            }
        }

        private void AddFingerprintButton_Clicked()
        {
            if (new EntryDialog("User id", i => ushort.TryParse(i.ToString(), out var n), "Need to be a valid user id").TryShow(out var input))
            {
                var dialog = new FingerprintDialog("Add Fingerprint", "Can't add fingerprint, try to place your finger flat on the sensor");

                Task.Run(() =>
                {
                    switch (_fingerprintSensor.AddFingerprint(ushort.Parse(input.ToString()), UserPermission.Level1))
                    {
                        case ResponseType.Success:
                            dialog.Cancel();

                            Application.MainLoop.Invoke(() => MessageBox.Query("Add Fingerprint", "Successfully added fingerprint", "Ok"));

                            break;
                        case ResponseType.Full:
                            dialog.ErrorMessage = "Sensor full, can't add more users";

                            dialog.CancelAndShowError();

                            break;
                        case ResponseType.NoUser:
                            dialog.ErrorMessage = "Can't add fingerprint, invalid id";

                            dialog.CancelAndShowError();

                            break;
                        case ResponseType.FingerOccupied:
                            dialog.ErrorMessage = "Can't add fingerprint, finger already registered";

                            dialog.CancelAndShowError();

                            break;
                        case ResponseType.UserOccupied:
                            dialog.ErrorMessage = "Can't add fingerprint, id already used";

                            dialog.CancelAndShowError();

                            break;
                        default:
                            dialog.CancelAndShowError();
                            break;
                    }
                });

                dialog.Show();

                UpdateUserCount();
            }
        }

        private void ReadFingerprintButton_Clicked()
        {
            var dialog = new FingerprintDialog("Read Fingerprint", "Can't read fingerprint, try to place your finger flat on the sensor");

            (ushort userID, UserPermission permission) userInfo = default;

            Task.Run(() =>
            {
                if (_fingerprintSensor.TryComparison1N(out userInfo))
                {
                    dialog.Cancel();
                }
                else
                {
                    dialog.CancelAndShowError();
                }
            });

            dialog.Show();

            if (userInfo != default)
            {
                MessageBox.Query("Read Fingerprint", $"Successfully read fingerprint\n\nUser ID: {userInfo.userID}\nPermissions: {userInfo.permission}\n ", "Ok");
            }
        }

        private void UserCountButton_Clicked()
        {
            UpdateUserCount();
        }
        private void WriteOut(string text)
        {
            Task.Run(() =>
            {
                lock (_outputFileLock)
                {
                    File.AppendAllText(OutputFilePath, text);
                }
            });
        }

        private void ChangeConfig()
        {
            File.Delete(SettingsFilePath);

            Application.Run(new SettingsDisplay());

            InitSensor();

            UpdateSerialPort();
        }

        private void Quit_Clicked()
        {
            Application.RequestStop();
        }
    }
}