mirror of
https://github.com/R0m1k3/xtremflow.git
synced 2026-10-12 01:36:20 +02:00
320 lines
8.5 KiB
Dart
320 lines
8.5 KiB
Dart
import 'dart:io';
|
|
import 'package:sqlite3/sqlite3.dart';
|
|
import 'package:uuid/uuid.dart';
|
|
import '../models/user.dart';
|
|
import '../models/playlist.dart';
|
|
import '../models/session.dart' as models;
|
|
import '../utils/password_hasher.dart';
|
|
|
|
class AppDatabase {
|
|
late final Database _db;
|
|
final _uuid = const Uuid();
|
|
|
|
/// Initialize database and create tables
|
|
Future<void> init() async {
|
|
final dbPath = '/app/data/xtremflow.db';
|
|
|
|
// Ensure data directory exists
|
|
final dir = Directory('/app/data');
|
|
if (!await dir.exists()) {
|
|
await dir.create(recursive: true);
|
|
}
|
|
|
|
_db = sqlite3.open(dbPath);
|
|
|
|
await _createTables();
|
|
print('Database initialized: $dbPath');
|
|
}
|
|
|
|
/// Create database tables
|
|
Future<void> _createTables() async {
|
|
// Users table
|
|
_db.execute('''
|
|
CREATE TABLE IF NOT EXISTS users (
|
|
id TEXT PRIMARY KEY,
|
|
username TEXT UNIQUE NOT NULL,
|
|
password_hash TEXT NOT NULL,
|
|
is_admin INTEGER DEFAULT 0,
|
|
created_at TEXT DEFAULT CURRENT_TIMESTAMP,
|
|
updated_at TEXT DEFAULT CURRENT_TIMESTAMP
|
|
)
|
|
''');
|
|
|
|
// Playlists table
|
|
_db.execute('''
|
|
CREATE TABLE IF NOT EXISTS playlists (
|
|
id TEXT PRIMARY KEY,
|
|
user_id TEXT NOT NULL,
|
|
name TEXT NOT NULL,
|
|
server_url TEXT NOT NULL,
|
|
username TEXT NOT NULL,
|
|
password TEXT NOT NULL,
|
|
dns TEXT,
|
|
created_at TEXT DEFAULT CURRENT_TIMESTAMP,
|
|
updated_at TEXT DEFAULT CURRENT_TIMESTAMP,
|
|
FOREIGN KEY (user_id) REFERENCES users(id) ON DELETE CASCADE
|
|
)
|
|
''');
|
|
|
|
// Sessions table
|
|
_db.execute('''
|
|
CREATE TABLE IF NOT EXISTS sessions (
|
|
id TEXT PRIMARY KEY,
|
|
user_id TEXT NOT NULL,
|
|
token TEXT UNIQUE NOT NULL,
|
|
expires_at TEXT NOT NULL,
|
|
created_at TEXT DEFAULT CURRENT_TIMESTAMP,
|
|
FOREIGN KEY (user_id) REFERENCES users(id) ON DELETE CASCADE
|
|
)
|
|
''');
|
|
|
|
// Indexes
|
|
_db.execute('CREATE INDEX IF NOT EXISTS idx_sessions_token ON sessions(token)');
|
|
_db.execute('CREATE INDEX IF NOT EXISTS idx_sessions_expires ON sessions(expires_at)');
|
|
_db.execute('CREATE INDEX IF NOT EXISTS idx_playlists_user ON playlists(user_id)');
|
|
}
|
|
|
|
/// Seed default admin user if no users exist
|
|
Future<void> seedAdmin() async {
|
|
final result = _db.select('SELECT COUNT(*) as count FROM users');
|
|
final count = result.first['count'] as int;
|
|
|
|
if (count == 0) {
|
|
final adminId = _uuid.v4();
|
|
final passwordHash = PasswordHasher.hash('admin');
|
|
|
|
_db.execute('''
|
|
INSERT INTO users (id, username, password_hash, is_admin)
|
|
VALUES (?, ?, ?, 1)
|
|
''', [adminId, 'admin', passwordHash]);
|
|
|
|
print('Default admin user created (username: admin, password: admin)');
|
|
}
|
|
}
|
|
|
|
// ==================== Users ====================
|
|
|
|
/// Find user by username
|
|
User? findUserByUsername(String username) {
|
|
final result = _db.select(
|
|
'SELECT * FROM users WHERE username = ?',
|
|
[username],
|
|
);
|
|
|
|
if (result.isEmpty) return null;
|
|
return User.fromMap(result.first);
|
|
}
|
|
|
|
/// Find user by ID
|
|
User? findUserById(String userId) {
|
|
final result = _db.select(
|
|
'SELECT * FROM users WHERE id = ?',
|
|
[userId],
|
|
);
|
|
|
|
if (result.isEmpty) return null;
|
|
return User.fromMap(result.first);
|
|
}
|
|
|
|
/// Verify user credentials
|
|
User? verifyCredentials(String username, String password) {
|
|
final result = _db.select(
|
|
'SELECT * FROM users WHERE username = ?',
|
|
[username],
|
|
);
|
|
|
|
if (result.isEmpty) return null;
|
|
|
|
final passwordHash = result.first['password_hash'] as String;
|
|
if (!PasswordHasher.verify(password, passwordHash)) {
|
|
return null;
|
|
}
|
|
|
|
return User.fromMap(result.first);
|
|
}
|
|
|
|
/// Create new user
|
|
User createUser(String username, String password, {bool isAdmin = false}) {
|
|
final userId = _uuid.v4();
|
|
final passwordHash = PasswordHasher.hash(password);
|
|
|
|
_db.execute('''
|
|
INSERT INTO users (id, username, password_hash, is_admin)
|
|
VALUES (?, ?, ?, ?)
|
|
''', [userId, username, passwordHash, isAdmin ? 1 : 0]);
|
|
|
|
return User(
|
|
id: userId,
|
|
username: username,
|
|
isAdmin: isAdmin,
|
|
createdAt: DateTime.now(),
|
|
);
|
|
}
|
|
|
|
/// Get all users
|
|
List<User> getAllUsers() {
|
|
final result = _db.select('SELECT * FROM users ORDER BY username ASC');
|
|
return result.map((row) => User.fromMap(row)).toList();
|
|
}
|
|
|
|
/// Update user password
|
|
void updateUserPassword(String userId, String newPassword) {
|
|
final passwordHash = PasswordHasher.hash(newPassword);
|
|
_db.execute(
|
|
'UPDATE users SET password_hash = ?, updated_at = CURRENT_TIMESTAMP WHERE id = ?',
|
|
[passwordHash, userId],
|
|
);
|
|
}
|
|
|
|
/// Update user administration status
|
|
void updateUserAdminStatus(String userId, bool isAdmin) {
|
|
_db.execute(
|
|
'UPDATE users SET is_admin = ?, updated_at = CURRENT_TIMESTAMP WHERE id = ?',
|
|
[isAdmin ? 1 : 0, userId],
|
|
);
|
|
}
|
|
|
|
/// Delete user
|
|
void deleteUser(String userId) {
|
|
_db.execute('DELETE FROM users WHERE id = ?', [userId]);
|
|
}
|
|
|
|
// ==================== Sessions ====================
|
|
|
|
/// Create new session
|
|
models.Session createSession(String userId, {Duration? duration}) {
|
|
final sessionId = _uuid.v4();
|
|
final token = _uuid.v4();
|
|
final expiresAt = DateTime.now().add(duration ?? const Duration(days: 7));
|
|
|
|
_db.execute('''
|
|
INSERT INTO sessions (id, user_id, token, expires_at)
|
|
VALUES (?, ?, ?, ?)
|
|
''', [sessionId, userId, token, expiresAt.toIso8601String()]);
|
|
|
|
return models.Session(
|
|
id: sessionId,
|
|
userId: userId,
|
|
token: token,
|
|
expiresAt: expiresAt,
|
|
createdAt: DateTime.now(),
|
|
);
|
|
}
|
|
|
|
/// Find session by token
|
|
models.Session? findSessionByToken(String token) {
|
|
final result = _db.select(
|
|
'SELECT * FROM sessions WHERE token = ?',
|
|
[token],
|
|
);
|
|
|
|
if (result.isEmpty) return null;
|
|
|
|
final session = models.Session.fromMap(result.first);
|
|
|
|
// Check if expired
|
|
if (session.isExpired) {
|
|
deleteSession(token);
|
|
return null;
|
|
}
|
|
|
|
return session;
|
|
}
|
|
|
|
/// Delete session (logout)
|
|
void deleteSession(String token) {
|
|
_db.execute('DELETE FROM sessions WHERE token = ?', [token]);
|
|
}
|
|
|
|
/// Clean expired sessions
|
|
void cleanExpiredSessions() {
|
|
_db.execute(
|
|
'DELETE FROM sessions WHERE expires_at < ?',
|
|
[DateTime.now().toIso8601String()],
|
|
);
|
|
}
|
|
|
|
// ==================== Playlists ====================
|
|
|
|
/// Get all playlists for a user
|
|
List<Playlist> getPlaylists(String userId) {
|
|
final result = _db.select(
|
|
'SELECT * FROM playlists WHERE user_id = ? ORDER BY created_at DESC',
|
|
[userId],
|
|
);
|
|
|
|
return result.map((row) => Playlist.fromMap(row)).toList();
|
|
}
|
|
|
|
/// Get playlist by ID
|
|
Playlist? getPlaylistById(String playlistId) {
|
|
final result = _db.select(
|
|
'SELECT * FROM playlists WHERE id = ?',
|
|
[playlistId],
|
|
);
|
|
|
|
if (result.isEmpty) return null;
|
|
return Playlist.fromMap(result.first);
|
|
}
|
|
|
|
/// Create new playlist
|
|
Playlist createPlaylist({
|
|
required String userId,
|
|
required String name,
|
|
required String serverUrl,
|
|
required String username,
|
|
required String password,
|
|
String? dns,
|
|
}) {
|
|
final playlistId = _uuid.v4();
|
|
final now = DateTime.now().toIso8601String();
|
|
|
|
_db.execute('''
|
|
INSERT INTO playlists (id, user_id, name, server_url, username, password, dns, created_at, updated_at)
|
|
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
|
|
''', [playlistId, userId, name, serverUrl, username, password, dns, now, now]);
|
|
|
|
return Playlist(
|
|
id: playlistId,
|
|
userId: userId,
|
|
name: name,
|
|
serverUrl: serverUrl,
|
|
username: username,
|
|
password: password,
|
|
dns: dns,
|
|
createdAt: DateTime.parse(now),
|
|
updatedAt: DateTime.parse(now),
|
|
);
|
|
}
|
|
|
|
/// Update playlist
|
|
Playlist updatePlaylist({
|
|
required String playlistId,
|
|
required String name,
|
|
required String serverUrl,
|
|
required String username,
|
|
required String password,
|
|
String? dns,
|
|
}) {
|
|
final now = DateTime.now().toIso8601String();
|
|
|
|
_db.execute('''
|
|
UPDATE playlists
|
|
SET name = ?, server_url = ?, username = ?, password = ?, dns = ?, updated_at = ?
|
|
WHERE id = ?
|
|
''', [name, serverUrl, username, password, dns, now, playlistId]);
|
|
|
|
return getPlaylistById(playlistId)!;
|
|
}
|
|
|
|
/// Delete playlist
|
|
void deletePlaylist(String playlistId) {
|
|
_db.execute('DELETE FROM playlists WHERE id = ?', [playlistId]);
|
|
}
|
|
|
|
/// Close database connection
|
|
void close() {
|
|
_db.dispose();
|
|
}
|
|
}
|