/*----------------------------------------------------------

This Source Code Form is subject to the terms of the

Mozilla Public License, v.2.0. If a copy of the MPL

was not distributed with this file, You can obtain one

at http://mozilla.org/MPL/2.0/.

----------------------------------------------------------*/

using System;

using System.Collections.Concurrent;

using System.Collections.Generic;

using System.Linq;

using System.Threading;

using ScriptEngine.Machine;

namespace OneScript.DebugServices

{

public class ThreadManager : IDisposable

{

private readonly IDictionary<int, MachineWaitToken> _machinesOnThreads = new ConcurrentDictionary<int, MachineWaitToken>();

public MachineWaitToken GetTokenForThread(int threadId)

{

return _machinesOnThreads[threadId];

}

public MachineWaitToken GetTokenForCurrentThread()

{

return GetTokenForThread(Thread.CurrentThread.ManagedThreadId);

}

public event EventHandler<ThreadStoppedEventArgs> ThreadStopped;

public void AttachToCurrentThread()

{

var machine = MachineInstance.Current;

_machinesOnThreads[Thread.CurrentThread.ManagedThreadId] = new MachineWaitToken()

{

Machine = machine

};

machine.MachineStopped += Machine_MachineStopped;

}

public void DetachFromCurrentThread()

{

var threadId = Thread.CurrentThread.ManagedThreadId;

DetachFromThread(threadId);

}

public void DetachFromThread(int threadId)

{

if (_machinesOnThreads.TryGetValue(threadId, out var t))

{

t.Machine.MachineStopped -= Machine_MachineStopped;

_machinesOnThreads.Remove(threadId);

// Если машина была остановлена - продолжаем её уже без остановок

t.Machine.UnsetDebugMode();

t.Set();

}

}

public MachineWaitToken[] GetAllTokens()

{

return _machinesOnThreads.Values.ToArray();

}

private void Machine_MachineStopped(object sender, MachineStoppedEventArgs e)

{

var args = new ThreadStoppedEventArgs

{

Machine = (MachineInstance)sender,

ThreadId = e.ThreadId,

StopReason = e.Reason,

ErrorMessage = e.ErrorMessage

};

ThreadStopped?.Invoke(this, args);

}

public int[] GetAllThreadIds()

{

return _machinesOnThreads.Keys.ToArray();

}

public void ReleaseAllThreads()

{

var tokens = GetAllTokens();

foreach (var machineWaitToken in tokens)

{

machineWaitToken.Machine.MachineStopped -= Machine_MachineStopped;

machineWaitToken.Dispose();

}

_machinesOnThreads.Clear();

}

public void Dispose()

{

ReleaseAllThreads();

}

}

}