app.py

#!/usr/bin/env python3
import requests
import os
from datetime import datetime, timedelta
import sys
from zoneinfo import ZoneInfo, ZoneInfoNotFoundError
try:
    from version import get_version
except ImportError as e:
    print(f"Warning: Could not import version module: {e}")
    def get_version():
        return "unknown"


# Basis-URLs
env_name = 'idleoutpostclaimer'
BASE_STORE = 'https://store.xsolla.com'
USERID_SERVICE = 'https://sb-user-id-service.xsolla.com/api/v1/user-id'
LOG_FILE = 'claim_rewards.log'

# Zeitzone konfigurieren
def get_timezone():
    tz_str = os.environ.get('TZ', 'Europe/Berlin')
    try:
        return ZoneInfo(tz_str)
    except ZoneInfoNotFoundError:
        log(f"Warnung: Zeitzone '{tz_str}' nicht gefunden. Fallback auf UTC.")
        return ZoneInfo("UTC")

TIMEZONE = get_timezone()

# Konfiguriere hier deine Game-ID
USER_GAME_ID = os.environ.get('USER_GAME_ID')

# Payload für den User-ID-Service
USER_ID_PAYLOAD = {
    "settings": {"projectId":256000, "merchantId":329415},
    "loginId":"048e3522-75bd-43f5-95da-6ec145b9723a",
    "webhookUrl":"https://vd.appquantum.tech/webhook/user_validation",
    "user": {"id":USER_GAME_ID, "country":"DE"},
    "isUserIdFromWebhook": False
}

# Claim-Endpunkte
ENDPOINTS = {
    'shovels': '/api/v2/project/256000/free/item/com.rockbite.zombieoutpost.webshop.dailyshovels',
    'tickets': '/api/v2/project/256000/free/item/com.rockbite.zombieoutpost.webshop.dailyadtickets',
    'legendary': '/api/v2/project/256000/free/item/com.rockbite.zombieoutpost.webshop.weeklylegendarychest',
    'weekly': '/api/v2/project/256000/free/item/com.rockbite.zombieoutpost.webshop.weeklybonus2',
    'weekly_gems': '/api/v2/project/256000/free/item/com.rockbite.zombieoutpost.webshop.weeklygems'
}

DAILY_REWARDS = ('shovels', 'tickets')
WEEKLY_REWARDS = ('legendary', 'weekly', 'weekly_gems')

REWARD_NAMES = {
    'shovels': 'Shovels',
    'tickets': 'Tickets',
    'legendary': 'Legendary',
    'weekly': 'Weekly',
    'weekly_gems': 'Weekly Gems'
}


def log(message: str):
    """
    Protokolliert eine Nachricht mit Zeitstempel auf der Konsole.
    """
    timestamp = datetime.now(TIMEZONE).strftime("%d.%m.%y-%H:%M")
    line = f"[{timestamp}] {message}"
    print(line)


def setup_session():
    session = requests.Session()
    headers = {
        'Content-Type': 'application/json',
        'Accept': 'application/json',
        'User-Agent': 'Mozilla/5.0'
    }
    # 1) Token vom User-ID-Service holen
    resp = session.post(USERID_SERVICE, json=USER_ID_PAYLOAD, headers=headers)
    resp.raise_for_status()
    token = resp.json().get('token')
    if not token:
        log('❗ Token konnte nicht vom User-ID-Service abgerufen werden.')
        raise RuntimeError('Kein Token erhalten vom User-ID-Service')
    log('🔑 Token erfolgreich erhalten.')
    # 2) Authorization-Header setzen
    session.headers.update({
        'Authorization': f'Bearer {token}',
        'Accept': 'application/json',
        'User-Agent': 'Mozilla/5.0'
    })
    return session


def claim(session, key: str):
    item_name = REWARD_NAMES.get(key, key.capitalize())
    url = BASE_STORE + ENDPOINTS[key]
    log(f"Versuche, '{item_name}' zu claimen...")
    resp = session.post(url, json={})

    if resp.status_code in (200, 204):
        log(f"✅ '{item_name}' erfolgreich geclaimed.")
    elif resp.status_code == 422:
        log(f"🔒 '{item_name}' wurde heute bereits geclaimed (Kauflimit erreicht).")
    else:
        log(f"❗ Fehler beim Claimen von '{item_name}'. Status-Code: {resp.status_code}")
        log(f"   Server-Antwort: {resp.text}")


def claim_all_rewards(session):
    log("\n--- Tägliche Belohnungen ---")
    for key in DAILY_REWARDS:
        claim(session, key)

    log("\n--- Wöchentliche Belohnungen ---")
    for key in WEEKLY_REWARDS:
        claim(session, key)


def show_startup_message():
    now = datetime.now(TIMEZONE)
    # Cron is set to 02:00
    next_run = now.replace(hour=2, minute=0, second=0, microsecond=0)
    if now.hour >= 2:
        next_run += timedelta(days=1)

    time_diff = next_run - now
    hours, remainder = divmod(time_diff.seconds, 3600)
    minutes, _ = divmod(remainder, 60)

    # Get version with error handling
    try:
        version = get_version()
        if version and version.strip():
            log(f"🚀 Idle Outpost Claimer v{version} gestartet.")
        else:
            log("🚀 Idle Outpost Claimer v[version unavailable] gestartet.")
    except Exception as e:
        log(f"🚀 Idle Outpost Claimer v[error: {e}] gestartet.")
    
    log(f"Nächster automatischer Claim um {next_run.strftime('%H:%M')}. Das ist in {hours} Stunden und {minutes} Minuten.")


if __name__ == '__main__':
    if len(sys.argv) > 1 and sys.argv[1] == '--status':
        show_startup_message()
        exit(0)

    if not USER_GAME_ID:
        print("Fehler: Die Umgebungsvariable USER_GAME_ID wurde nicht gesetzt.")
        print("Bitte setze sie und starte den Container neu.")
        exit(1)

    log("⚙️  Führe planmäßigen Claim aus...")
    # Session initialisieren
    sess = setup_session()

    claim_all_rewards(sess)

    log("\n🏁 Alle Aktionen abgeschlossen.")