From 1eca7f3ff9378379bd8c41548c0ab03e6c8c855f Mon Sep 17 00:00:00 2001
From: Bryce <bryce@integreatconsult.com>
Date: Tue, 5 Aug 2025 07:32:56 -0700
Subject: [PATCH] support toggle.

---
 app/models.py                                 |   1 +
 app/routes.py                                 |  24 +++++++
 app/templates/partials/folder_card.html       |  50 ++++++++++++++
 app/templates/partials/folders_list.html      |  43 +-----------
 .../versions/f8ba65458ba2_adding_toggle.py    |  32 +++++++++
 tests/test_routes.py                          |  63 +++++++++++++++++-
 tmp/.hold                                     |   0
 .../example.com/user1/dovecot-uidlist         |   1 +
 .../example.com/user1/dovecot-uidvalidity     |   1 +
 .../user1/dovecot-uidvalidity.6891fad8        |   0
 .../example.com/user1/dovecot.index.log       | Bin 0 -> 232 bytes
 .../example.com/user1/dovecot.list.index.log  | Bin 0 -> 512 bytes
 .../example.com/user1/dovecot.mailbox.log     | Bin 0 -> 24 bytes
 tmp/imap-data/example.com/user1/maildirfolder |   0
 tmp/imap-data/example.com/user1/subscriptions |   3 +
 15 files changed, 177 insertions(+), 41 deletions(-)
 create mode 100644 app/templates/partials/folder_card.html
 create mode 100644 migrations/versions/f8ba65458ba2_adding_toggle.py
 create mode 100644 tmp/.hold
 create mode 100644 tmp/imap-data/example.com/user1/dovecot-uidlist
 create mode 100644 tmp/imap-data/example.com/user1/dovecot-uidvalidity
 create mode 100644 tmp/imap-data/example.com/user1/dovecot-uidvalidity.6891fad8
 create mode 100644 tmp/imap-data/example.com/user1/dovecot.index.log
 create mode 100644 tmp/imap-data/example.com/user1/dovecot.list.index.log
 create mode 100644 tmp/imap-data/example.com/user1/dovecot.mailbox.log
 create mode 100644 tmp/imap-data/example.com/user1/maildirfolder
 create mode 100644 tmp/imap-data/example.com/user1/subscriptions

diff --git a/app/models.py b/app/models.py
index 8b88d17..4156eab 100644
--- a/app/models.py
+++ b/app/models.py
@@ -39,5 +39,6 @@ class Folder(Base):
     name = db.Column(db.String(255), nullable=False)
     rule_text = db.Column(db.Text)
     priority = db.Column(db.Integer)
+    organize_enabled = db.Column(db.Boolean, default=True)
     
     user = db.relationship('User', backref=db.backref('folders', lazy=True))
\ No newline at end of file
diff --git a/app/routes.py b/app/routes.py
index 731d2d6..a3a591f 100644
--- a/app/routes.py
+++ b/app/routes.py
@@ -116,6 +116,30 @@ def delete_folder(folder_id):
         folders = Folder.query.filter_by(user_id=current_user.id).all()
         return render_template('partials/folders_list.html', folders=folders)
 
+@main.route('/api/folders/<folder_id>/toggle', methods=['PUT'])
+@login_required
+def toggle_folder_organize(folder_id):
+    try:
+        # Find the folder by ID and ensure it belongs to the current user
+        folder = Folder.query.filter_by(id=folder_id, user_id=current_user.id).first()
+        
+        if not folder:
+            return jsonify({'error': 'Folder not found'}), 404
+            
+        # Toggle the organize_enabled flag
+        folder.organize_enabled = not folder.organize_enabled
+        
+        db.session.commit()
+        
+        # Return just the updated folder card HTML for this specific folder
+        return render_template('partials/folder_card.html', folder=folder)
+        
+    except Exception as e:
+        # Print unhandled exceptions to the console as required
+        logging.exception("Error toggling folder organize flag: %s", e)
+        db.session.rollback()
+        return jsonify({'error': 'An unexpected error occurred'}), 500
+
 @main.route('/api/folders/<folder_id>/edit', methods=['GET'])
 @login_required
 def edit_folder_modal(folder_id):
diff --git a/app/templates/partials/folder_card.html b/app/templates/partials/folder_card.html
new file mode 100644
index 0000000..9dad1da
--- /dev/null
+++ b/app/templates/partials/folder_card.html
@@ -0,0 +1,50 @@
+<div id="folder-{{ folder.id }}" class="card bg-base-100 shadow-xl border border-base-300 hover:shadow-lg transition-shadow duration-200">
+    <div class="card-body">
+        <div class="flex justify-between items-start mb-2">
+            <h3 class="text-xl font-bold truncate">{{ folder.name }}</h3>
+            <div class="flex space-x-2">
+            <button class="btn btn-sm btn-outline" 
+                    hx-get="/api/folders/{{ folder.id }}/edit"
+                    hx-target="#modal-holder"
+                    hx-swap="innerHTML"
+                    hx-trigger="click">
+                <i class="fas fa-edit"></i>
+            </button>
+            <button class="btn btn-sm btn-outline btn-error" 
+                    hx-delete="/api/folders/{{ folder.id }}" 
+                    hx-target="#folders-list" 
+                    hx-swap="innerHTML"
+                    hx-confirm="Are you sure you want to delete this folder?">
+                <i class="fas fa-trash"></i>
+            </button>
+        </div>
+        </div>
+
+        <div class="bg-base-200 rounded-box p-4 mb-4">
+            <p class="text-base-content/80">{{ folder.rule_text }}</p>
+        </div>
+        
+        <div class="flex justify-between items-center mt-2">
+            {% if folder.priority == 1 %}
+                <span class="badge badge-error">High Priority</span>
+            {% elif folder.priority == -1 %}
+                <span class="badge badge-info">Low Priority</span>
+            {% else %}
+                <span class="badge badge-primary">Normal Priority</span>
+            {% endif %}
+            <div class="flex items-center space-x-2">
+                <span class="text-xs">Organize:</span>
+                <input 
+                    type="checkbox"
+                    class="toggle toggle-sm toggle-success"
+                    {% if folder.organize_enabled %}checked="checked"{% endif %}
+                    hx-put="/api/folders/{{ folder.id }}/toggle"
+                    hx-target="#folder-{{ folder.id }}"
+                    hx-swap="outerHTML"
+                    hx-trigger="click"
+                    aria-label="Toggle organize enabled">
+            </input>
+            </div>
+        </div>
+    </div>
+</div>
\ No newline at end of file
diff --git a/app/templates/partials/folders_list.html b/app/templates/partials/folders_list.html
index c416aa8..9fec5a4 100644
--- a/app/templates/partials/folders_list.html
+++ b/app/templates/partials/folders_list.html
@@ -1,43 +1,6 @@
-<div class="grid grid-cols-1 md:grid-cols-2 lg:grid-cols-3 gap-6">
+<div id="folders-list" class="grid grid-cols-1 md:grid-cols-2 lg:grid-cols-3 gap-6">
     {% for folder in folders %}
-    <div class="card bg-base-100 shadow-xl border border-base-300 hover:shadow-lg transition-shadow duration-200">
-        <div class="card-body">
-            <div class="flex justify-between items-start mb-2">
-                <h3 class="text-xl font-bold truncate">{{ folder.name }}</h3>
-                <div class="flex space-x-2">
-                <button class="btn btn-sm btn-outline" 
-                        hx-get="/api/folders/{{ folder.id }}/edit"
-                        hx-target="#modal-holder"
-                        hx-swap="innerHTML"
-                        hx-trigger="click">
-                    <i class="fas fa-edit"></i>
-                </button>
-                <button class="btn btn-sm btn-outline btn-error" 
-                        hx-delete="/api/folders/{{ folder.id }}" 
-                        hx-target="#folders-list" 
-                        hx-swap="innerHTML"
-                        hx-confirm="Are you sure you want to delete this folder?">
-                    <i class="fas fa-trash"></i>
-                </button>
-            </div>
-            </div>
-
-            <div class="bg-base-200 rounded-box p-4 mb-4">
-                <p class="text-base-content/80">{{ folder.rule_text }}</p>
-            </div>
-            
-            <div class="flex justify-between items-center mt-2">
-                {% if folder.priority == 1 %}
-                    <span class="badge badge-error">High Priority</span>
-                {% elif folder.priority == -1 %}
-                    <span class="badge badge-info">Low Priority</span>
-                {% else %}
-                    <span class="badge badge-primary">Normal Priority</span>
-                {% endif %}
-                <span class="text-xs badge badge-secondary">0 emails</span>
-            </div>
-        </div>
-    </div>
+        {% include 'partials/folder_card.html' %}
     {% else %}
     <div class="col-span-full text-center py-12 bg-base-100 rounded-box shadow-lg border border-dashed border-base-300">
         <div class="text-5xl mb-4 text-primary">
@@ -45,7 +8,7 @@
         </div>
         <h3 class="text-2xl font-bold mb-2">No folders yet</h3>
         <p class="mb-6 text-base-content/70">Add your first folder to get started organizing your emails.</p>
-        <button class="btn btn-primary btn-lg" 
+        <button class="btn btn-primary btn-lg"
                 hx-get="/api/folders/new"
                 hx-target="#modal-holder"
                 hx-swap="innerHTML"
diff --git a/migrations/versions/f8ba65458ba2_adding_toggle.py b/migrations/versions/f8ba65458ba2_adding_toggle.py
new file mode 100644
index 0000000..5eb5ff4
--- /dev/null
+++ b/migrations/versions/f8ba65458ba2_adding_toggle.py
@@ -0,0 +1,32 @@
+"""adding toggle
+
+Revision ID: f8ba65458ba2
+Revises: 0bcf54a7f2a7
+Create Date: 2025-08-05 06:58:59.265593
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = 'f8ba65458ba2'
+down_revision = '0bcf54a7f2a7'
+branch_labels = None
+depends_on = None
+
+
+def upgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table('folders', schema=None) as batch_op:
+        batch_op.add_column(sa.Column('organize_enabled', sa.Boolean(), nullable=True))
+
+    # ### end Alembic commands ###
+
+
+def downgrade():
+    # ### commands auto generated by Alembic - please adjust! ###
+    with op.batch_alter_table('folders', schema=None) as batch_op:
+        batch_op.drop_column('organize_enabled')
+
+    # ### end Alembic commands ###
diff --git a/tests/test_routes.py b/tests/test_routes.py
index 29302f2..c43c80f 100644
--- a/tests/test_routes.py
+++ b/tests/test_routes.py
@@ -1,5 +1,6 @@
 import pytest
 from app.models import User, Folder
+from app import db
 import uuid
 from bs4 import BeautifulSoup
 
@@ -308,4 +309,64 @@ def test_edit_folder_content_whitespace_handling(authenticated_client, mock_fold
     assert updated_folder is not None
     assert updated_folder.name == 'Updated Folder With Whitespace'  # Should be trimmed
     assert updated_folder.rule_text == 'Updated rule with whitespace around it'  # Should be trimmed
-    assert updated_folder.priority == int(test_priority)
\ No newline at end of file
+    assert updated_folder.priority == int(test_priority)
+    
+def test_toggle_folder_organize_enabled(authenticated_client, mock_folder):
+    """Test toggling the organize_enabled flag for a folder."""
+    # Verify initial state is True (default)
+    assert mock_folder.organize_enabled is True
+    
+    # Toggle the flag
+    response = authenticated_client.put(f'/api/folders/{mock_folder.id}/toggle')
+    
+    # Should return 200 OK
+    assert response.status_code == 200
+    
+    # Verify the folder was updated in the database
+    updated_folder = Folder.query.filter_by(id=mock_folder.id).first()
+    assert updated_folder is not None
+    assert updated_folder.organize_enabled is False
+    
+    # Toggle again to make sure it works both ways
+    response2 = authenticated_client.put(f'/api/folders/{mock_folder.id}/toggle')
+    
+    # Should return 200 OK
+    assert response2.status_code == 200
+    
+    # Verify the folder was updated in the database again
+    updated_folder2 = Folder.query.filter_by(id=mock_folder.id).first()
+    assert updated_folder2 is not None
+    assert updated_folder2.organize_enabled is True
+
+def test_toggle_folder_organize_enabled_not_found(authenticated_client, mock_user):
+    """Test toggling organize_enabled flag for a non-existent folder."""
+    # Try to toggle a folder that doesn't exist
+    response = authenticated_client.put('/api/folders/999/toggle')
+    
+    # Should return 404 Not Found
+    assert response.status_code == 404
+    
+def test_toggle_folder_organize_enabled_unauthorized(authenticated_client, mock_user, app):
+    """Test toggling organize_enabled flag for a folder that doesn't belong to the user."""
+    # Create a folder that belongs to a different user
+    other_user = User(email='other@example.com', first_name='Other', last_name='User')
+    other_user.set_password('password')
+    
+    with app.app_context():
+        db.session.add(other_user)
+        db.session.commit()
+        
+        # Create a folder for the other user
+        other_folder = Folder(
+            user_id=other_user.id,
+            name='Other User Folder',
+            rule_text='Test rule text'
+        )
+        db.session.add(other_folder)
+        db.session.commit()
+        
+        # Try to toggle the flag for the folder that doesn't belong to authenticated user
+        response = authenticated_client.put(f'/api/folders/{other_folder.id}/toggle')
+        
+        # Should return 404 Not Found (folder not found due to authorization check)
+        assert response.status_code == 404
\ No newline at end of file
diff --git a/tmp/.hold b/tmp/.hold
new file mode 100644
index 0000000..e69de29
diff --git a/tmp/imap-data/example.com/user1/dovecot-uidlist b/tmp/imap-data/example.com/user1/dovecot-uidlist
new file mode 100644
index 0000000..c1adf5b
--- /dev/null
+++ b/tmp/imap-data/example.com/user1/dovecot-uidlist
@@ -0,0 +1 @@
+3 V1754397400 N1 Gdd33f323d8fa916813000000739f6a6d
diff --git a/tmp/imap-data/example.com/user1/dovecot-uidvalidity b/tmp/imap-data/example.com/user1/dovecot-uidvalidity
new file mode 100644
index 0000000..db78e65
--- /dev/null
+++ b/tmp/imap-data/example.com/user1/dovecot-uidvalidity
@@ -0,0 +1 @@
+6891fad8
\ No newline at end of file
diff --git a/tmp/imap-data/example.com/user1/dovecot-uidvalidity.6891fad8 b/tmp/imap-data/example.com/user1/dovecot-uidvalidity.6891fad8
new file mode 100644
index 0000000..e69de29
diff --git a/tmp/imap-data/example.com/user1/dovecot.index.log b/tmp/imap-data/example.com/user1/dovecot.index.log
new file mode 100644
index 0000000000000000000000000000000000000000..5cdc6b91055b031e56a0313d075dd3070a97d5ff
GIT binary patch
literal 232
zcmZQ%)?m2tYhnfy0|J2fj8HC&ZfIy|W?<kDr~rxs*_{py3<Cdw0HjO>L;wLJ13N=*
zVrEWCW)VnD4^Th=BnvVRge_hfE2FbFvTCRp1N8#cw<!S4lwe>1tM>uYAoGFZV0%3v
rYPubOG|XNWAR7c28F(2oQi^m7@(YZ0OEXh)fMP&>EkFq%fT#ih8Sp<4

literal 0
HcmV?d00001

diff --git a/tmp/imap-data/example.com/user1/dovecot.list.index.log b/tmp/imap-data/example.com/user1/dovecot.list.index.log
new file mode 100644
index 0000000000000000000000000000000000000000..6b3eaec1e96aaf0fbeb1700a74099503791f5fa8
GIT binary patch
literal 512
zcmZQ%)?m2tYhnfy0|J2fj8HC&ZfIy|Q(#~akYHc|tM>uY|A7D~-U1SL0P+KXdVu_9
z1_llR8wk6@0VoGj2jnt<%#s1BV+7)y%;FM|ULZ#RBm<Jer<Vm}090>rX;LvlFT_1C
zy(b{%v^xOp1?d0*5Cdc{2msZ!L+s>-ILOn_$v*-t1(It4hJyf-pHa*K>VUc*Y)%&g
z6GSac{axeF%Al|l2C6HbpOp(T6=-IU0?^D51{H<|Peiza<gfvd8x(-*LE*UoXfTiu
T4o_&<Qsf?x+mOu%x}O06m|9<g

literal 0
HcmV?d00001

diff --git a/tmp/imap-data/example.com/user1/dovecot.mailbox.log b/tmp/imap-data/example.com/user1/dovecot.mailbox.log
new file mode 100644
index 0000000000000000000000000000000000000000..96bbc7c5eb8fe13679241c648acf90e01c24b54b
GIT binary patch
literal 24
gcmZQ!U|_hzX6<@XKjr&;Ubzc$xzUCh6Mx+R09nZiHUIzs

literal 0
HcmV?d00001

diff --git a/tmp/imap-data/example.com/user1/maildirfolder b/tmp/imap-data/example.com/user1/maildirfolder
new file mode 100644
index 0000000..e69de29
diff --git a/tmp/imap-data/example.com/user1/subscriptions b/tmp/imap-data/example.com/user1/subscriptions
new file mode 100644
index 0000000..6e1e1ce
--- /dev/null
+++ b/tmp/imap-data/example.com/user1/subscriptions
@@ -0,0 +1,3 @@
+V	2
+
+INBOX