Update main.py

main.py

@@ -1266,9 +1266,182 @@ def admin_get_all_deals():
         return handle_route_errors(e, uid_context=admin_uid)
 # --- END OF ADMIN ALL LISTINGS/DEALS ---
 
+# --- ADMIN ENDPOINTS FOR WHATSAPP-MANAGED FARMER ACCOUNTS ---
 
-# --- (The rest of your script: Admin approval of deals, Chat, Notifications, AI, main block) ---
-# Ensure all these sections are present and use the handle_route_errors and app=db_app pattern.
+@app.route('/api/admin/whatsapp-farmers/create', methods=['POST'])
+def admin_create_whatsapp_farmer():
+    auth_header = request.headers.get('Authorization')
+    admin_uid = None
+    try:
+        admin_uid = verify_admin(auth_header)
+        if not FIREBASE_INITIALIZED: 
+            return jsonify({'error': 'Server configuration error: Firebase not ready.'}), 503
+
+        data = request.get_json()
+        name = data.get('name')
+        phone_number = data.get('phone_number') # Expecting E.164 format ideally
+        location = data.get('location', "")
+        initial_notes = data.get('initial_notes', "")
+
+        if not name or not phone_number:
+            return jsonify({'error': 'Farmer name and phone number are required.'}), 400
+        
+        # Basic validation for phone number format (can be enhanced)
+        if not re.match(r"^\+?[1-9]\d{1,14}$", phone_number):
+            return jsonify({'error': 'Invalid phone number format. Please use E.164 like +263771234567.'}), 400
+
+        # Check if phone number is already in use by a fully registered webapp user
+        # This query can be slow if you have many users. Consider indexing phone_number.
+        users_ref = db.reference('users', app=db_app)
+        existing_user_query = users_ref.order_by_child('phone_number').equal_to(phone_number).limit_to_first(1).get()
+        if existing_user_query:
+            for existing_uid, existing_user_data in existing_user_query.items():
+                if existing_user_data.get('account_type') == 'webapp_registered':
+                    return jsonify({'error': f'This phone number is already registered to a web app user (UID: {existing_uid}).'}), 409
+                # Optionally, handle if already a whatsapp_managed account with this number
+                # if existing_user_data.get('account_type') == 'whatsapp_managed':
+                #    return jsonify({'error': f'This phone number is already managed for another WhatsApp farmer (UID: {existing_uid}).'}), 409
+
+
+        # Generate a unique ID for this WhatsApp-managed farmer
+        # This UID will be used in the 'users' node. No Firebase Auth user created at this stage.
+        farmer_uid = f"wpfarmer_{str(uuid.uuid4())}"
+
+        profile_data = {
+            'uid': farmer_uid, # Store the UID within the profile too for convenience
+            'name': name,
+            'phone_number': phone_number,
+            'email': f"{farmer_uid}@whatsapp.tunasonga.internal", # Placeholder email
+            'location': location,
+            'roles': {'farmer': True, 'buyer': False, 'transporter': False},
+            'account_type': "whatsapp_managed",
+            'is_placeholder_account': True, # Indicates this was admin-created before web registration
+            'can_login_webapp': False, # Cannot log in to web app yet
+            'managed_by_admin_uid': admin_uid,
+            'created_at': datetime.now(timezone.utc).isoformat(),
+            'created_by_admin_uid': admin_uid, # Explicitly track creator
+            'whatsapp_interaction_log': [{ # Start a log
+                'timestamp': datetime.now(timezone.utc).isoformat(),
+                'action': 'account_created_by_admin',
+                'admin_uid': admin_uid,
+                'notes': f"Initial notes: {initial_notes}" if initial_notes else "Account created."
+            }]
+        }
+
+        db.reference(f'users/{farmer_uid}', app=db_app).set(profile_data)
+
+        return jsonify({
+            'success': True, 
+            'message': f'WhatsApp-managed farmer account created for {name} ({phone_number}).',
+            'farmer_uid': farmer_uid,
+            'profile': profile_data
+        }), 201
+
+    except Exception as e:
+        return handle_route_errors(e, uid_context=admin_uid)
+
+
+@app.route('/api/admin/whatsapp-farmers', methods=['GET'])
+def admin_list_whatsapp_farmers():
+    auth_header = request.headers.get('Authorization')
+    admin_uid = None
+    try:
+        admin_uid = verify_admin(auth_header)
+        if not FIREBASE_INITIALIZED: 
+            return jsonify({'error': 'Server configuration error: Firebase not ready.'}), 503
+
+        users_ref = db.reference('users', app=db_app)
+        # Querying for account_type. This requires an index on 'account_type' in Firebase rules for performance.
+        # ".indexOn": ["account_type"] under your "users" rules.
+        whatsapp_farmers_query = users_ref.order_by_child('account_type').equal_to('whatsapp_managed').get()
+        
+        whatsapp_farmers = whatsapp_farmers_query or {}
+        
+        return jsonify(whatsapp_farmers), 200
+
+    except Exception as e:
+        return handle_route_errors(e, uid_context=admin_uid)
+
+@app.route('/api/admin/whatsapp-farmers/<farmer_uid>', methods=['GET'])
+def admin_get_whatsapp_farmer(farmer_uid):
+    auth_header = request.headers.get('Authorization')
+    admin_uid = None
+    try:
+        admin_uid = verify_admin(auth_header)
+        if not FIREBASE_INITIALIZED: 
+            return jsonify({'error': 'Server configuration error: Firebase not ready.'}), 503
+
+        farmer_ref = db.reference(f'users/{farmer_uid}', app=db_app)
+        farmer_data = farmer_ref.get()
+
+        if not farmer_data or farmer_data.get('account_type') != 'whatsapp_managed':
+            return jsonify({'error': 'WhatsApp-managed farmer not found or not of correct type.'}), 404
+        
+        return jsonify(farmer_data), 200
+
+    except Exception as e:
+        return handle_route_errors(e, uid_context=admin_uid)
+
+
+@app.route('/api/admin/whatsapp-farmers/<farmer_uid>/update', methods=['PUT'])
+def admin_update_whatsapp_farmer(farmer_uid):
+    auth_header = request.headers.get('Authorization')
+    admin_uid = None
+    try:
+        admin_uid = verify_admin(auth_header)
+        if not FIREBASE_INITIALIZED: 
+            return jsonify({'error': 'Server configuration error: Firebase not ready.'}), 503
+
+        farmer_ref = db.reference(f'users/{farmer_uid}', app=db_app)
+        farmer_data = farmer_ref.get()
+
+        if not farmer_data or farmer_data.get('account_type') != 'whatsapp_managed':
+            return jsonify({'error': 'WhatsApp-managed farmer not found or not of correct type.'}), 404
+
+        data = request.get_json()
+        updates = {}
+        allowed_fields_to_update = ['name', 'location', 'phone_number'] # Define what admin can change
+
+        for field in allowed_fields_to_update:
+            if field in data:
+                updates[field] = data[field]
+        
+        if not updates and 'additional_notes' not in data : # Check if any valid fields or notes are provided
+             return jsonify({'error': 'No valid fields to update or notes provided.'}), 400
+
+        # Log the update action
+        if data.get('additional_notes'):
+            log_entry_key = db.reference(f'users/{farmer_uid}/whatsapp_interaction_log', app=db_app).push().key
+            log_entry = {
+                'timestamp': datetime.now(timezone.utc).isoformat(),
+                'action': 'admin_update_notes',
+                'admin_uid': admin_uid,
+                'notes': data.get('additional_notes')
+            }
+            # To add to the log, we need to fetch existing logs or use a transaction if it's critical
+            # For simplicity here, we'll just push a new entry.
+            # If you want to update the main profile and add a log entry atomically,
+            # you'd construct a single update payload for farmer_ref.update()
+            db.reference(f'users/{farmer_uid}/whatsapp_interaction_log/{log_entry_key}', app=db_app).set(log_entry)
+
+
+        if updates: # If there are profile fields to update
+            updates['last_updated_by_admin_uid'] = admin_uid
+            updates['last_updated_at'] = datetime.now(timezone.utc).isoformat()
+            farmer_ref.update(updates)
+        
+        updated_farmer_data = farmer_ref.get() # Get the latest data
+
+        return jsonify({
+            'success': True, 
+            'message': f'WhatsApp-managed farmer {farmer_uid} updated.',
+            'profile': updated_farmer_data
+        }), 200
+
+    except Exception as e:
+        return handle_route_errors(e, uid_context=admin_uid)
+
+# --- END OF WHATSAPP-MANAGED FARMER ACCOUNT ENDPOINTS ---
 
 @app.route('/api/admin/deals/pending', methods=['GET'])
 def admin_get_pending_deals():