From c9c4ab787fba70389679676a35ff738d1806c9be Mon Sep 17 00:00:00 2001
From: Marten Rebane <marten.rebane@nortal.com>
Date: Mon, 30 Mar 2026 17:54:00 +0300
Subject: [PATCH] Use correct icon in crypto view

---
 .../UI/Component/Container/Crypto/EncryptView.swift    | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/RIADigiDoc/UI/Component/Container/Crypto/EncryptView.swift b/RIADigiDoc/UI/Component/Container/Crypto/EncryptView.swift
index 84f70b8f..dc76d1ac 100644
--- a/RIADigiDoc/UI/Component/Container/Crypto/EncryptView.swift
+++ b/RIADigiDoc/UI/Component/Container/Crypto/EncryptView.swift
@@ -159,6 +159,14 @@ struct EncryptView: View {
     private var containerExtension: String {
         URL(fileURLWithPath: viewModel.containerName).pathExtension
     }
+    
+    private var containerIcon: String {
+        viewModel.isContainerDecrypted
+            ? "ic_m3_encrypted_off_48pt_wght400"
+            : (viewModel.isContainerEncrypted
+                ? "ic_m3_encrypted_48pt_wght400"
+                : "ic_m3_folder_48pt_wght400")
+    }
 
     @State private var containerLoadingTask: Task<Void, Never>?
 
@@ -196,7 +204,7 @@ struct EncryptView: View {
                         ScrollView {
                             VStack {
                                 ContainerNameView(
-                                    icon: "ic_m3_stylus_note_48pt_wght400",
+                                    icon: containerIcon,
                                     containerNameTitle: containerNameTitle,
                                     name: $viewModel.containerName,
                                     isEditContainerButtonShown: viewModel.isEditButtonShown,