drunken-chunk 1.0.0

package/drunken_chunk/db.py

@@ -0,0 +1,245 @@
+import os
+import sqlite3
+from datetime import datetime
+
+DB_DIR = os.path.expanduser("~/.drunken_chunk")
+DB_PATH = os.environ.get("DRUNKEN_CHUNK_DB_PATH", os.path.join(DB_DIR, "drunken_chunk.db"))
+
+def get_connection():
+    """Initializes the database directory and returns a sqlite3 connection."""
+    db_dir = os.path.dirname(DB_PATH)
+    if db_dir:
+        os.makedirs(db_dir, exist_ok=True)
+    conn = sqlite3.connect(DB_PATH)
+    conn.row_factory = sqlite3.Row
+    return conn
+
+def init_db():
+    """Creates tables if they do not exist."""
+    conn = get_connection()
+    cursor = conn.cursor()
+
+    # Profile Table
+    cursor.execute("""
+        CREATE TABLE IF NOT EXISTS profile (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            name TEXT NOT NULL,
+            weight REAL NOT NULL,        -- In kg
+            height REAL NOT NULL,        -- In cm
+            wake_time TEXT NOT NULL,     -- Format: HH:MM
+            sleep_time TEXT NOT NULL,    -- Format: HH:MM
+            reminder_interval INTEGER NOT NULL, -- In minutes
+            custom_target INTEGER,       -- In ml (optional override)
+            joined_date TEXT NOT NULL
+        )
+    """)
+
+    # Water Intake Log Table
+    cursor.execute("""
+        CREATE TABLE IF NOT EXISTS water_log (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            timestamp TEXT NOT NULL,     -- Format: YYYY-MM-DD HH:MM:SS
+            amount INTEGER NOT NULL,     -- In ml
+            type TEXT NOT NULL           -- 'direct' or name of shortcut
+        )
+    """)
+
+    # Shortcuts Table
+    cursor.execute("""
+        CREATE TABLE IF NOT EXISTS shortcuts (
+            name TEXT PRIMARY KEY,
+            amount INTEGER NOT NULL      -- In ml
+        )
+    """)
+
+    # Reminder Daemon State Table
+    cursor.execute("""
+        CREATE TABLE IF NOT EXISTS reminder_state (
+            key TEXT PRIMARY KEY,
+            value TEXT NOT NULL
+        )
+    """)
+
+    # Insert default active state for reminder if not present
+    cursor.execute("INSERT OR IGNORE INTO reminder_state (key, value) VALUES ('is_active', '0')")
+    cursor.execute("INSERT OR IGNORE INTO reminder_state (key, value) VALUES ('last_notified', '')")
+
+    # Add default shortcuts if table is empty
+    cursor.execute("SELECT COUNT(*) FROM shortcuts")
+    if cursor.fetchone()[0] == 0:
+        default_shortcuts = [
+            ("glass", 250),
+            ("cup", 150),
+            ("bottle", 750),
+            ("flask", 500)
+        ]
+        cursor.executemany("INSERT INTO shortcuts (name, amount) VALUES (?, ?)", default_shortcuts)
+
+    conn.commit()
+    conn.close()
+
+# ---------------- PROFILE OPERATIONS ----------------
+
+def get_profile():
+    """Gets the user profile. Returns dict or None."""
+    conn = get_connection()
+    cursor = conn.cursor()
+    cursor.execute("SELECT * FROM profile ORDER BY id DESC LIMIT 1")
+    row = cursor.fetchone()
+    conn.close()
+    return dict(row) if row else None
+
+def save_profile(name, weight, height, wake_time, sleep_time, reminder_interval, custom_target=None):
+    """Saves or updates user profile."""
+    conn = get_connection()
+    cursor = conn.cursor()
+    # Check if there is already a profile
+    cursor.execute("SELECT id FROM profile LIMIT 1")
+    row = cursor.fetchone()
+    
+    now_str = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    if row:
+        profile_id = row['id']
+        cursor.execute("""
+            UPDATE profile 
+            SET name = ?, weight = ?, height = ?, wake_time = ?, sleep_time = ?, 
+                reminder_interval = ?, custom_target = ?
+            WHERE id = ?
+        """, (name, weight, height, wake_time, sleep_time, reminder_interval, custom_target, profile_id))
+    else:
+        cursor.execute("""
+            INSERT INTO profile (name, weight, height, wake_time, sleep_time, reminder_interval, custom_target, joined_date)
+            VALUES (?, ?, ?, ?, ?, ?, ?, ?)
+        """, (name, weight, height, wake_time, sleep_time, reminder_interval, custom_target, now_str))
+        
+    conn.commit()
+    conn.close()
+
+# ---------------- LOGGING OPERATIONS ----------------
+
+def log_water(amount, drink_type="direct", timestamp=None):
+    """Logs water consumption at a specific time (defaults to now)."""
+    if not timestamp:
+        timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    conn = get_connection()
+    cursor = conn.cursor()
+    cursor.execute("""
+        INSERT INTO water_log (timestamp, amount, type)
+        VALUES (?, ?, ?)
+    """, (timestamp, amount, drink_type))
+    conn.commit()
+    conn.close()
+
+def delete_log(log_id):
+    """Deletes a water log entry by its ID."""
+    conn = get_connection()
+    cursor = conn.cursor()
+    cursor.execute("DELETE FROM water_log WHERE id = ?", (log_id,))
+    conn.commit()
+    conn.close()
+
+def get_today_logs():
+    """Retrieves all water logs for today."""
+    today = datetime.now().strftime("%Y-%m-%d")
+    conn = get_connection()
+    cursor = conn.cursor()
+    cursor.execute("""
+        SELECT * FROM water_log 
+        WHERE timestamp LIKE ? 
+        ORDER BY timestamp ASC
+    """, (f"{today}%",))
+    rows = cursor.fetchall()
+    conn.close()
+    return [dict(r) for r in rows]
+
+def get_history_logs(days=7):
+    """Retrieves water logs for the last N days."""
+    conn = get_connection()
+    cursor = conn.cursor()
+    cursor.execute("""
+        SELECT * FROM water_log 
+        WHERE timestamp >= datetime('now', ?) 
+        ORDER BY timestamp DESC
+    """, (f"-{days} days",))
+    rows = cursor.fetchall()
+    conn.close()
+    return [dict(r) for r in rows]
+
+def get_all_logs():
+    """Retrieves all logs in the database."""
+    conn = get_connection()
+    cursor = conn.cursor()
+    cursor.execute("SELECT * FROM water_log ORDER BY timestamp DESC")
+    rows = cursor.fetchall()
+    conn.close()
+    return [dict(r) for r in rows]
+
+# ---------------- SHORTCUT OPERATIONS ----------------
+
+def get_shortcuts():
+    """Retrieves all shortcuts."""
+    conn = get_connection()
+    cursor = conn.cursor()
+    cursor.execute("SELECT * FROM shortcuts ORDER BY amount ASC")
+    rows = cursor.fetchall()
+    conn.close()
+    return {r['name']: r['amount'] for r in rows}
+
+def save_shortcut(name, amount):
+    """Saves or updates a shortcut."""
+    conn = get_connection()
+    cursor = conn.cursor()
+    cursor.execute("""
+        INSERT INTO shortcuts (name, amount) 
+        VALUES (?, ?)
+        ON CONFLICT(name) DO UPDATE SET amount = excluded.amount
+    """, (name.lower(), amount))
+    conn.commit()
+    conn.close()
+
+def delete_shortcut(name):
+    """Deletes a shortcut."""
+    conn = get_connection()
+    cursor = conn.cursor()
+    cursor.execute("DELETE FROM shortcuts WHERE name = ?", (name.lower(),))
+    conn.commit()
+    conn.close()
+
+# ---------------- DAEMON STATE OPERATIONS ----------------
+
+def set_daemon_state(key, value):
+    """Sets a configuration state for the daemon."""
+    conn = get_connection()
+    cursor = conn.cursor()
+    cursor.execute("""
+        INSERT INTO reminder_state (key, value)
+        VALUES (?, ?)
+        ON CONFLICT(key) DO UPDATE SET value = excluded.value
+    """, (key, str(value)))
+    conn.commit()
+    conn.close()
+
+def get_daemon_state(key, default=None):
+    """Retrieves a configuration state for the daemon."""
+    conn = get_connection()
+    cursor = conn.cursor()
+    cursor.execute("SELECT value FROM reminder_state WHERE key = ?", (key,))
+    row = cursor.fetchone()
+    conn.close()
+    return row['value'] if row else default
+
+def undo_last_log():
+    """Deletes the most recent water log entry. Returns dict of deleted entry or None."""
+    conn = get_connection()
+    cursor = conn.cursor()
+    cursor.execute("SELECT * FROM water_log ORDER BY id DESC LIMIT 1")
+    row = cursor.fetchone()
+    if row:
+        log_dict = dict(row)
+        cursor.execute("DELETE FROM water_log WHERE id = ?", (log_dict['id'],))
+        conn.commit()
+        conn.close()
+        return log_dict
+    conn.close()
+    return None
+

package/drunken_chunk/profile.py

@@ -0,0 +1,232 @@
+import math
+import random
+from .utils import Colors, colorize
+
+def calculate_bmi(weight_kg, height_cm):
+    """Calculates BMI (Body Mass Index)."""
+    if height_cm <= 0:
+        return 0
+    height_m = height_cm / 100.0
+    return weight_kg / (height_m * height_m)
+
+def calculate_recommended_intake(weight_kg, height_cm):
+    """
+    Calculates recommended daily water intake in ml.
+    Standard baseline: 35 ml per kg of body weight.
+    Adjustments: High BMI bodies require slightly more water for metabolic activity.
+    """
+    if weight_kg <= 0 or height_cm <= 0:
+        return 2000 # default fallback
+        
+    baseline = weight_kg * 35  # in ml
+    
+    # BMI adjustment
+    bmi = calculate_bmi(weight_kg, height_cm)
+    if bmi > 25:
+        # Increase target by 1% per BMI unit above 25, capped at 15%
+        adjustment_factor = min(1.15, 1.0 + (bmi - 25) * 0.01)
+        recommended = baseline * adjustment_factor
+    elif bmi < 18.5 and bmi > 0:
+        # Decrease slightly but not below baseline * 0.9
+        adjustment_factor = max(0.9, 1.0 - (18.5 - bmi) * 0.01)
+        recommended = baseline * adjustment_factor
+    else:
+        recommended = baseline
+        
+    # Round to nearest 50ml
+    return int(math.ceil(recommended / 50.0)) * 50
+
+def get_bmi_category(bmi):
+    if bmi < 18.5:
+        return "Underweight", Colors.YELLOW
+    elif 18.5 <= bmi < 25:
+        return "Normal weight", Colors.GREEN
+    elif 25 <= bmi < 30:
+        return "Overweight", Colors.YELLOW
+    else:
+        return "Obese", Colors.RED
+
+def get_hydration_level(consumed_ml, target_ml):
+    """
+    Determines hydration level and gives a status string with emojis.
+    Returns: (level_name, color, description_text)
+    """
+    if target_ml <= 0:
+        return "Unknown", Colors.GRAY, "Set up your target first."
+        
+    ratio = consumed_ml / target_ml
+    
+    if ratio == 0:
+        return "Dry Soil [DRY]", Colors.RED, "No water recorded yet today. Drink up!"
+    elif ratio < 0.25:
+        return "Dehydrated Cactus [CRITICAL]", Colors.RED, "Very low hydration! Your body is in emergency drought mode."
+    elif ratio < 0.50:
+        return "Thirsty Camel [LOW]", Colors.YELLOW, "Below average hydration. Time to fill up your hump."
+    elif ratio < 0.75:
+        return "Wandering Nomad [MODERATE]", Colors.CYAN, "Decent progress, but you are still wandering in the hydration desert."
+    elif ratio < 1.00:
+        return "Hydrated Dolphin [HYDRATED]", Colors.GREEN, "Excellent hydration! You are swimming smoothly and feeling alert."
+    else:
+        return "Poseidon / Water God [OPTIMAL]", Colors.MAGENTA, "Perfect hydration! You have achieved peak fluid flow."
+
+def get_profile_insights(profile, consumed_today):
+    """Generates detailed body and hydration insights."""
+    weight = profile['weight']
+    height = profile['height']
+    bmi = calculate_bmi(weight, height)
+    bmi_cat, bmi_color = get_bmi_category(bmi)
+    
+    target = profile['custom_target'] if profile['custom_target'] else calculate_recommended_intake(weight, height)
+    ratio = consumed_today / target
+    
+    level_name, level_color, level_desc = get_hydration_level(consumed_today, target)
+    
+    insights = []
+    insights.append(f"• Your BMI is {colorize(f'{bmi:.1f}', Colors.BOLD)} ({colorize(bmi_cat, bmi_color)}).")
+    
+    # Calculate metabolic demand info
+    if bmi > 25:
+        insights.append("• Higher body mass increases metabolic demands; your target has been scaled up accordingly.")
+    elif bmi < 18.5:
+        insights.append("• Lower body mass reduces base water requirement slightly, but don't compromise on regular sips.")
+    else:
+        insights.append("• Your body mass indexes in the normal range. Standard metabolic demand applies.")
+
+    # Hydration level specific recommendations
+    if ratio < 0.5:
+        insights.append(f"• {colorize('WARNING:', Colors.RED + Colors.BOLD)} Low hydration leads to fatigue, headaches, and low focus.")
+        insights.append("• Action item: Set a shorter reminder interval or make sure your shortcuts are logged.")
+    elif ratio < 1.0:
+        insights.append("• You are on the right track! Keeping a steady flow prevents afternoon crashes.")
+    else:
+        insights.append(f"• {colorize('CONGRATS!', Colors.GREEN + Colors.BOLD)} You've met your daily target. Extra consumption will be flushed out, keep listening to your body.")
+        
+    return {
+        "target": target,
+        "bmi": bmi,
+        "bmi_category": bmi_cat,
+        "bmi_color": bmi_color,
+        "level_name": level_name,
+        "level_color": level_color,
+        "level_desc": level_desc,
+        "ratio": ratio,
+        "insights": insights
+    }
+
+def get_situational_art_and_quote(ratio):
+    """Returns a tuple of (ascii_art_str, funny_quote_str) based on hydration ratio."""
+    if ratio == 0:
+        art = r"""
+  ╔═════════════════╗
+  ║  :(  SYSTEM     ║
+  ║                 ║
+  ║  thirst.sys     ║
+  ║  has crashed    ║
+  ║                 ║
+  ║  ERR: 0x00H2O  ║
+  ╚═════════════════╝
+        """
+        quotes = [
+            "Warning: You are literally 1HP bro XD. Drink some water immediately before you despawn!",
+            "Your kidneys are crying: 'Lag! Lag! I am at 1HP!' Go drink some water before you disconnect.",
+            "You are at 1HP, out of mana, and your organs are throwing 500 Internal Server Errors. DRINK WATER!",
+            "Keyboard warrior down! You are literally at 1HP bro. Throw a health potion down your throat.",
+            "System check: Dehydration level 100%. Please plug in your water cable.",
+            "Your kidney is running a garbage collection loop. Give it some water to clear the heap."
+        ]
+        quote = random.choice(quotes)
+    elif ratio < 0.25:
+        art = r"""
+      _______
+    .'_______'.
+   /  X     X  \
+  |      ^      |
+  |    -----    |
+   '._________.'
+   [ GAME  OVER ]
+   [ 0 CREDITS  ]
+        """
+        quotes = [
+            "Low health alert! 15HP remaining. Spiky, dry, and lagging hard. Recharge your shield cell!",
+            "Danger: Shield down, health critical. If a bug attacks you now, you're dead. Sip some water.",
+            "Your cells are begging for a heal. Current status: lagging in the lobby with 15HP.",
+            "You are drier than a poorly documented legacy codebase. Drink up to restore some frames.",
+            "Organ performance degraded. Latency rising. Sip some water to avoid thermal throttling.",
+            "Your skin cells are looking like dried raisins. Help them out, drink a glass."
+        ]
+        quote = random.choice(quotes)
+    elif ratio < 0.50:
+        art = r"""
+   (\(\
+   (>_<)  "pls send h2o"
+   (  )(")(")
+   ─────────────
+   // TODO: drink
+   // FIXME: NOW
+        """
+        quotes = [
+            "Half health! Camel mode activated. Storing water in your hump? biological storage failing!",
+            "40HP remaining. Your compilation times are starting to slow down. Lubricate the gears!",
+            "A wild dehydration monster appeared. You are holding your ground at 40HP, but barely.",
+            "Your humps are getting flat. Storing code snippets instead of water? Go grab a flask.",
+            "Halfway to the safe zone. Your brain is running on low power mode. Take a gulp.",
+            "Biological buffer is half empty. Grab a glass to keep the pipeline flowing."
+        ]
+        quote = random.choice(quotes)
+    elif ratio < 0.75:
+        art = r"""
+   (\(\
+   (-.-)  "still alive..."
+   (  )(")(")
+   ─────────────
+   > running...
+   H2O : MEDIUM
+        """
+        quotes = [
+            "60% HP. Desert nomad status. You've escaped absolute drought, but you're not safe yet. Take a gulp.",
+            "60HP. Decent performance, but your brain buffer is half empty. Clear the cache with a glass.",
+            "You are in the mid-game. Neither dehydrated nor fully hydrated. Take a sip to secure the win.",
+            "Organ systems running at 60% capacity. You're writing code, but it's probably missing tests. Drink water!",
+            "Consistency check: Passing. Keep this steady flow to prevent the afternoon crash."
+        ]
+        quote = random.choice(quotes)
+    elif ratio < 1.00:
+        art = r"""
+  .──────────────────.
+  │ $ make hydrate   │
+  │ [█████████░░]    │
+  │  ✓ BUILD: OK     │
+  │  ✓ 0 warnings    │
+  │  ✓ still alive   │
+  '──────────────────'
+        """
+        quotes = [
+            "Smooth swimming! 85HP. High focus, fast reflexes. You are gliding through code like a dolphin.",
+            "85HP. Excellent latency. Your brain-computer interface is fully lubricated. Keep it going!",
+            "Organ performance: Optimal. You're close to achieving the God Buff. One more sip!",
+            "Gliding through issues and pull requests. Hydration ratio: Excellent. You're styling on them.",
+            "Your organs are running in liquid-cooled mode. Peak performance detected.",
+            "You are in the zone. Clean focus, hydrated joints. Keep it up and hit the goal."
+        ]
+        quote = random.choice(quotes)
+    else:
+        art = r"""
+      ⚡  ⚡  ⚡
+      .───────.
+     /  [MAX]  \
+    |  BUFFED!! |
+     \_________/
+   >> GOD MODE ON
+        """
+        quotes = [
+            "MAX HEALTH achieved! Transcended. Water deity status. The fluid gods bow to you!",
+            "100HP. Full shield. God Mode activated. You can compile C++ with your bare hands now.",
+            "You are literally composed of 99% pure hydration. Even your code is running in liquid cooling.",
+            "Water God status. Your kidneys have reached Nirvana. Go code something impossible!",
+            "Target unlocked. Hydration database fully synced. You are peak human right now."
+        ]
+        quote = random.choice(quotes)
+        
+    return art.strip("\n"), quote
+
+