/**
 * Jooby https://jooby.io
 * Apache License Version 2.0 https://jooby.io/LICENSE.txt
 * Copyright 2014 Edgar Espina
 */
package io.jooby.internal;

import io.jooby.Context;
import io.jooby.Session;
import io.jooby.SessionToken;
import io.jooby.SessionStore;

import javax.annotation.Nonnull;
import java.time.Instant;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

public class MemorySessionStore implements SessionStore {
  private static class SessionData {
    private Instant lastAccessedTime;
    private Instant creationTime;
    private Map hash;

    public SessionData(Instant creationTime, Instant lastAccessedTime, Map hash) {
      this.creationTime = creationTime;
      this.lastAccessedTime = lastAccessedTime;
      this.hash = hash;
    }
  }

  private ConcurrentHashMap<String, SessionData> sessions = new ConcurrentHashMap<>();

  private SessionToken token;

  public MemorySessionStore(SessionToken token) {
    this.token = token;
  }

  @Override public Session newSession(Context ctx) {
    String sessionId = token.newToken();
    SessionData data = sessions.computeIfAbsent(sessionId, sid -> {
      Instant now = Instant.now();
      return new SessionData(now, now, new ConcurrentHashMap());
    });

    Session session = restore(ctx, sessionId, data);

    token.saveToken(ctx, sessionId);
    return session;
  }

  @Override public Session findSession(Context ctx) {
    String sessionId = token.findToken(ctx);
    if (sessionId == null) {
      return null;
    }
    SessionData data = sessions.get(sessionId);
    if (data != null) {
      Session session = restore(ctx, sessionId, data);
      token.saveToken(ctx, sessionId);
      return session;
    }
    return null;
  }

  @Override public void deleteSession(@Nonnull Context ctx, @Nonnull Session session) {
    String sessionId = session.getId();
    sessions.remove(sessionId);
    token.deleteToken(ctx, sessionId);
  }

  @Override public void touchSession(@Nonnull Context ctx, @Nonnull Session session) {
    saveSession(ctx, session);
  }

  @Override public void saveSession(Context ctx, @Nonnull Session session) {
    String sessionId = session.getId();
    sessions.put(sessionId,
        new SessionData(session.getCreationTime(), Instant.now(), session.toMap()));
    token.saveToken(ctx, sessionId);
  }

  private Session restore(Context ctx, String sessionId, SessionData data) {
    return Session.create(ctx, sessionId, data.hash)
        .setLastAccessedTime(data.lastAccessedTime)
        .setCreationTime(data.creationTime);
  }
}

@@ -2057,16 +2057,59 @@
 ssion);%0A
+    token.saveToken(ctx, session.getId());%0A
   %7D%0A%0A  @
@@ -2339,45 +2339,8 @@
 ));%0A
-    token.saveToken(ctx, sessionId);%0A
   %7D%0A

remove savetoken from savesession call (no context modification is allowed at this point)