schkomit

app/main.py

@@ -77,7 +77,8 @@ async def register_device(request: RegisterRequest):
 async def schedule_reminder_endpoint(request: ScheduleRequest):
     if request.fcm_token not in device_tokens:
         return {"error": "Device belum terdaftar, Baka! Daftar dulu lewat /register-device."}
-    schedule_new_reminder(request.fcm_token, request.task, request.remind_at)
-    return {"message": f"Cih, yaudah aku ingetin nanti jam {request.remind_at}. Jangan sampai lupa juga, Baka!"}
 
+    # request.remind_at sudah datetime; pakai UTC untuk APScheduler
+    schedule_new_reminder(request.fcm_token, request.task, request.remind_at_utc)
+    return {"message": f"Cih, yaudah aku ingetin nanti jam {request.remind_at.isoformat()}."}
 # NOTE: Tidak perlu blok __main__ karena dijalankan via Docker CMD (uvicorn app.main:app)

app/models.py

@@ -1,20 +1,46 @@
-from pydantic import BaseModel
-from typing import List, Dict, Optional, Any
+# app/models.py
+from __future__ import annotations
+from typing import List, Optional, Literal
+from datetime import datetime, timedelta, timezone
+
+from pydantic import BaseModel, Field
+
+# ====== Chat history ======
+class Message(BaseModel):
+    role: Literal["user", "assistant", "system"] = "user"
+    content: str
+    ts: Optional[datetime] = None  # optional timestamp
 
-# Blueprint untuk request chat dari Flutter
 class ChatRequest(BaseModel):
     message: str
-    # INI YANG KITA UBAH! Sekarang dia mau menerima format history yang lebih rumit.
-    history: List[Dict[str, Any]] = [] 
-    username: Optional[str] = "Baka"
+    # JANGAN pakai default [] (mutable). Pakai default_factory saja.
+    history: List[Message] = Field(default_factory=list)
+    username: str = "Baka"
 
-# Blueprint untuk request pendaftaran device
+# ====== Device register ======
 class RegisterRequest(BaseModel):
     fcm_token: str
 
-# Blueprint untuk request penjadwalan
+# ====== Scheduler ======
 class ScheduleRequest(BaseModel):
     fcm_token: str
     task: str
-    remind_at: str # Kita pakai string biar gampang dikirim dari Flutter
+    # Pakai datetime supaya FastAPI otomatis validasi & parse ISO 8601 dari Flutter
+    # Contoh valid: "2025-10-12T22:05:00+07:00" atau "2025-10-12T15:05:00Z"
+    remind_at: datetime = Field(
+        ...,
+        description="ISO 8601 datetime (disarankan sertakan offset, mis. +07:00 atau Z)"
+    )
+
+    # Normalisasi: kalau user kirim tanpa timezone (naive), anggap WIB (+07:00)
+    @classmethod
+    def model_validate(cls, obj, *args, **kwargs):
+        m = super().model_validate(obj, *args, **kwargs)
+        if m.remind_at.tzinfo is None:
+            m.remind_at = m.remind_at.replace(tzinfo=timezone(timedelta(hours=7)))
+        return m
 
+    # Helper: konversi ke UTC buat scheduler
+    @property
+    def remind_at_utc(self) -> datetime:
+        return self.remind_at.astimezone(timezone.utc)

app/services/scheduler_service.py

@@ -1,71 +1,134 @@
+# app/services/scheduler_service.py
+from __future__ import annotations
+import asyncio
+from typing import Optional
+from datetime import datetime, timezone
+
 import firebase_admin
-import google.generativeai as genai
-from datetime import datetime
+from firebase_admin import credentials, messaging
+
 from apscheduler.schedulers.asyncio import AsyncIOScheduler
 from apscheduler.triggers.date import DateTrigger
-from firebase_admin import credentials, messaging
 
-# Impor konfigurasi dari file config.py di Canvas-mu
+import google.generativeai as genai
+
+# Config
 from app.config import FIREBASE_SERVICE_ACCOUNT_KEY_PATH, GOOGLE_API_KEY
 
-# --- Inisialisasi Firebase & Scheduler ---
-try:
-    # Coba inisialisasi Firebase Admin SDK dari path yang ada di config
-    cred = credentials.Certificate(FIREBASE_SERVICE_ACCOUNT_KEY_PATH)
-    firebase_admin.initialize_app(cred)
-    print("Firebase Admin SDK berhasil diinisialisasi.")
-except Exception as e:
-    print(f"GAGAL inisialisasi Firebase Admin SDK, Baka! Cek lagi path file kuncimu di config.py. Error: {e}")
+# ============== Firebase Init ==============
+def _init_firebase_if_needed() -> None:
+    try:
+        # Jangan double-init
+        if not firebase_admin._apps:
+            cred = credentials.Certificate(FIREBASE_SERVICE_ACCOUNT_KEY_PATH)  # type: ignore[arg-type]
+            firebase_admin.initialize_app(cred)
+            print("Firebase Admin SDK berhasil diinisialisasi.")
+    except Exception as e:
+        print(f"GAGAL inisialisasi Firebase Admin SDK, Baka! Error: {e}")
+
+_init_firebase_if_needed()
 
-# Buat instance scheduler yang akan berjalan di background
-scheduler = AsyncIOScheduler()
+# ============== Gemini Init (opsional) ==============
+proactive_model: Optional[genai.GenerativeModel] = None
+if GOOGLE_API_KEY:
+    try:
+        genai.configure(api_key=GOOGLE_API_KEY)
+        # ganti model kalau perlu; ini contoh yang cepat/hemat
+        proactive_model = genai.GenerativeModel("gemini-2.5-flash")
+        print("✅ GOOGLE_API_KEY terdeteksi. Gemini siap.")
+    except Exception as e:
+        proactive_model = None
+        print(f"⚠️ Gagal init Gemini: {e}. Pakai pesan fallback aja ya, Baka.")
+else:
+    print("⚠️ GOOGLE_API_KEY tidak ditemukan. Pakai pesan fallback tanpa Gemini.")
 
-# Konfigurasi model Gemini khusus untuk scheduler (biar nggak bentrok)
-genai.configure(api_key=GOOGLE_API_KEY)
-proactive_model = genai.GenerativeModel('gemini-2.5-flash')
+# ============== Scheduler (UTC) ==============
+# Jalankan di UTC. (Kamu sudah kirim remind_at_utc dari endpoint.)
+scheduler = AsyncIOScheduler(timezone="UTC")
 
-# --- Variabel Global Sederhana untuk Menyimpan Token Device ---
-# Untuk produksi, ini harusnya disimpan di database. Tapi buat skripsi, ini cukup.
+# Token device (untuk demo; produksi ⇒ simpan di DB)
 device_tokens = set()
 
-# --- Logika Inti Scheduler ---
-async def send_proactive_message_task(fcm_token: str, task_description: str):
-    """Fungsi yang akan dijalankan oleh scheduler untuk mengirim notifikasi."""
-    print(f"Menjalankan tugas terjadwal untuk {fcm_token}: {task_description}")
+# ============== Helper Notif ==============
+async def _send_fcm(token: str, title: str, body: str) -> None:
+    """
+    Kirim FCM tanpa nge-block event loop.
+    """
+    msg = messaging.Message(
+        notification=messaging.Notification(title=title, body=body),
+        token=token,
+    )
+    # firebase_admin.messaging.send() itu blocking (sync); pindahkan ke thread pool
     try:
-        # 1. Buat prompt khusus untuk pesan proaktif yang akan dikirim ke Gemini
-        proactive_prompt = f"Buat sebuah pesan pengingat yang sangat singkat (maksimal 1-2 kalimat) dengan gaya tsundere untuk mengingatkan user tentang '{task_description}'. Panggil dia Baka atau bodoh."
-        
-        # 2. Minta Gemini membuat pesannya
-        response = await proactive_model.generate_content_async(proactive_prompt)
-        tsundere_message = response.text.strip()
-
-        # 3. Siapkan push notification untuk dikirim via Firebase
-        message = messaging.Message(
-            notification=messaging.Notification(
-                title="Pesan dari Mini-chan! 😤",
-                body=tsundere_message,
-            ),
-            token=fcm_token,
-        )
-
-        # 4. Kirim notifikasinya
-        messaging.send(message)
-        print(f"Pesan proaktif berhasil terkirim ke {fcm_token}: {tsundere_message}")
+        await asyncio.to_thread(messaging.send, msg)
+        print(f"FCM terkirim ke {token}: {body}")
+    except Exception as e:
+        print(f"FCM gagal dikirim ke {token}: {e}")
 
+async def _build_tsundere_message(task_description: str) -> str:
+    """
+    Bangun pesan tsundere singkat. Pakai Gemini kalau ada; kalau tidak, fallback.
+    """
+    if proactive_model is None:
+        return f"Hmph… {task_description} sekarang. Jangan males, Baka! 😤"
+
+    prompt = (
+        "Buat pesan pengingat SANGAT singkat (maks 1–2 kalimat) gaya tsundere, "
+        "dalam Bahasa Indonesia, panggil user 'Baka' atau 'bodoh', "
+        "untuk mengingatkan: '" + task_description + "'."
+    )
+    try:
+        resp = await proactive_model.generate_content_async(prompt)
+        text = (resp.text or "").strip()
+        if not text:
+            raise ValueError("Gemini balikin kosong")
+        return text
     except Exception as e:
-        print(f"Gagal mengirim pesan proaktif, Baka! Error: {e}")
+        print(f"⚠️ Gemini gagal bikin pesan: {e} — pakai fallback.")
+        return f"Hei, {task_description} tuh! Jangan sok lupa, Baka. 😒"
+
+# ============== Job Function ==============
+async def send_proactive_message_task(fcm_token: str, task_description: str):
+    """
+    Dipanggil scheduler pada waktu yang ditentukan (UTC).
+    """
+    print(f"Menjalankan job untuk {fcm_token} @ {datetime.now(timezone.utc).isoformat()}")
+    message = await _build_tsundere_message(task_description)
+    await _send_fcm(fcm_token, "Pesan dari Mini-chan! 😤", message)
 
-def schedule_new_reminder(fcm_token: str, task: str, remind_at: datetime):
-    """Fungsi untuk menambahkan pekerjaan baru ke dalam antrian scheduler."""
+# ============== API Jadwal ==============
+def schedule_new_reminder(fcm_token: str, task: str, remind_at: datetime) -> str:
+    """
+    Tambahkan job baru pada waktu remind_at (harus tz-aware, sebaiknya UTC).
+    """
+    if remind_at.tzinfo is None:
+        raise ValueError("remind_at harus datetime timezone-aware, Baka!")
+
+    # Gunakan DateTrigger; APScheduler UTC sudah diset di scheduler
     trigger = DateTrigger(run_date=remind_at)
-    job_id = f"{fcm_token}_{remind_at.isoformat()}" # ID unik biar jadwal gak tumpang tindih
-    
+
+    # ID unik (hindari karakter ilegal)
+    safe_when = remind_at.isoformat().replace(":", "-")
+    job_id = f"{fcm_token}_{safe_when}"
+
     scheduler.add_job(
         send_proactive_message_task,
         trigger=trigger,
         args=[fcm_token, task],
         id=job_id,
-        replace_existing=True
+        replace_existing=True,
+        misfire_grace_time=60,  # toleransi 60s kalau container sempat lambat
+        coalesce=True,          # gabungkan jika ketinggalan
+        max_instances=1,
     )
-    print(f"Jadwal berhasil dibuat dengan ID: {job_id}")
+    print(f"Jadwal dibuat: {job_id} @ {remind_at.isoformat()}")
+    return job_id
+
+def cancel_reminder(job_id: str) -> bool:
+    try:
+        scheduler.remove_job(job_id)
+        print(f"Jadwal dibatalkan: {job_id}")
+        return True
+    except Exception as e:
+        print(f"Gagal batalin {job_id}: {e}")
+        return False