private class KiterjStat { public long Db; public long Meret; }

private class Eredmeny
{
    public long Fajlok, Mappak, Hibak, OsszMeret;
    public Dictionary<string, KiterjStat> Kiterj = new Dictionary<string, KiterjStat>();
    public List<KeyValuePair<string, long>> Legnagyobbak = new List<KeyValuePair<string, long>>();
    public List<string> HibaSzovegek = new List<string>();
}

private enum Allapot { Kesz, Fut, Szunet, Megszakitas }

private Allapot allapot = Allapot.Kesz;
private CancellationTokenSource forras;
private ManualResetEventSlim szunet = new ManualResetEventSlim(true);
private Stopwatch ora = new Stopwatch();
private bool bezarasra = false;

// a háttérszál és a felület közös adatai
private long szamolt, feldolgozott, osszes, osszMeret;
private volatile int fazis;
private volatile string aktualis = "";

private void Form1_Load(object sender, EventArgs e)
{
    tmrKijelzo.Interval = 200;
    pb.AccessibleName = "Mappa feldolgozása";
    Beallit(Allapot.Kesz);
}

private void Beallit(Allapot a)
{
    allapot = a;

    btnIndit.Enabled = a == Allapot.Kesz;
    btnSzunet.Enabled = a == Allapot.Fut || a == Allapot.Szunet;
    btnMegse.Enabled = a == Allapot.Fut || a == Allapot.Szunet;
    txtMappa.Enabled = a == Allapot.Kesz;
    btnTallozas.Enabled = a == Allapot.Kesz;
    chkAlkonyvtar.Enabled = a == Allapot.Kesz;
    nudIdo.Enabled = a == Allapot.Kesz;
    btnSzunet.Text = a == Allapot.Szunet ? "Folytatás" : "Szünet";
    this.UseWaitCursor = a == Allapot.Fut || a == Allapot.Megszakitas;

    if (a == Allapot.Megszakitas) { lblAllapot.Text = "Megszakítás folyamatban..."; }
}

private void btnTallozas_Click(object sender, EventArgs e)
{
    using (FolderBrowserDialog d = new FolderBrowserDialog())
    {
        if (Directory.Exists(txtMappa.Text)) { d.SelectedPath = txtMappa.Text; }

        if (d.ShowDialog(this) == DialogResult.OK) { txtMappa.Text = d.SelectedPath; }
    }
}

private async void btnIndit_Click(object sender, EventArgs e)
{
    string mappa = txtMappa.Text.Trim();

    if (!Directory.Exists(mappa)) { tsslUzenet.Text = "A megadott mappa nem létezik."; return; }

    bool alkonyvtar = chkAlkonyvtar.Checked;
    int idoMp = (int)nudIdo.Value;

    // minden indításnál nullázunk
    Interlocked.Exchange(ref szamolt, 0);
    Interlocked.Exchange(ref feldolgozott, 0);
    Interlocked.Exchange(ref osszes, 0);
    Interlocked.Exchange(ref osszMeret, 0);
    aktualis = "";
    fazis = 1;
    szunet.Set();
    lstLegnagyobb.Items.Clear(); lstKiterjesztes.Items.Clear(); lstNaplo.Items.Clear();
    lblSzazalek.Text = ""; lblMostani.Text = "";

    pb.Style = ProgressBarStyle.Marquee;   // 1. fázis: határozatlan
    pb.MarqueeAnimationSpeed = 30;
    lblAllapot.Text = "Fájlok keresése...";

    CancellationTokenSource felhasznalo = new CancellationTokenSource();
    CancellationTokenSource ido = idoMp > 0 ? new CancellationTokenSource(TimeSpan.FromSeconds(idoMp)) : new CancellationTokenSource();
    CancellationTokenSource egyutt = CancellationTokenSource.CreateLinkedTokenSource(felhasznalo.Token, ido.Token);

    forras = felhasznalo;
    Eredmeny er = new Eredmeny();
    string uzenet;

    ora.Restart();
    tmrKijelzo.Start();
    Beallit(Allapot.Fut);

    try
    {
        CancellationToken token = egyutt.Token;

        await Task.Run(() => Dolgoz(mappa, alkonyvtar, er, token), token);

        uzenet = "Kész: " + er.Fajlok + " fájl, " + Meret(er.OsszMeret) + ", " + ora.Elapsed.TotalSeconds.ToString("0.0") + " mp.";
    }
    catch (OperationCanceledException)
    {
        uzenet = (ido.IsCancellationRequested ? "Időtúllépés (" + idoMp + " mp)" : "Megszakítva") + ": " + er.Fajlok + " fájl feldolgozva.";
    }
    catch (Exception ex)
    {
        uzenet = "Hiba: " + ex.Message;
    }
    finally
    {
        tmrKijelzo.Stop();
        ora.Stop();
        egyutt.Dispose(); ido.Dispose(); felhasznalo.Dispose();
        forras = null;
    }

    if (IsDisposed) { return; }

    Kijelez(er);   // a háttérszál befejezte: az Eredmeny biztonságosan olvasható

    pb.MarqueeAnimationSpeed = 0;
    pb.Style = ProgressBarStyle.Continuous;
    long ossz = Interlocked.Read(ref osszes);
    pb.Maximum = (int)Math.Max(1, Math.Min(int.MaxValue, ossz));
    pb.Value = (int)Math.Min(pb.Maximum, Interlocked.Read(ref feldolgozott));
    lblAllapot.Text = uzenet;
    tsslUzenet.Text = uzenet;
    lblMostani.Text = "";
    fazis = 0;
    Beallit(Allapot.Kesz);
}

// ---- a háttérmunka: vezérlőt NEM ér ----
private void Dolgoz(string mappa, bool alkonyvtar, Eredmeny er, CancellationToken token)
{
    List<string> fajlok = new List<string>();
    Stack<string> verem = new Stack<string>();

    verem.Push(mappa);

    // 1. fázis: bejárás
    while (verem.Count > 0)
    {
        token.ThrowIfCancellationRequested();
        szunet.Wait(token);

        string m = verem.Pop();

        aktualis = m;

        try
        {
            foreach (string f in Directory.EnumerateFiles(m)) { fajlok.Add(f); Interlocked.Increment(ref szamolt); }

            if (alkonyvtar)
            {
                foreach (string d in Directory.EnumerateDirectories(m))
                {
                    if ((File.GetAttributes(d) & FileAttributes.ReparsePoint) == 0) { verem.Push(d); }   // hivatkozást nem követünk
                }
            }

            er.Mappak++;
        }
        catch (UnauthorizedAccessException) { er.Hibak++; er.HibaSzovegek.Add(m + ": nincs hozzáférés"); }
        catch (IOException ex) { er.Hibak++; er.HibaSzovegek.Add(m + ": " + ex.Message); }
    }

    Interlocked.Exchange(ref osszes, fajlok.Count);
    fazis = 2;

    // 2. fázis: feldolgozás
    foreach (string f in fajlok)
    {
        token.ThrowIfCancellationRequested();
        szunet.Wait(token);   // szünetben itt áll; a megszakítás kiszabadítja

        aktualis = f;

        try
        {
            long meret = new FileInfo(f).Length;
            string kit = Path.GetExtension(f).ToLowerInvariant();

            if (kit.Length == 0) { kit = "(nincs)"; }

            KiterjStat ks;

            if (!er.Kiterj.TryGetValue(kit, out ks)) { ks = new KiterjStat(); er.Kiterj[kit] = ks; }

            ks.Db++;
            ks.Meret += meret;
            er.Fajlok++;
            er.OsszMeret += meret;
            Interlocked.Add(ref osszMeret, meret);

            er.Legnagyobbak.Add(new KeyValuePair<string, long>(f, meret));

            if (er.Legnagyobbak.Count > 20)
            {
                er.Legnagyobbak.Sort((a, b) => b.Value.CompareTo(a.Value));
                er.Legnagyobbak.RemoveRange(10, er.Legnagyobbak.Count - 10);
            }
        }
        catch (IOException ex) { er.Hibak++; er.HibaSzovegek.Add(Path.GetFileName(f) + ": " + ex.Message); }
        catch (UnauthorizedAccessException) { er.Hibak++; er.HibaSzovegek.Add(Path.GetFileName(f) + ": nincs hozzáférés"); }

        Interlocked.Increment(ref feldolgozott);
    }
}

// ---- a felület frissítése: 200 ms-onként, a felületi szálon ----
private void tmrKijelzo_Tick(object sender, EventArgs e)
{
    lblIdo.Text = ora.Elapsed.ToString(@"mm\:ss");

    string ak = aktualis;

    lblMostani.Text = ak.Length > 70 ? "..." + ak.Substring(ak.Length - 67) : ak;

    if (fazis == 1)
    {
        lblSzazalek.Text = Interlocked.Read(ref szamolt) + " fájl eddig";
    }
    else if (fazis == 2)
    {
        long k = Interlocked.Read(ref feldolgozott), o = Interlocked.Read(ref osszes);

        if (pb.Style != ProgressBarStyle.Continuous)
        {
            pb.MarqueeAnimationSpeed = 0;
            pb.Style = ProgressBarStyle.Continuous;
            pb.Minimum = 0;
            pb.Maximum = (int)Math.Max(1, Math.Min(int.MaxValue, o));
            lblAllapot.Text = allapot == Allapot.Szunet ? "Szünet" : "Feldolgozás...";
        }

        pb.Value = (int)Math.Min(pb.Maximum, k);

        double arany = o == 0 ? 1 : (double)k / o;
        string szoveg = k + " / " + o + " (" + arany.ToString("P0") + ")";

        if (arany >= 0.05 && ora.Elapsed.TotalSeconds >= 3 && k < o)
        {
            szoveg += ", még kb. " + Hatralevo(TimeSpan.FromSeconds(ora.Elapsed.TotalSeconds / arany - ora.Elapsed.TotalSeconds));
        }

        lblSzazalek.Text = szoveg;
    }
}

private string Hatralevo(TimeSpan t)
{
    if (t.TotalSeconds < 5) { return "néhány másodperc"; }
    if (t.TotalSeconds < 90) { return Math.Round(t.TotalSeconds / 5) * 5 + " mp"; }

    return Math.Round(t.TotalMinutes) + " perc";
}

private string Meret(long b)
{
    string[] egyseg = { "B", "KB", "MB", "GB", "TB" };
    double v = b;
    int i = 0;

    while (v >= 1024 && i < egyseg.Length - 1) { v /= 1024; i++; }

    return v.ToString(i == 0 ? "0" : "0.##") + " " + egyseg[i];
}

private void Kijelez(Eredmeny er)
{
    er.Legnagyobbak.Sort((a, b) => b.Value.CompareTo(a.Value));

    foreach (KeyValuePair<string, long> p in er.Legnagyobbak.Take(10)) { lstLegnagyobb.Items.Add(Meret(p.Value) + "   " + p.Key); }

    foreach (KeyValuePair<string, KiterjStat> p in er.Kiterj.OrderByDescending(x => x.Value.Meret))
    {
        lstKiterjesztes.Items.Add(p.Key + ": " + p.Value.Db + " db, " + Meret(p.Value.Meret));
    }

    lstNaplo.Items.Add("Mappák: " + er.Mappak + ", fájlok: " + er.Fajlok + ", összméret: " + Meret(er.OsszMeret) + ", hibák: " + er.Hibak);

    foreach (string h in er.HibaSzovegek.Take(200)) { lstNaplo.Items.Add("⚠ " + h); }
}

// ---- szünet, megszakítás, bezárás ----
private void btnSzunet_Click(object sender, EventArgs e)
{
    if (allapot == Allapot.Fut)
    {
        szunet.Reset();
        ora.Stop();
        Beallit(Allapot.Szunet);
        lblAllapot.Text = "Szünet";
    }
    else if (allapot == Allapot.Szunet)
    {
        szunet.Set();
        ora.Start();
        Beallit(Allapot.Fut);
        lblAllapot.Text = fazis == 1 ? "Fájlok keresése..." : "Feldolgozás...";
    }
}

private void btnMegse_Click(object sender, EventArgs e)
{
    Beallit(Allapot.Megszakitas);

    if (forras != null) { forras.Cancel(); }
}

protected override bool ProcessCmdKey(ref Message msg, Keys keyData)
{
    if (keyData == Keys.Escape && btnMegse.Enabled) { btnMegse.PerformClick(); return true; }

    return base.ProcessCmdKey(ref msg, keyData);
}

private async void Form1_FormClosing(object sender, FormClosingEventArgs e)
{
    if (forras != null && !bezarasra)
    {
        e.Cancel = true;
        bezarasra = true;
        forras.Cancel();

        while (forras != null) { await Task.Delay(50); }

        Close();
    }
}