90 lines
3.3 KiB
Python
90 lines
3.3 KiB
Python
import re
|
|
import glob
|
|
|
|
# 1. ServerConfig missing fields
|
|
for path in ['tests/security_test.rs', 'tests/integration_test.rs']:
|
|
try:
|
|
with open(path, 'r') as f:
|
|
content = f.read()
|
|
|
|
content = re.sub(
|
|
r'server:\s*nx9_auth::config::ServerConfig\s*\{\s*host:\s*"127\.0\.0\.1"\.into\(\),\s*port:\s*8080,?\s*\}',
|
|
r'server: nx9_auth::config::ServerConfig {\n host: "127.0.0.1".into(),\n port: 8080,\n cookie_secure: false,\n production: false,\n }',
|
|
content
|
|
)
|
|
|
|
with open(path, 'w') as f:
|
|
f.write(content)
|
|
except FileNotFoundError:
|
|
pass
|
|
|
|
# 2. cli_test.rs
|
|
path = 'tests/cli_test.rs'
|
|
try:
|
|
with open(path, 'r') as f:
|
|
content = f.read()
|
|
|
|
content = content.replace('nx9_auth::db::run_sqlite_migrations(&provider)', 'nx9_auth::db::run_migrations(&pool)')
|
|
content = content.replace('nx9_auth::db::run_sqlite_migrations(&pool)', 'nx9_auth::db::run_migrations(&pool)')
|
|
|
|
# username_exists
|
|
content = content.replace(
|
|
'nx9_auth::db::repository::users::username_exists(\n &provider,',
|
|
'provider.users().username_exists('
|
|
)
|
|
content = content.replace(
|
|
'nx9_auth::db::repository::users::username_exists(&provider,',
|
|
'provider.users().username_exists('
|
|
)
|
|
|
|
# Ensure provider is instantiated for `provider.users().username_exists` in cli_test.rs if needed.
|
|
# Actually, cli_test.rs does NOT have a provider. It has a pool!
|
|
# Wait, `provider` was in the compile error: `tests/cli_test.rs:160: &provider not found in this scope`.
|
|
# Let me just provide a provider if pool is there!
|
|
content = content.replace(
|
|
'let admin_exists = provider.users().username_exists(',
|
|
'let provider = std::sync::Arc::new(nx9_auth::db::provider::SqliteProvider::new(pool.clone()));\n let admin_exists = provider.users().username_exists('
|
|
)
|
|
|
|
with open(path, 'w') as f:
|
|
f.write(content)
|
|
except FileNotFoundError:
|
|
pass
|
|
|
|
# 3. migration_compatibility.rs
|
|
path = 'tests/migration_compatibility.rs'
|
|
try:
|
|
with open(path, 'r') as f:
|
|
content = f.read()
|
|
|
|
content = content.replace(
|
|
'let admin_role = provider.roles().find_by_name(',
|
|
'let provider = std::sync::Arc::new(nx9_auth::db::provider::SqliteProvider::new(pool.clone()));\n let admin_role = provider.roles().find_by_name('
|
|
)
|
|
|
|
with open(path, 'w') as f:
|
|
f.write(content)
|
|
except FileNotFoundError:
|
|
pass
|
|
|
|
# 4. src/main.rs - E0308 PostgresProvider::new(pool) where pool is SqlitePool
|
|
path = 'src/main.rs'
|
|
try:
|
|
with open(path, 'r') as f:
|
|
content = f.read()
|
|
|
|
# Replace the conflicting conditional logic to just use SqliteProvider for now.
|
|
# Or properly cfg(feature).
|
|
# Since we must keep SQLite exclusively per instructions:
|
|
content = re.sub(
|
|
r'#\[cfg\(feature = "postgres"\)\].*?let provider.*?SqliteProvider::new\(pool\)\)\s*\};',
|
|
r'let provider: std::sync::Arc<dyn db::provider::DatabaseProvider> = std::sync::Arc::new(db::provider::SqliteProvider::new(pool));',
|
|
content,
|
|
flags=re.DOTALL
|
|
)
|
|
|
|
with open(path, 'w') as f:
|
|
f.write(content)
|
|
except FileNotFoundError:
|
|
pass
|