main.py

# main.py
from prompts import ask_choice, ask_multi_states, ask_multi_jobtypes, ask_int
from writers import open_csv_writer

from scrapers.seek_scraper import scrape_seek
from scrapers.indeed_scraper import scrape_indeed
from scrapers.jora_scraper import scrape_jora

def main():
    print("=== JOB SCRAPING MACHINE (Integrated System) ===")

    platform = ask_choice(
        "Mau scraping platform apa?",
        choices=["seek", "indeed", "jora"],
        allow_all=True
    )

    keyword = input("\nMasukkan keyword/job title (contoh: Software Engineer): ").strip()
    if not keyword:
        print("❌ Keyword kosong. Batal.")
        return

    chosen_states = ask_multi_states()      # [] berarti ALL
    chosen_job_types = ask_multi_jobtypes() # [] berarti ALL

    seek_pages = ask_int("\nMax pages SEEK", 1)
    indeed_pages = ask_int("Pages per state INDEED", 1)
    jora_pages = ask_int("Max pages JORA", 5)

    out = input("\nNama output CSV (default jobs_all_platforms.csv): ").strip() or "jobs_all_platforms.csv"
    f, writer = open_csv_writer(out)

    try:
        if platform in ["all", "seek"]:
            scrape_seek(keyword, seek_pages, chosen_states, chosen_job_types, writer)

        if platform in ["all", "indeed"]:
            scrape_indeed(keyword, indeed_pages, chosen_states, chosen_job_types, writer)

        if platform in ["all", "jora"]:
            scrape_jora(keyword, jora_pages, chosen_states, chosen_job_types, writer)

    finally:
        f.close()

    print(f"\n✅ Done. Output: {out}")

if __name__ == "__main__":
    main()