Nemcio/CyberPanel
1
0
Fork 0
mirror of https://github.com/usmannasir/cyberpanel.git synced 2025-11-12 16:26:12 +01:00
Code Issues Packages Projects Releases Wiki Activity

bug fix: add payment method for free scan users

Browse Source
This commit is contained in:
usmannasir
2025-06-27 14:51:33 +05:00
parent 08d9e113ce
commit 4e773a5488
1 changed files with 34 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
aiScanner/aiScannerManager.py
View File

@@ -477,13 +477,43 @@ class AIScannerManager:
# Field doesn't exist yet, allow access for now # Field doesn't exist yet, allow access for now
pass pass
# Check if user has scanner configured # Check if user has scanner configured (create if VPS user)
try: try:
scanner_settings = AIScannerSettings.objects.get(admin=admin) scanner_settings = AIScannerSettings.objects.get(admin=admin)
if not scanner_settings.is_payment_configured or not scanner_settings.api_key: if not scanner_settings.is_payment_configured or not scanner_settings.api_key:
return JsonResponse({'success': False, 'error': 'Initial payment setup required first'}) # Check if this is a VPS with free scans
server_ip = ACLManager.fetchIP()
vps_info = self.check_vps_free_scans(server_ip)
if vps_info.get('is_vps'):
# VPS users can add payment methods without initial setup
# Get or create VPS API key
vps_key_data = self.get_or_create_vps_api_key(server_ip)
if vps_key_data and vps_key_data.get('api_key'):
# Use VPS API key for adding payment method
api_key_to_use = vps_key_data.get('api_key')
else:
return JsonResponse({'success': False, 'error': 'Failed to authenticate VPS'})
else:
return JsonResponse({'success': False, 'error': 'Initial payment setup required first'})
else:
api_key_to_use = scanner_settings.api_key
except AIScannerSettings.DoesNotExist: except AIScannerSettings.DoesNotExist:
return JsonResponse({'success': False, 'error': 'Scanner not configured'}) # Check if this is a VPS with free scans
server_ip = ACLManager.fetchIP()
vps_info = self.check_vps_free_scans(server_ip)
if vps_info.get('is_vps'):
# VPS users can add payment methods without initial setup
# Get or create VPS API key
vps_key_data = self.get_or_create_vps_api_key(server_ip)
if vps_key_data and vps_key_data.get('api_key'):
# Use VPS API key for adding payment method
api_key_to_use = vps_key_data.get('api_key')
else:
return JsonResponse({'success': False, 'error': 'Failed to authenticate VPS'})
else:
return JsonResponse({'success': False, 'error': 'Scanner not configured'})
# Get admin email and domain # Get admin email and domain
cyberpanel_host = request.get_host() # Keep full host including port cyberpanel_host = request.get_host() # Keep full host including port
@@ -493,7 +523,7 @@ class AIScannerManager:
self.logger.writeToFile(f'[AIScannerManager.addPaymentMethod] Setting up new payment method for {admin.userName} (API key authentication)') self.logger.writeToFile(f'[AIScannerManager.addPaymentMethod] Setting up new payment method for {admin.userName} (API key authentication)')
# Call platform API to add payment method # Call platform API to add payment method
setup_data = self.setup_add_payment_method(scanner_settings.api_key, admin_email, cyberpanel_host) setup_data = self.setup_add_payment_method(api_key_to_use, admin_email, cyberpanel_host)
if setup_data: if setup_data:
self.logger.writeToFile(f'[AIScannerManager.addPaymentMethod] Payment method setup successful for {admin_email}') self.logger.writeToFile(f'[AIScannerManager.addPaymentMethod] Payment method setup successful for {admin_email}')