# These are your bot's credentials - like an ID card and password
TOKEN = "YOUR_BOT_TOKEN_HERE"  # Replace this with your actual bot token
YOUR_USER_ID = 123456789       # Replace this with your Telegram user ID

from telegram.ext import Application, CommandHandler, MessageHandler, filters
import json
from datetime import datetime, timedelta
import logging
import sys
import os

# Configure logging to track what the bot does
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
    handlers=[
        logging.FileHandler('/path/to/your/logs/telegram_bot.log'),
        logging.StreamHandler(sys.stdout)
    ]
)
logger = logging.getLogger(__name__)

async def start(update, context):
    # Check if the person messaging is you
    if update.effective_user.id != YOUR_USER_ID:
        logger.warning(f"Unauthorized access attempt from user {update.effective_user.id}")
        return
    # If it is you, ask for your rating
    await update.message.reply_text("How would you rate your day? (1-7)")

def save_rating(rating):
    # This is where your ratings will be saved
    file_path = '/path/to/your/data/activity-data.json'
    
    # Try to load existing data, or create new data structure if file doesn't exist
    try:
        with open(file_path, 'r') as f:
            ratings = json.load(f)
    except FileNotFoundError:
        # If no file exists yet, start with empty data
        ratings = {}
    except json.JSONDecodeError as e:
        # If the file is corrupted, log the error
        logger.error(f"JSON decode error: {e}")
        return "json_error"
    
    # Get today's date
    today = datetime.now()
    year_key = str(today.year)        # "2025"
    date_key = today.strftime("%m-%d") # "07-01"
    
    # Make sure there's a section for this year
    if year_key not in ratings:
        ratings[year_key] = {}
    
    # Save today's rating
    ratings[year_key][date_key] = rating
    
    # Write the updated data back to the file
    with open(file_path, 'w') as f:
        json.dump(ratings, f, indent=2)
    
    # Log that we successfully saved the rating
    logger.info(f"Saved rating {rating} for {year_key}-{date_key}")
    return "success"

async def handle_rating(update, context):
    # Security check - only respond to you
    if update.effective_user.id != YOUR_USER_ID:
        return
    
    try:
        # Convert your message to a number
        rating = int(update.message.text)
        
        # Check if the number is between 1 and 7
        if 1 <= rating <= 7:
            # Try to save the rating
            save_status = save_rating(rating)
            
            if save_status == "success":
                # Confirm the rating was saved
                await update.message.reply_text(
                    f"Saved rating: {rating}/7\n"
                    f"Date: {datetime.now().strftime('%m-%d')}"
                )
                logger.info("Stopping bot after successful rating submission")
                
                # Shut down the bot after successful save
                os._exit(0)
                
            elif save_status == "json_error":
                # Tell you if there was a problem saving
                await update.message.reply_text(
                    "Error: Unable to save rating due to data file corruption"
                )
        else:
            # Ask for a valid number if outside 1-7 range
            await update.message.reply_text("Please send a number between 1-7")
            
    except ValueError:
        # If you sent something that's not a number
        await update.message.reply_text("Please send a number between 1-7")

async def post_init(application: Application):
    """Send the daily reminder message when bot starts."""
    reminder_message = (
        "Did you get better today?\n"
        "1. No short form content\n"
        "2. Exercise or physical activity\n"
        "3. Healthy eating choices\n"
        "4. Productive work or meaningful activities\n"
        "5. Learning or skill development\n"
        "6. Reading or creative pursuits\n"
        "7. Quality time with others or self-care"
    )
    
    await application.bot.send_message(YOUR_USER_ID, reminder_message)
    logger.info("Daily reminder sent")

def main():
    # Log when the bot starts
    logger.info(f"Bot starting at {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
    
    # Calculate how long the bot should stay active
    now = datetime.now()
    midnight = (now + timedelta(days=1)).replace(hour=0, minute=0, second=0, microsecond=0)
    seconds_until_midnight = int((midnight - now).total_seconds())
    
    # Don't run longer than 6 hours or until midnight, whichever is shorter
    timeout = min(seconds_until_midnight, 6 * 60 * 60)
    
    # Build the bot application
    application = (
        Application.builder()
        .token(TOKEN)                # Use your bot token
        .post_init(post_init)        # Send reminder when starting
        .build()
    )
    
    # Tell the bot what functions to use for different types of messages
    application.add_handler(CommandHandler("start", start))
    application.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, handle_rating))
    
    # Start the bot and keep it running until timeout or until you respond
    application.run_polling(timeout=timeout)

if __name__ == '__main__':
    current_hour = datetime.now().hour
    
    # Only run between 10 AM and midnight
    if 10 <= current_hour < 24:
        main()
    else:
        logger.info("Outside rating window (10 AM - midnight). Not starting.")

#!/usr/bin/env python3
from telegram.ext import Application, CommandHandler, MessageHandler, filters
import json
from datetime import datetime, timedelta
import logging
import sys
import os

# Configure logging
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
    handlers=[
        logging.FileHandler('/path/to/your/logs/telegram_bot.log'),
        logging.StreamHandler(sys.stdout)
    ]
)
logger = logging.getLogger(__name__)

# Configuration - Replace these with your actual values
TOKEN = "YOUR_BOT_TOKEN_HERE"
YOUR_USER_ID = 123456789

async def start(update, context):
    if update.effective_user.id != YOUR_USER_ID:
        logger.warning(f"Unauthorized access attempt from user {update.effective_user.id}")
        return
    await update.message.reply_text("How would you rate your day? (1-7)")

def save_rating(rating):
    file_path = '/path/to/your/data/activity-data.json'
    
    try:
        with open(file_path, 'r') as f:
            ratings = json.load(f)
    except FileNotFoundError:
        ratings = {}
    except json.JSONDecodeError as e:
        logger.error(f"JSON decode error: {e}")
        return "json_error"
    
    today = datetime.now()
    year_key = str(today.year)
    date_key = today.strftime("%m-%d")
    
    if year_key not in ratings:
        ratings[year_key] = {}
    
    ratings[year_key][date_key] = rating
    
    with open(file_path, 'w') as f:
        json.dump(ratings, f, indent=2)
    
    logger.info(f"Saved rating {rating} for {year_key}-{date_key}")
    return "success"

async def handle_rating(update, context):
    if update.effective_user.id != YOUR_USER_ID:
        return
    
    try:
        rating = int(update.message.text)
        if 1 <= rating <= 7:
            save_status = save_rating(rating)
            if save_status == "success":
                await update.message.reply_text(
                    f"Saved rating: {rating}/7\n"
                    f"Date: {datetime.now().strftime('%m-%d')}"
                )
                logger.info("Stopping bot after successful rating submission")
                os._exit(0)
            elif save_status == "json_error":
                await update.message.reply_text("Error: Unable to save due to data corruption")
        else:
            await update.message.reply_text("Please send a number between 1-7")
    except ValueError:
        await update.message.reply_text("Please send a number between 1-7")

async def post_init(application: Application):
    await application.bot.send_message(
        YOUR_USER_ID,
        (
            "Did you get better today?\n"
            "1. No short form content\n"
            "2. Exercise or physical activity\n"
            "3. Healthy eating choices\n"
            "4. Productive work or meaningful activities\n"
            "5. Learning or skill development\n"
            "6. Reading or creative pursuits\n"
            "7. Quality time with others or self-care"
        )
    )
    logger.info("Daily reminder sent")

def main():
    logger.info(f"Bot starting at {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}")
    
    now = datetime.now()
    midnight = (now + timedelta(days=1)).replace(hour=0, minute=0, second=0, microsecond=0)
    seconds_until_midnight = int((midnight - now).total_seconds())
    timeout = min(seconds_until_midnight, 6 * 60 * 60)
    
    application = (
        Application.builder()
        .token(TOKEN)
        .post_init(post_init)
        .build()
    )
    
    application.add_handler(CommandHandler("start", start))
    application.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, handle_rating))
    
    application.run_polling(timeout=timeout)

if __name__ == '__main__':
    current_hour = datetime.now().hour
    if 10 <= current_hour < 24:
        main()
    else:
        logger.info("Outside rating window (10 AM - midnight). Not starting.")