package org.syndicate_lang.actors;

import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.*;
import java.util.concurrent.atomic.AtomicLong;
import java.util.function.Consumer;
import java.util.function.Supplier;
import java.util.logging.Level;
import java.util.logging.Logger;

/**
 * I represent the shared execution concepts for a collection of objects; I am roughly analogous to the E concept of a Vat.
 */
public class Actor implements Executor {
    private final static ThreadLocal<Actor> _currentActor = new ThreadLocal<>();
    private final static AtomicLong _count = new AtomicLong(0);
    private final static AtomicLong _actorId = new AtomicLong(0);
    protected final static ExecutorService _executor = Executors.newWorkStealingPool();
    protected final static ScheduledExecutorService _scheduledExecutor = Executors.newScheduledThreadPool(Runtime.getRuntime().availableProcessors());

    private final String _name;
    private final Logger _logger;

    private boolean _alive = true;
    private Throwable _exitReason = null;
    private boolean _isCounted = true;
    private Set<Actor> _links = null;
    private Map<Object, Remote<IMonitorHandler>> _monitors = null;
    private Consumer<Actor> _exitTrap = null;

    public static Actor current() {
        return _currentActor.get();
    }

    public static Logger log() {
        return current().getLogger();
    }

    public static<T> Remote<T> ref(T o) {
        return current().proxyFor(o);
    }

    public Actor() {
        this("" + _actorId.incrementAndGet());
    }

    public Actor(String debugName) {
        this._name = debugName;
        this._logger = Logger.getLogger(this.getClass().getSimpleName() + "(" + this._name + ")");
        _count.incrementAndGet();
    }

    public static<T> Remote<T> forObject(T o) {
        return new Actor().proxyFor(o);
    }

    public static<T> Promise<Remote<T>> boot(Supplier<T> f) {
        Promise<Remote<T>> p = new Promise<>();
        Actor a = new Actor();
        a.execute(
                () -> p.resolveCalling(() -> Actor.ref(f.get())),
                () -> p.rejectWith(new ActorTerminated(a)));
        return p;
    }

    public String getName() {
        return _name;
    }

    public Throwable getExitReason() {
        return _exitReason;
    }

    public Logger getLogger() {
        return _logger;
    }

    public synchronized boolean isDaemon() {
        return _alive && !_isCounted;
    }

    public synchronized Actor daemonize() {
        this._releaseCount();
        return this;
    }

    private void _releaseCount() {
        if (_isCounted) {
            _isCounted = false;
            synchronized (_count) {
                if (_count.decrementAndGet() == 0) {
                    _count.notifyAll();
                }
            }
        }
    }

    public String toString() {
        return super.toString() + "(" + this._name + ")";
    }

    public<T> Remote<T> proxyFor(T o) {
        return new Remote<T>(this, o);
    }

    private void _perform(Runnable work, Runnable ifNotAlive) {
        synchronized (this) {
            _currentActor.set(this);
            try {
                if (!_alive) {
                    if (ifNotAlive != null) ifNotAlive.run();
                } else {
                    try {
                        work.run();
                    } catch (Throwable exn) {
                        this._stop(false, exn);
                    }
                }
            } finally {
                _currentActor.set(null);
            }
        }
    }

    public Promise<?> stop() {
        return stop(null);
    }

    public Promise<?> stop(Throwable reason) {
        if (current() == this) {
            this._stop(true, reason);
            return Promise.resolved();
        } else {
            Promise<?> p = new Promise<>();
            this.execute(() -> {
                this._stop(true, reason);
                p.resolve();
            }, p::resolve);
            return p;
        }
    }

    private synchronized void _stop(boolean normally, Throwable reason) {
        if (_alive) {
            _alive = false;
            _exitReason = reason;
            if (normally) {
                getLogger().log(Level.FINE, "Actor stopped", reason);
            } else {
                getLogger().log(Level.SEVERE, "Actor terminated with error", reason);
            }
            Set<Actor> linkedPeers = _links;
            if (linkedPeers != null) {
                _links = null;
                for (var peer : linkedPeers) {
                    peer.notifyExit(this);
                }
            }
            Map<Object, Remote<IMonitorHandler>> monitoringPeers = _monitors;
            if (monitoringPeers != null) {
                _monitors = null;
                for (var entry : monitoringPeers.entrySet()) {
                    final var ref = entry.getKey();
                    entry.getValue().async((h) -> h.handleMonitor(this, ref));
                }
            }
            _releaseCount();
        }
    }

    @Override
    public void execute(Runnable work) {
        this.execute(work, null);
    }

    public void execute(Runnable work, Runnable ifNotAlive) {
        this.later(0, work, ifNotAlive);
    }

    public void later(long delayMilliseconds, Runnable work) {
        this.later(delayMilliseconds, work, null);
    }

    public void later(long delayMilliseconds, Runnable work, Runnable ifNotAlive) {
        if (delayMilliseconds == 0) {
            _executor.execute(() -> this._perform(work, ifNotAlive));
        } else {
            _scheduledExecutor.schedule(() -> this._perform(work, ifNotAlive), delayMilliseconds, TimeUnit.MILLISECONDS);
        }
    }

    public PeriodicTimer every(long periodMilliseconds, Runnable f) {
        return every(0, periodMilliseconds, f);
    }

    public PeriodicTimer every(long initialDelayMilliseconds, long periodMilliseconds, Runnable f) {
        return new PeriodicTimer(
                _scheduledExecutor.scheduleAtFixedRate(
                        () -> this._perform(f, null),
                        initialDelayMilliseconds,
                        periodMilliseconds,
                        TimeUnit.MILLISECONDS));
    }

    public static void awaitAll() throws InterruptedException {
        while (_count.get() > 0) {
            synchronized (_count) {
                _count.wait(1000);
            }
        }
        _executor.shutdown();
        _scheduledExecutor.shutdown();
        _executor.awaitTermination(5, TimeUnit.MINUTES);
        _scheduledExecutor.awaitTermination(5, TimeUnit.MINUTES);
    }

    public static void convenientLogging() {
        System.setProperty("java.util.logging.SimpleFormatter.format",
                "%1$tY-%1$tm-%1$td %1$tH:%1$tM:%1$tS.%1$tL %4$s %3$s %5$s%6$s%n");
    }

    public void link() {
        final Actor peer = Actor.current();
        this.linkPeer(peer);
        peer.linkPeer(this);
    }

    public void unlink() {
        final Actor peer = Actor.current();
        this.unlinkPeer(peer);
        peer.unlinkPeer(this);
    }

    private synchronized void linkPeer(Actor peer) {
        if (this._alive) {
            if (_links == null) _links = new HashSet<>();
            _links.add(peer);
        } else {
            peer.notifyExit(this);
        }
    }

    private void notifyExit(final Actor exitingPeer) {
        this.execute(() -> {
            this.unlinkPeer(exitingPeer);
            if (this._exitTrap != null) {
                this._exitTrap.accept(exitingPeer);
            } else {
                this._stop(exitingPeer.getExitReason() == null, new ActorTerminated(exitingPeer));
            }
        });
    }

    private synchronized void unlinkPeer(Actor peer) {
        if (_links != null) {
            _links.remove(peer);
            if (_links.isEmpty()) {
                _links = null;
            }
        }
    }

    public void trapExits(Consumer<Actor> handler) {
        this._exitTrap = handler;
    }

    public synchronized Object monitor(Consumer<Actor> handler) {
        Object ref = new Object();
        monitor(ref, (actor, _ref) -> handler.accept(actor));
        return ref;
    }

    public synchronized void monitor(final Object ref, IMonitorHandler handler) {
        if (this._alive) {
            if (_monitors == null) _monitors = new HashMap<>();
            _monitors.put(ref, Actor.ref(handler));
        } else {
            Actor.ref(handler).async((h) -> h.handleMonitor(this, ref));
        }
    }
}