GameServer.java

package emu.grasscutter.server.game;

import emu.grasscutter.GameConstants;
import emu.grasscutter.Grasscutter;
import emu.grasscutter.command.CommandMap;
import emu.grasscutter.database.DatabaseHelper;
import emu.grasscutter.game.Account;
import emu.grasscutter.game.battlepass.BattlePassSystem;
import emu.grasscutter.game.combine.CombineManger;
import emu.grasscutter.game.drop.DropSystem;
import emu.grasscutter.game.dungeons.DungeonSystem;
import emu.grasscutter.game.dungeons.challenge.DungeonChallenge;
import emu.grasscutter.game.expedition.ExpeditionSystem;
import emu.grasscutter.game.gacha.GachaSystem;
import emu.grasscutter.game.managers.AnnouncementSystem;
import emu.grasscutter.game.managers.CookingManager;
import emu.grasscutter.game.managers.InventorySystem;
import emu.grasscutter.game.managers.MultiplayerSystem;
import emu.grasscutter.game.managers.chat.ChatManager;
import emu.grasscutter.game.managers.chat.ChatManagerHandler;
import emu.grasscutter.game.managers.energy.EnergyManager;
import emu.grasscutter.game.managers.stamina.StaminaManager;
import emu.grasscutter.game.player.Player;
import emu.grasscutter.game.quest.QuestSystem;
import emu.grasscutter.game.shop.ShopSystem;
import emu.grasscutter.game.tower.TowerSystem;
import emu.grasscutter.game.world.World;
import emu.grasscutter.game.world.WorldDataSystem;
import emu.grasscutter.net.packet.PacketHandler;
import emu.grasscutter.net.proto.SocialDetailOuterClass.SocialDetail;
import emu.grasscutter.server.event.types.ServerEvent;
import emu.grasscutter.server.event.game.ServerTickEvent;
import emu.grasscutter.server.event.internal.ServerStartEvent;
import emu.grasscutter.server.event.internal.ServerStopEvent;
import emu.grasscutter.server.scheduler.ServerTaskScheduler;
import emu.grasscutter.task.TaskMap;
import kcp.highway.ChannelConfig;
import kcp.highway.KcpServer;
import lombok.Getter;

import java.net.InetSocketAddress;
import java.time.Instant;
import java.time.OffsetDateTime;
import java.util.*;
import java.util.concurrent.ConcurrentHashMap;

import static emu.grasscutter.utils.Language.translate;
import static emu.grasscutter.Configuration.*;

@Getter
public final class GameServer extends KcpServer {
    // Game server base
    private final InetSocketAddress address;
	private final GameServerPacketHandler packetHandler;
    private final Map<Integer, Player> players;
    private final Set<World> worlds;

    // Server systems
	private final InventorySystem inventorySystem;
	private final GachaSystem gachaSystem;
	private final ShopSystem shopSystem;
	private final MultiplayerSystem multiplayerSystem;
	private final DungeonSystem dungeonSystem;
	private final ExpeditionSystem expeditionSystem;
	private final DropSystem dropSystem;
	private final WorldDataSystem worldDataSystem;
	private final BattlePassSystem battlePassSystem;
	private final CombineManger combineSystem;
	private final TowerSystem towerSystem;
	private final AnnouncementSystem announcementSystem;
	private final QuestSystem questSystem;
	
	// Extra
	private final ServerTaskScheduler scheduler;
    private final CommandMap commandMap;
    private final TaskMap taskMap;
	
	private ChatManagerHandler chatManager;

	public GameServer() {
		this(getAdapterInetSocketAddress());
	}

	public GameServer(InetSocketAddress address) {
		ChannelConfig channelConfig = new ChannelConfig();
		channelConfig.nodelay(true, 40, 2, true);
		channelConfig.setMtu(1400);
		channelConfig.setSndwnd(256);
		channelConfig.setRcvwnd(256);
		channelConfig.setTimeoutMillis(30 * 1000);//30s
		channelConfig.setUseConvChannel(true);
		channelConfig.setAckNoDelay(false);

		this.init(GameSessionManager.getListener(),channelConfig,address);

		DungeonChallenge.initialize();
		EnergyManager.initialize();
		StaminaManager.initialize();
		CookingManager.initialize();
		CombineManger.initialize();

		// Game Server base
		this.address = address;
		this.packetHandler = new GameServerPacketHandler(PacketHandler.class);
		this.players = new ConcurrentHashMap<>();
		this.worlds = Collections.synchronizedSet(new HashSet<>());
		
		// Extra
		this.scheduler = new ServerTaskScheduler();
		this.commandMap = new CommandMap(true);
        this.taskMap = new TaskMap(true);

		// Create game systems
		this.inventorySystem = new InventorySystem(this);
		this.gachaSystem = new GachaSystem(this);
		this.shopSystem = new ShopSystem(this);
		this.multiplayerSystem = new MultiplayerSystem(this);
		this.dungeonSystem = new DungeonSystem(this);
		this.dropSystem = new DropSystem(this);
		this.expeditionSystem = new ExpeditionSystem(this);
		this.combineSystem = new CombineManger(this);
		this.towerSystem = new TowerSystem(this);
		this.worldDataSystem = new WorldDataSystem(this);
		this.battlePassSystem = new BattlePassSystem(this);
		this.announcementSystem = new AnnouncementSystem(this);
		this.questSystem = new QuestSystem(this);
		
		// Chata manager
		this.chatManager = new ChatManager(this);
		
		// Hook into shutdown event.
		Runtime.getRuntime().addShutdownHook(new Thread(this::onServerShutdown));
	}
	
    @Deprecated
    public ChatManagerHandler getChatManager() {
        return chatManager;
    }
    
    @Deprecated
    public void setChatManager(ChatManagerHandler chatManager) {
        this.chatManager = chatManager;
    }
    
    public ChatManagerHandler getChatSystem() {
        return chatManager;
    }

    public void setChatSystem(ChatManagerHandler chatManager) {
        this.chatManager = chatManager;
    }

	private static InetSocketAddress getAdapterInetSocketAddress(){
		InetSocketAddress inetSocketAddress;
		if(GAME_INFO.bindAddress.equals("")){
			inetSocketAddress=new InetSocketAddress(GAME_INFO.bindPort);
		}else{
			inetSocketAddress=new InetSocketAddress(
					GAME_INFO.bindAddress,
					GAME_INFO.bindPort
			);
		}
		return inetSocketAddress;
	}

	public void registerPlayer(Player player) {
		getPlayers().put(player.getUid(), player);
	}

	public Player getPlayerByUid(int id) {
		return this.getPlayerByUid(id, false);
	}

	public Player getPlayerByUid(int id, boolean allowOfflinePlayers) {
		// Console check
		if (id == GameConstants.SERVER_CONSOLE_UID) {
			return null;
		}

		// Get from online players
		Player player = this.getPlayers().get(id);

		if (!allowOfflinePlayers) {
			return player;
		}

		// Check database if character isnt here
		if (player == null) {
			player = DatabaseHelper.getPlayerByUid(id);
		}

		return player;
	}

	public Player getPlayerByAccountId(String accountId) {
		Optional<Player> playerOpt = getPlayers().values().stream().filter(player -> player.getAccount().getId().equals(accountId)).findFirst();
		return playerOpt.orElse(null);
	}

	public SocialDetail.Builder getSocialDetailByUid(int id) {
		// Get from online players
		Player player = this.getPlayerByUid(id, true);

		if (player == null) {
			return null;
		}

		return player.getSocialDetail();
	}

	public Account getAccountByName(String username) {
		Optional<Player> playerOpt = getPlayers().values().stream().filter(player -> player.getAccount().getUsername().equals(username)).findFirst();
		if (playerOpt.isPresent()) {
			return playerOpt.get().getAccount();
		}
		return DatabaseHelper.getAccountByName(username);
	}

    public synchronized void onTick() {
        var tickStart = Instant.now();

        // Tick worlds.
        Iterator<World> it = this.getWorlds().iterator();
        while (it.hasNext()) {
            World world = it.next();

            if (world.getPlayerCount() == 0) {
                it.remove();
            }

            world.onTick();
        }

        // Tick players.
        for (Player player : this.getPlayers().values()) {
            player.onTick();
        }

        // Tick scheduler.
        this.getScheduler().runTasks();

        // Call server tick event.
        ServerTickEvent event = new ServerTickEvent(tickStart, Instant.now());
        event.call();
    }

	public void registerWorld(World world) {
		this.getWorlds().add(world);
	}

	public void deregisterWorld(World world) {
		// TODO Auto-generated method stub

	}

	public void start() {
		// Schedule game loop.
		Timer gameLoop = new Timer();
		gameLoop.scheduleAtFixedRate(new TimerTask() {
			@Override
			public void run() {
				try {
					onTick();
				} catch (Exception e) {
					Grasscutter.getLogger().error(translate("messages.game.game_update_error"), e);
				}
			}
		}, new Date(), 1000L);
		Grasscutter.getLogger().info(translate("messages.status.free_software"));
		Grasscutter.getLogger().info(translate("messages.game.port_bind", Integer.toString(address.getPort())));
		ServerStartEvent event = new ServerStartEvent(ServerEvent.Type.GAME, OffsetDateTime.now());
		event.call();
	}

	public void onServerShutdown() {
		ServerStopEvent event = new ServerStopEvent(ServerEvent.Type.GAME, OffsetDateTime.now()); event.call();

		// Kick and save all players
		List<Player> list = new ArrayList<>(this.getPlayers().size());
		list.addAll(this.getPlayers().values());

		for (Player player : list) {
			player.getSession().close();
		}
	}
}