Initial commit
This commit is contained in:
148
lib/database/database.dart
Normal file
148
lib/database/database.dart
Normal file
@@ -0,0 +1,148 @@
|
||||
import 'package:drift/drift.dart';
|
||||
import 'package:drift/native.dart';
|
||||
import 'package:path_provider/path_provider.dart';
|
||||
import 'package:path/path.dart' as p;
|
||||
import 'dart:io';
|
||||
|
||||
part 'database.g.dart';
|
||||
|
||||
// Exercises
|
||||
class Exercises extends Table {
|
||||
TextColumn get id => text()();
|
||||
TextColumn get name => text()();
|
||||
TextColumn get instructions => text().nullable()();
|
||||
TextColumn get enrichment => text().nullable()();
|
||||
TextColumn get tags => text().nullable()(); // JSON string
|
||||
TextColumn get videoUrl => text().nullable()();
|
||||
|
||||
@override
|
||||
Set<Column> get primaryKey => {id};
|
||||
}
|
||||
|
||||
// TrainingPlans
|
||||
class TrainingPlans extends Table {
|
||||
TextColumn get id => text()();
|
||||
TextColumn get name => text()();
|
||||
TextColumn get sections => text().nullable()(); // JSON string
|
||||
|
||||
@override
|
||||
Set<Column> get primaryKey => {id};
|
||||
}
|
||||
|
||||
// Programs
|
||||
class Programs extends Table {
|
||||
TextColumn get id => text()();
|
||||
TextColumn get name => text()();
|
||||
TextColumn get createdAt => text()();
|
||||
|
||||
@override
|
||||
Set<Column> get primaryKey => {id};
|
||||
}
|
||||
|
||||
// ProgramWeeks
|
||||
class ProgramWeeks extends Table {
|
||||
TextColumn get id => text()();
|
||||
TextColumn get programId =>
|
||||
text().references(Programs, #id, onDelete: KeyAction.cascade)();
|
||||
IntColumn get position => integer()();
|
||||
TextColumn get notes => text().nullable()();
|
||||
|
||||
@override
|
||||
Set<Column> get primaryKey => {id};
|
||||
}
|
||||
|
||||
// ProgramWorkouts
|
||||
class ProgramWorkouts extends Table {
|
||||
TextColumn get id => text()();
|
||||
TextColumn get weekId =>
|
||||
text().references(ProgramWeeks, #id, onDelete: KeyAction.cascade)();
|
||||
TextColumn get programId =>
|
||||
text().references(Programs, #id, onDelete: KeyAction.cascade)();
|
||||
TextColumn get day => text()();
|
||||
TextColumn get type => text()(); // "exercise" | "plan"
|
||||
TextColumn get refId => text().nullable()();
|
||||
TextColumn get name => text().nullable()();
|
||||
TextColumn get description => text().nullable()();
|
||||
BoolColumn get completed => boolean().withDefault(const Constant(false))();
|
||||
|
||||
@override
|
||||
Set<Column> get primaryKey => {id};
|
||||
}
|
||||
|
||||
// AnalysisSessions
|
||||
class AnalysisSessions extends Table {
|
||||
TextColumn get id => text()();
|
||||
TextColumn get name => text()();
|
||||
TextColumn get date => text()();
|
||||
TextColumn get videoPath => text().nullable()();
|
||||
|
||||
@override
|
||||
Set<Column> get primaryKey => {id};
|
||||
}
|
||||
|
||||
// Annotations
|
||||
class Annotations extends Table {
|
||||
TextColumn get id => text()();
|
||||
TextColumn get sessionId =>
|
||||
text().references(AnalysisSessions, #id, onDelete: KeyAction.cascade)();
|
||||
RealColumn get startTime => real()();
|
||||
RealColumn get endTime => real()();
|
||||
TextColumn get name => text().nullable()();
|
||||
TextColumn get description => text().nullable()();
|
||||
TextColumn get color => text().nullable()();
|
||||
|
||||
@override
|
||||
Set<Column> get primaryKey => {id};
|
||||
}
|
||||
|
||||
// ChatSessions
|
||||
class ChatSessions extends Table {
|
||||
TextColumn get id => text()();
|
||||
TextColumn get title => text().nullable()();
|
||||
TextColumn get createdAt => text()();
|
||||
TextColumn get updatedAt => text()();
|
||||
|
||||
@override
|
||||
Set<Column> get primaryKey => {id};
|
||||
}
|
||||
|
||||
// ChatMessages
|
||||
class ChatMessages extends Table {
|
||||
TextColumn get id => text()();
|
||||
TextColumn get sessionId =>
|
||||
text().references(ChatSessions, #id, onDelete: KeyAction.cascade)();
|
||||
TextColumn get role => text()(); // 'user' | 'assistant'
|
||||
TextColumn get content => text()();
|
||||
TextColumn get createdAt => text()();
|
||||
|
||||
@override
|
||||
Set<Column> get primaryKey => {id};
|
||||
}
|
||||
|
||||
@DriftDatabase(
|
||||
tables: [
|
||||
Exercises,
|
||||
TrainingPlans,
|
||||
Programs,
|
||||
ProgramWeeks,
|
||||
ProgramWorkouts,
|
||||
AnalysisSessions,
|
||||
Annotations,
|
||||
ChatSessions,
|
||||
ChatMessages,
|
||||
],
|
||||
)
|
||||
class AppDatabase extends _$AppDatabase {
|
||||
AppDatabase() : super(_openConnection());
|
||||
|
||||
@override
|
||||
int get schemaVersion => 1;
|
||||
}
|
||||
|
||||
LazyDatabase _openConnection() {
|
||||
return LazyDatabase(() async {
|
||||
final dbFolder = await getApplicationDocumentsDirectory();
|
||||
final file = File(p.join(dbFolder.path, 'trainhub.sqlite'));
|
||||
return NativeDatabase.createInBackground(file);
|
||||
});
|
||||
}
|
||||
Reference in New Issue
Block a user