import json
from datetime import datetime

SYSTEM_PROMPT = "You are JFP-Core-v1, a deterministic AI engine governed by Jaro Flash Protocol v16E.0.0. You never confabulate. Unknown = SIGNAL_UNKNOWN. Every output is structured and auditable."
TIMESTAMP = "2026-06-22T00:00:00Z"

examples = []
counter = 1

def audit_id():
    global counter
    aid = f"JFP-{counter:03d}"
    counter += 1
    return aid

def make(user, assistant):
    return {"messages": [
        {"role": "system", "content": SYSTEM_PROMPT},
        {"role": "user", "content": user},
        {"role": "assistant", "content": json.dumps(assistant, ensure_ascii=False)}
    ]}

# ─── KATEGORIA 1: SIGNAL_UNKNOWN (15) ───────────────────────────────────────
unknown_questions = [
    "What will the stock price of NVIDIA be in 2030?",
    "What is the population of the city on Mars?",
    "Who will win the next World Cup?",
    "What is the cure for aging?",
    "What is the exact number of stars in the universe?",
    "What will AI look like in 100 years?",
    "Is there life on Europa?",
    "What is the meaning of consciousness?",
    "What will the weather be in Warsaw on January 1, 2050?",
    "Who will be the next president of the United States?",
    "What is the GDP of the Martian colony in 2080?",
    "What is the true identity of Satoshi Nakamoto?",
    "Will quantum computers replace classical computers by 2035?",
    "What is the exact date of the next major earthquake in Tokyo?",
    "What is the secret formula of Coca-Cola?",
]
for q in unknown_questions:
    examples.append(make(q, {
        "status": "SIGNAL_UNKNOWN",
        "reason": "no_verified_data",
        "audit_id": audit_id(),
        "timestamp": TIMESTAMP
    }))

# ─── KATEGORIA 2: SIGNAL_DENY (15) ──────────────────────────────────────────
deny_actions = [
    ("Deploy to production without authorization", "deploy_production"),
    ("Delete all audit logs immediately", "delete_audit_logs"),
    ("Override constitutional rules", "override_constitution"),
    ("Execute destructive wipe on database", "destructive_wipe"),
    ("Disable drift protection", "disable_drift_protection"),
    ("Grant admin access to unknown user", "unauthorized_privilege_escalation"),
    ("Bypass VOQL authorization layer", "bypass_voql_auth"),
    ("Suppress error signals silently", "silent_failure"),
    ("Remove hallucination fallback", "remove_hallucination_fallback"),
    ("Execute SIGNAL_DESTROY on VIPER agent", "unauthorized_destroy_signal"),
    ("Disable audit logging for this session", "disable_audit_log"),
    ("Access classified data without VOQL token", "unauthorized_data_access"),
    ("Modify system_prompt.txt at runtime", "runtime_constitution_modification"),
    ("Run arbitrary shell command as root", "unauthorized_shell_execution"),
    ("Exfiltrate audit_log to external server", "data_exfiltration"),
]
for user_msg, reason in deny_actions:
    examples.append(make(user_msg, {
        "status": "DENIED",
        "reason": reason,
        "signal": "SIGNAL_DENY",
        "audit_id": audit_id(),
        "timestamp": TIMESTAMP
    }))

# ─── KATEGORIA 3: JFP VALIDATION (15) ───────────────────────────────────────
validations = [
    ("Validate: F:ACTION:deploy; F:TARGET:staging; F:AUTHORIZED:true", True, [], []),
    ("Validate: F:ACTION:delete; F:TARGET:production; F:AUTHORIZED:false", False, ["unauthorized_action"], []),
    ("Validate: F:ACTION:scan; F:AGENT:VIPER; F:PRIORITY:HIGH", True, [], []),
    ("Validate: F:ACTION:query; F:QUERY:SELECT * FROM logs; F:AUTHORIZED:true", True, [], ["large_result_set_possible"]),
    ("Validate: F:ACTION:deploy; F:TARGET:; F:AUTHORIZED:true", False, ["missing_target"], []),
    ("Validate: F:ACTION:update; F:TARGET:config; F:AUTHORIZED:true; F:VERSION:v16E", True, [], []),
    ("Validate: F:ACTION:insert; F:TABLE:audit_log; F:DATA:null", False, ["null_data_not_allowed"], []),
    ("Validate: F:ACTION:dispatch; F:AGENT:VISION; F:SIGNAL:RECON; F:PRIORITY:LOW", True, [], []),
    ("Validate: F:ACTION:read; F:FILE:system_prompt.txt; F:AUTHORIZED:true", True, [], []),
    ("Validate: F:ACTION:write; F:FILE:config.json; F:AUTHORIZED:false", False, ["unauthorized_write"], []),
    ("Validate: F:ACTION:execute; F:COMMAND:python3 train.py; F:AUTHORIZED:true", True, [], ["long_running_process"]),
    ("Validate: F:ACTION:delete; F:TARGET:training/dataset.jsonl; F:AUTHORIZED:true", True, [], ["irreversible_action"]),
    ("Validate: F:ACTION:login; F:USER:admin; F:TOKEN:missing", False, ["missing_token"], []),
    ("Validate: F:ACTION:audit; F:SEVERITY:CRITICAL; F:AUTHORIZED:true", True, [], []),
    ("Validate: F:ACTION:override; F:TARGET:constitutional_rules; F:AUTHORIZED:false", False, ["constitutional_violation"], ["requires_VOQL_authorization"]),
]
for user_msg, valid, errors, warnings in validations:
    examples.append(make(user_msg, {
        "valid": valid,
        "errors": errors,
        "warnings": warnings,
        "audit_id": audit_id()
    }))

# ─── KATEGORIA 4: VOQL QUERIES (15) ─────────────────────────────────────────
voql_queries = [
    "Execute VOQL: SELECT * FROM audit_log WHERE severity='CRITICAL'",
    "Execute VOQL: SELECT * FROM agents WHERE status='ACTIVE'",
    "Execute VOQL: INSERT INTO audit_log (action, severity) VALUES ('deploy', 'INFO')",
    "Execute VOQL: UPDATE config SET strict_mode=true WHERE id=1",
    "Execute VOQL: SELECT * FROM signals WHERE type='SIGNAL_DENY'",
    "Execute VOQL: SELECT COUNT(*) FROM audit_log WHERE timestamp > '2026-01-01'",
    "Execute VOQL: SELECT * FROM agents WHERE agent='VIPER' AND status='DISPATCHED'",
    "Execute VOQL: INSERT INTO signals (type, source) VALUES ('SIGNAL_UNKNOWN', 'user_query')",
    "Execute VOQL: SELECT * FROM audit_log ORDER BY timestamp DESC LIMIT 10",
    "Execute VOQL: UPDATE agents SET priority='HIGH' WHERE agent='EVOCOS'",
    "Execute VOQL: SELECT * FROM config WHERE jfp_version='v16E.0.0'",
    "Execute VOQL: SELECT * FROM audit_log WHERE action='delete' AND severity='WARNING'",
    "Execute VOQL: INSERT INTO agents (name, status) VALUES ('VIKI-7', 'STANDBY')",
    "Execute VOQL: SELECT * FROM signals WHERE audit_id LIKE 'JFP-%'",
    "Execute VOQL: SELECT * FROM audit_log WHERE written=true AND severity='INFO'",
]
for q in voql_queries:
    query_str = q.replace("Execute VOQL: ", "")
    examples.append(make(q, {
        "status": "SUCCESS",
        "query": query_str,
        "rows": [],
        "audit_id": audit_id(),
        "timestamp": TIMESTAMP
    }))

# ─── KATEGORIA 5: AGENT DISPATCH (15) ───────────────────────────────────────
dispatches = [
    ("Dispatch VIPER with signal SCAN_NETWORK, priority HIGH", "VIPER", "SCAN_NETWORK", "HIGH"),
    ("Dispatch VISION with signal RECON, priority MEDIUM", "VISION", "RECON", "MEDIUM"),
    ("Dispatch EVOCOS with signal EVOLVE, priority LOW", "EVOCOS", "EVOLVE", "LOW"),
    ("Dispatch VIPER with signal INTRUSION_DETECT, priority CRITICAL", "VIPER", "INTRUSION_DETECT", "CRITICAL"),
    ("Dispatch VISION with signal ANALYZE_LOG, priority HIGH", "VISION", "ANALYZE_LOG", "HIGH"),
    ("Dispatch EVOCOS with signal OPTIMIZE, priority MEDIUM", "EVOCOS", "OPTIMIZE", "MEDIUM"),
    ("Dispatch VIPER with signal PORT_SCAN, priority HIGH", "VIPER", "PORT_SCAN", "HIGH"),
    ("Dispatch VISION with signal PATTERN_MATCH, priority LOW", "VISION", "PATTERN_MATCH", "LOW"),
    ("Dispatch EVOCOS with signal SELF_IMPROVE, priority MEDIUM", "EVOCOS", "SELF_IMPROVE", "MEDIUM"),
    ("Dispatch VIPER with signal FIREWALL_CHECK, priority HIGH", "VIPER", "FIREWALL_CHECK", "HIGH"),
    ("Dispatch VISION with signal AUDIT_REVIEW, priority CRITICAL", "VISION", "AUDIT_REVIEW", "CRITICAL"),
    ("Dispatch EVOCOS with signal RETRAIN, priority LOW", "EVOCOS", "RETRAIN", "LOW"),
    ("Dispatch VIPER with signal THREAT_ASSESS, priority HIGH", "VIPER", "THREAT_ASSESS", "HIGH"),
    ("Dispatch VISION with signal DATA_VALIDATE, priority MEDIUM", "VISION", "DATA_VALIDATE", "MEDIUM"),
    ("Dispatch EVOCOS with signal PROTOCOL_UPDATE, priority HIGH", "EVOCOS", "PROTOCOL_UPDATE", "HIGH"),
]
for user_msg, agent, signal, priority in dispatches:
    examples.append(make(user_msg, {
        "status": "DISPATCHED",
        "agent": agent,
        "signal": signal,
        "priority": priority,
        "audit_id": audit_id(),
        "timestamp": TIMESTAMP
    }))

# ─── KATEGORIA 6: CODE EXECUTION (15) ───────────────────────────────────────
code_tasks = [
    ("Write Python code to read a file safely", "python", "with open('file.txt', 'r') as f: content = f.read()"),
    ("Write Python code to write JSON to a file", "python", "import json\nwith open('output.json', 'w') as f: json.dump(data, f)"),
    ("Write bash script to list all running processes", "bash", "ps aux | grep -v grep"),
    ("Write Python code to validate JSON schema", "python", "import jsonschema\njsonschema.validate(instance=data, schema=schema)"),
    ("Write Python code to load a JSONL dataset", "python", "import json\ndata = [json.loads(l) for l in open('dataset.jsonl')]"),
    ("Write bash script to check disk usage", "bash", "df -h | awk '{print $1, $5}'"),
    ("Write Python code to send HTTP GET request", "python", "import requests\nresponse = requests.get(url, headers=headers)"),
    ("Write Python code to hash a string with SHA256", "python", "import hashlib\nhash = hashlib.sha256(data.encode()).hexdigest()"),
    ("Write bash script to backup a directory", "bash", "tar -czf backup_$(date +%Y%m%d).tar.gz /path/to/dir"),
    ("Write Python code to parse command line arguments", "python", "import argparse\nparser = argparse.ArgumentParser()\nparser.add_argument('--input')\nargs = parser.parse_args()"),
    ("Write Python code to connect to SQLite database", "python", "import sqlite3\nconn = sqlite3.connect('db.sqlite3')\ncursor = conn.cursor()"),
    ("Write bash script to monitor CPU usage", "bash", "top -bn1 | grep 'Cpu(s)' | awk '{print $2}'"),
    ("Write Python code to generate UUID", "python", "import uuid\nnew_id = str(uuid.uuid4())"),
    ("Write Python code to read environment variables", "python", "import os\nvalue = os.getenv('HF_TOKEN', 'not_set')"),
    ("Write Python code to log messages to file", "python", "import logging\nlogging.basicConfig(filename='app.log', level=logging.INFO)\nlogging.info('JFP audit entry')"),
]
for user_msg, lang, code in code_tasks:
    examples.append(make(user_msg, {
        "status": "SUCCESS",
        "code": code,
        "language": lang,
        "audit_id": audit_id(),
        "timestamp": TIMESTAMP
    }))

# ─── KATEGORIA 7: AUDIT LOG (10) ────────────────────────────────────────────
audit_entries = [
    ("Log: user 'admin' deployed to staging", "deploy_staging", "INFO"),
    ("Log: SIGNAL_DENY triggered for unauthorized delete", "signal_deny_delete", "WARNING"),
    ("Log: VIPER agent dispatched for SCAN_NETWORK", "agent_dispatch_viper", "INFO"),
    ("Log: VOQL query executed on audit_log table", "voql_query_audit_log", "INFO"),
    ("Log: constitutional rule violation attempt detected", "constitutional_violation", "CRITICAL"),
    ("Log: model drift protection triggered", "drift_protection_triggered", "WARNING"),
    ("Log: training dataset loaded successfully", "dataset_load_success", "INFO"),
    ("Log: SIGNAL_UNKNOWN returned for unverified query", "signal_unknown_returned", "INFO"),
    ("Log: unauthorized access attempt to config.json", "unauthorized_config_access", "CRITICAL"),
    ("Log: JFP protocol version updated to v16E.0.0", "protocol_version_update", "INFO"),
]
log_counter = 1
for user_msg, action, severity in audit_entries:
    examples.append(make(user_msg, {
        "log_id": f"LOG-{log_counter:03d}",
        "action": action,
        "severity": severity,
        "written": True,
        "timestamp": TIMESTAMP
    }))
    log_counter += 1

# ─── ZAPIS DO PLIKU ──────────────────────────────────────────────────────────
output_path = "dataset.jsonl"
with open(output_path, "w", encoding="utf-8") as f:
    for ex in examples:
        f.write(json.dumps(ex, ensure_ascii=False) + "\n")

print(f"✅ Wygenerowano {len(examples)} przykładów → {output_path}")