package scorekeep;

import java.io.File;

import java.util.Iterator;

import java.util.Map;

import java.util.HashMap;

import java.util.List;

import java.util.Set;

import java.util.HashSet;

import com.amazonaws.auth.AWSCredentials;

import com.amazonaws.auth.profile.ProfileCredentialsProvider;

import com.amazonaws.auth.AWSCredentialsProvider;

import com.amazonaws.services.dynamodbv2.AmazonDynamoDB;

import com.amazonaws.services.dynamodbv2.AmazonDynamoDBClientBuilder;

import com.amazonaws.services.dynamodbv2.datamodeling.DynamoDBMapper;

import com.amazonaws.services.dynamodbv2.datamodeling.DynamoDBQueryExpression;

import com.amazonaws.services.dynamodbv2.model.AttributeValue;

import com.amazonaws.regions.Regions;

import com.fasterxml.jackson.core.JsonFactory;

import com.fasterxml.jackson.core.JsonParser;

import com.fasterxml.jackson.databind.JsonNode;

import com.fasterxml.jackson.databind.ObjectMapper;

import com.fasterxml.jackson.databind.node.ObjectNode;

import java.lang.Exception;

import java.lang.Throwable;

public class GameModel {

/** AWS SDK credentials. */

private AmazonDynamoDB client = AmazonDynamoDBClientBuilder.standard()

.withRegion(Regions.fromName(System.getenv("AWS_REGION")))

.build();

private DynamoDBMapper mapper = new DynamoDBMapper(client);

private SessionModel sessionModel = new SessionModel();

public void saveGame(Game game) throws SessionNotFoundException {

try {

// check session

String sessionId = game.getSession();

if (sessionModel.loadSession(sessionId) == null ) {

throw new SessionNotFoundException(sessionId);

}

mapper.save(game);

} catch (Exception e) {

throw e;

}

}

public Game loadGame(String gameId) throws GameNotFoundException {

Game game = mapper.load(Game.class, gameId);

if ( game == null ) {

throw new GameNotFoundException(gameId);

}

return game;

}

public List<Game> loadGames(String sessionId) throws SessionNotFoundException {

if ( sessionModel.loadSession(sessionId) == null ) {

throw new SessionNotFoundException(sessionId);

}

Map<String, AttributeValue> eav = new HashMap<String, AttributeValue>();

eav.put(":val1", new AttributeValue().withS(sessionId));

Map<String, String> ean = new HashMap<String, String>();

ean.put("#key1", "session");

DynamoDBQueryExpression<Game> queryExpression = new DynamoDBQueryExpression<Game>()

.withIndexName("session-index")

.withExpressionAttributeValues(eav)

.withExpressionAttributeNames(ean)

.withKeyConditionExpression("#key1 = :val1")

.withConsistentRead(false);

List<Game> sessionGames = mapper.query(Game.class, queryExpression);

return sessionGames;

}

public void deleteGame(String sessionId, String gameId) throws GameNotFoundException {

Game game = mapper.load(Game.class, gameId);

if ( game == null ) {

throw new GameNotFoundException(gameId);

}

mapper.delete(game);

//delete game from session

Session session = mapper.load(Session.class, sessionId);

Set<String> sessionGames = session.getGames();

sessionGames.remove(gameId);

if (sessionGames.size() == 0) {

session.clearGames();

} else {

session.setGames(sessionGames);

}

mapper.save(session);

}

}