Added: Habit creation logic

app/models/Habit.php

@@ -0,0 +1,32 @@
+<?php
+
+class Habit {
+    private $db;
+
+    public function __construct() {
+        $this->db = Database::getInstance()->getConnection();
+    }
+
+    public function create($data) {
+        $stmt = $this->db->prepare("
+            INSERT INTO habits (user_id, title, frequency, custom_frequency, reward_points, created_at)
+            VALUES (?, ?, ?, ?, ?, NOW())
+        ");
+
+        $stmt->bind_param(
+            "isssi", // Bind types: int, string, string, string, int
+            $data['user_id'],
+            $data['name'],
+            $data['frequency'],
+            $data['custom_frequency'], // Bind the custom_frequency field
+            $data['reward_points']
+        );
+
+        if ($stmt->execute()) {
+            return true;
+        } else {
+            error_log("Failed to create habit: " . $stmt->error);
+            return false;
+        }
+    }
+}

app/models/User.php

@@ -42,7 +42,7 @@ class User {
     public function login($email, $password) {
         $hashedPassword = password_hash($password, PASSWORD_BCRYPT);
 
-        $stmt = $this->db->prepare("SELECT username, password FROM users WHERE email = ?");
+        $stmt = $this->db->prepare("SELECT id, username, password FROM users WHERE email = ?");
         $stmt->bind_param("s", $email);
         $stmt->execute();
         $result = $stmt->get_result();
@@ -52,6 +52,7 @@ class User {
             $user = $result->fetch_assoc();
             if (password_verify($password, $user['password'])) {
                 $_SESSION['user'] = [
+                    'id' => $user['id'],
                     'username' => $user['username'],
                     'email' => $email,
                 ];