Initial commit

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);
+  });
+}

lib/data/database/daos/analysis_dao.dart

@@ -0,0 +1,32 @@
+import 'package:drift/drift.dart';
+import 'package:trainhub_flutter/data/database/app_database.dart';
+
+part 'analysis_dao.g.dart';
+
+@DriftAccessor(tables: [AnalysisSessions, Annotations])
+class AnalysisDao extends DatabaseAccessor<AppDatabase>
+    with _$AnalysisDaoMixin {
+  AnalysisDao(super.db);
+
+  Future<List<AnalysisSession>> getAllSessions() =>
+      select(analysisSessions).get();
+
+  Future<AnalysisSession?> getSession(String id) =>
+      (select(analysisSessions)..where((t) => t.id.equals(id)))
+          .getSingleOrNull();
+
+  Future<void> insertSession(AnalysisSessionsCompanion entry) =>
+      into(analysisSessions).insert(entry);
+
+  Future<void> deleteSession(String id) =>
+      (delete(analysisSessions)..where((t) => t.id.equals(id))).go();
+
+  Future<List<Annotation>> getAnnotations(String sessionId) =>
+      (select(annotations)..where((t) => t.sessionId.equals(sessionId))).get();
+
+  Future<void> insertAnnotation(AnnotationsCompanion entry) =>
+      into(annotations).insert(entry);
+
+  Future<void> deleteAnnotation(String id) =>
+      (delete(annotations)..where((t) => t.id.equals(id))).go();
+}

lib/data/database/daos/analysis_dao.g.dart

@@ -0,0 +1,10 @@
+// GENERATED CODE - DO NOT MODIFY BY HAND
+
+part of 'analysis_dao.dart';
+
+// ignore_for_file: type=lint
+mixin _$AnalysisDaoMixin on DatabaseAccessor<AppDatabase> {
+  $AnalysisSessionsTable get analysisSessions =>
+      attachedDatabase.analysisSessions;
+  $AnnotationsTable get annotations => attachedDatabase.annotations;
+}

lib/data/database/daos/chat_dao.dart

@@ -0,0 +1,43 @@
+import 'package:drift/drift.dart';
+import 'package:trainhub_flutter/data/database/app_database.dart';
+
+part 'chat_dao.g.dart';
+
+@DriftAccessor(tables: [ChatSessions, ChatMessages])
+class ChatDao extends DatabaseAccessor<AppDatabase> with _$ChatDaoMixin {
+  ChatDao(super.db);
+
+  Future<List<ChatSession>> getAllSessions() =>
+      (select(chatSessions)
+            ..orderBy([
+              (t) => OrderingTerm(
+                  expression: t.createdAt, mode: OrderingMode.desc)
+            ]))
+          .get();
+
+  Future<ChatSession?> getSession(String id) =>
+      (select(chatSessions)..where((t) => t.id.equals(id)))
+          .getSingleOrNull();
+
+  Future<void> insertSession(ChatSessionsCompanion entry) =>
+      into(chatSessions).insert(entry);
+
+  Future<void> deleteSession(String id) =>
+      (delete(chatSessions)..where((t) => t.id.equals(id))).go();
+
+  Future<void> updateSessionTitle(String id, String title) =>
+      (update(chatSessions)..where((t) => t.id.equals(id)))
+          .write(ChatSessionsCompanion(title: Value(title)));
+
+  Future<List<ChatMessage>> getMessages(String sessionId) =>
+      (select(chatMessages)
+            ..where((t) => t.sessionId.equals(sessionId))
+            ..orderBy([
+              (t) =>
+                  OrderingTerm(expression: t.createdAt, mode: OrderingMode.asc)
+            ]))
+          .get();
+
+  Future<void> insertMessage(ChatMessagesCompanion entry) =>
+      into(chatMessages).insert(entry);
+}

lib/data/database/daos/chat_dao.g.dart

@@ -0,0 +1,9 @@
+// GENERATED CODE - DO NOT MODIFY BY HAND
+
+part of 'chat_dao.dart';
+
+// ignore_for_file: type=lint
+mixin _$ChatDaoMixin on DatabaseAccessor<AppDatabase> {
+  $ChatSessionsTable get chatSessions => attachedDatabase.chatSessions;
+  $ChatMessagesTable get chatMessages => attachedDatabase.chatMessages;
+}

lib/data/database/daos/exercise_dao.dart

@@ -0,0 +1,24 @@
+import 'package:drift/drift.dart';
+import 'package:trainhub_flutter/data/database/app_database.dart';
+
+part 'exercise_dao.g.dart';
+
+@DriftAccessor(tables: [Exercises])
+class ExerciseDao extends DatabaseAccessor<AppDatabase>
+    with _$ExerciseDaoMixin {
+  ExerciseDao(super.db);
+
+  Future<List<Exercise>> getAllExercises() => select(exercises).get();
+
+  Future<Exercise> getExerciseById(String id) =>
+      (select(exercises)..where((t) => t.id.equals(id))).getSingle();
+
+  Future<void> insertExercise(ExercisesCompanion entry) =>
+      into(exercises).insert(entry);
+
+  Future<void> updateExercise(String id, ExercisesCompanion entry) =>
+      (update(exercises)..where((t) => t.id.equals(id))).write(entry);
+
+  Future<void> deleteExercise(String id) =>
+      (delete(exercises)..where((t) => t.id.equals(id))).go();
+}

lib/data/database/daos/exercise_dao.g.dart

@@ -0,0 +1,8 @@
+// GENERATED CODE - DO NOT MODIFY BY HAND
+
+part of 'exercise_dao.dart';
+
+// ignore_for_file: type=lint
+mixin _$ExerciseDaoMixin on DatabaseAccessor<AppDatabase> {
+  $ExercisesTable get exercises => attachedDatabase.exercises;
+}

lib/data/database/daos/program_dao.dart

@@ -0,0 +1,60 @@
+import 'package:drift/drift.dart';
+import 'package:trainhub_flutter/data/database/app_database.dart';
+
+part 'program_dao.g.dart';
+
+@DriftAccessor(tables: [Programs, ProgramWeeks, ProgramWorkouts])
+class ProgramDao extends DatabaseAccessor<AppDatabase>
+    with _$ProgramDaoMixin {
+  ProgramDao(super.db);
+
+  Future<List<Program>> getAllPrograms() =>
+      (select(programs)
+            ..orderBy([
+              (t) =>
+                  OrderingTerm(expression: t.createdAt, mode: OrderingMode.desc)
+            ]))
+          .get();
+
+  Future<Program?> getProgram(String id) =>
+      (select(programs)..where((t) => t.id.equals(id))).getSingleOrNull();
+
+  Future<void> insertProgram(ProgramsCompanion entry) =>
+      into(programs).insert(entry);
+
+  Future<void> deleteProgram(String id) =>
+      (delete(programs)..where((t) => t.id.equals(id))).go();
+
+  Future<List<ProgramWeek>> getWeeks(String programId) =>
+      (select(programWeeks)
+            ..where((t) => t.programId.equals(programId))
+            ..orderBy([(t) => OrderingTerm(expression: t.position)]))
+          .get();
+
+  Future<void> insertWeek(ProgramWeeksCompanion entry) =>
+      into(programWeeks).insert(entry);
+
+  Future<void> deleteWeek(String id) =>
+      (delete(programWeeks)..where((t) => t.id.equals(id))).go();
+
+  Future<void> updateWeekNote(String weekId, String note) =>
+      (update(programWeeks)..where((t) => t.id.equals(weekId)))
+          .write(ProgramWeeksCompanion(notes: Value(note)));
+
+  Future<List<ProgramWorkout>> getWorkouts(String programId) =>
+      (select(programWorkouts)..where((t) => t.programId.equals(programId)))
+          .get();
+
+  Future<void> insertWorkout(ProgramWorkoutsCompanion entry) =>
+      into(programWorkouts).insert(entry);
+
+  Future<void> updateWorkout(String id, ProgramWorkoutsCompanion entry) =>
+      (update(programWorkouts)..where((t) => t.id.equals(id))).write(entry);
+
+  Future<void> deleteWorkout(String id) =>
+      (delete(programWorkouts)..where((t) => t.id.equals(id))).go();
+
+  Future<void> toggleWorkoutComplete(String id, bool currentStatus) =>
+      (update(programWorkouts)..where((t) => t.id.equals(id)))
+          .write(ProgramWorkoutsCompanion(completed: Value(!currentStatus)));
+}

lib/data/database/daos/program_dao.g.dart

@@ -0,0 +1,10 @@
+// GENERATED CODE - DO NOT MODIFY BY HAND
+
+part of 'program_dao.dart';
+
+// ignore_for_file: type=lint
+mixin _$ProgramDaoMixin on DatabaseAccessor<AppDatabase> {
+  $ProgramsTable get programs => attachedDatabase.programs;
+  $ProgramWeeksTable get programWeeks => attachedDatabase.programWeeks;
+  $ProgramWorkoutsTable get programWorkouts => attachedDatabase.programWorkouts;
+}

lib/data/database/daos/training_plan_dao.dart

@@ -0,0 +1,24 @@
+import 'package:drift/drift.dart';
+import 'package:trainhub_flutter/data/database/app_database.dart';
+
+part 'training_plan_dao.g.dart';
+
+@DriftAccessor(tables: [TrainingPlans])
+class TrainingPlanDao extends DatabaseAccessor<AppDatabase>
+    with _$TrainingPlanDaoMixin {
+  TrainingPlanDao(super.db);
+
+  Future<List<TrainingPlan>> getAllPlans() => select(trainingPlans).get();
+
+  Future<TrainingPlan> getPlanById(String id) =>
+      (select(trainingPlans)..where((t) => t.id.equals(id))).getSingle();
+
+  Future<void> insertPlan(TrainingPlansCompanion entry) =>
+      into(trainingPlans).insert(entry);
+
+  Future<void> updatePlan(String id, TrainingPlansCompanion entry) =>
+      (update(trainingPlans)..where((t) => t.id.equals(id))).write(entry);
+
+  Future<void> deletePlan(String id) =>
+      (delete(trainingPlans)..where((t) => t.id.equals(id))).go();
+}

lib/data/database/daos/training_plan_dao.g.dart

@@ -0,0 +1,8 @@
+// GENERATED CODE - DO NOT MODIFY BY HAND
+
+part of 'training_plan_dao.dart';
+
+// ignore_for_file: type=lint
+mixin _$TrainingPlanDaoMixin on DatabaseAccessor<AppDatabase> {
+  $TrainingPlansTable get trainingPlans => attachedDatabase.trainingPlans;
+}