Initial commit

2026-02-19 02:49:29 +01:00
commit 782986a632
148 changed files with 29230 additions and 0 deletions
--- a/lib/data/database/app_database.dart
+++ b/lib/data/database/app_database.dart
@@ -0,0 +1,152 @@
+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 'package:trainhub_flutter/core/constants/app_constants.dart';
+import 'package:trainhub_flutter/data/database/daos/exercise_dao.dart';
+import 'package:trainhub_flutter/data/database/daos/training_plan_dao.dart';
+import 'package:trainhub_flutter/data/database/daos/program_dao.dart';
+import 'package:trainhub_flutter/data/database/daos/analysis_dao.dart';
+import 'package:trainhub_flutter/data/database/daos/chat_dao.dart';
+import 'dart:io';
+
+part 'app_database.g.dart';
+
+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()();
+  TextColumn get videoUrl => text().nullable()();
+
+  @override
+  Set<Column> get primaryKey => {id};
+}
+
+class TrainingPlans extends Table {
+  TextColumn get id => text()();
+  TextColumn get name => text()();
+  TextColumn get sections => text().nullable()();
+
+  @override
+  Set<Column> get primaryKey => {id};
+}
+
+class Programs extends Table {
+  TextColumn get id => text()();
+  TextColumn get name => text()();
+  TextColumn get createdAt => text()();
+
+  @override
+  Set<Column> get primaryKey => {id};
+}
+
+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};
+}
+
+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()();
+  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};
+}
+
+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};
+}
+
+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};
+}
+
+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};
+}
+
+class ChatMessages extends Table {
+  TextColumn get id => text()();
+  TextColumn get sessionId =>
+      text().references(ChatSessions, #id, onDelete: KeyAction.cascade)();
+  TextColumn get role => text()();
+  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,
+  ],
+  daos: [
+    ExerciseDao,
+    TrainingPlanDao,
+    ProgramDao,
+    AnalysisDao,
+    ChatDao,
+  ],
+)
+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, AppConstants.databaseName));
+    return NativeDatabase.createInBackground(file);
+  });
+}