src/main/java/com/l2jserver/gameserver/model/MacroList.java - develop - tofu/l2interlude

src/main/java/com/l2jserver/gameserver/model/MacroList.java
191 строка6 KB
Zoey76
01 янв 2025, 07:57
01 янв 2025, 07:5716130a5
/*
 * Copyright © 2004-2025 L2J Server
 *
 * This file is part of L2J Server.
 *
 * L2J Server is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * L2J Server is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */
package com.l2jserver.gameserver.model;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.StringTokenizer;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.l2jserver.commons.database.ConnectionFactory;
import com.l2jserver.gameserver.enums.MacroType;
import com.l2jserver.gameserver.enums.ShortcutType;
import com.l2jserver.gameserver.model.actor.instance.L2PcInstance;
import com.l2jserver.gameserver.model.interfaces.IRestorable;
import com.l2jserver.gameserver.network.serverpackets.SendMacroList;
import com.l2jserver.gameserver.util.StringUtil;

public class MacroList implements IRestorable {
	private static final Logger LOG = LoggerFactory.getLogger(MacroList.class);
	
	private final L2PcInstance _owner;
	
	private int _revision;
	
	private int _macroId;
	
	private final Map<Integer, Macro> _macros = Collections.synchronizedMap(new LinkedHashMap<>());
	
	public MacroList(L2PcInstance owner) {
		_owner = owner;
		_revision = 1;
		_macroId = 1000;
	}
	
	public int getRevision() {
		return _revision;
	}
	
	public Map<Integer, Macro> getAllMacros() {
		return _macros;
	}
	
	public void registerMacro(Macro macro) {
		if (macro.getId() == 0) {
			do {
				macro.setId(_macroId++);
			}
			while (_macros.containsKey(macro.getId()));
			_macros.put(macro.getId(), macro);
			registerMacroInDb(macro);
		} else {
			final Macro old = _macros.put(macro.getId(), macro);
			if (old != null) {
				deleteMacroFromDb(old);
			}
			registerMacroInDb(macro);
		}
		sendUpdate();
	}
	
	public void deleteMacro(int id) {
		final Macro removed = _macros.remove(id);
		if (removed != null) {
			deleteMacroFromDb(removed);
		}
		
		final Shortcut[] allShortCuts = _owner.getAllShortCuts();
		for (Shortcut sc : allShortCuts) {
			if ((sc.getId() == id) && (sc.getType() == ShortcutType.MACRO)) {
				_owner.deleteShortCut(sc.getSlot(), sc.getPage());
			}
		}
		
		sendUpdate();
	}
	
	public void sendUpdate() {
		_revision++;
		final Collection<Macro> allMacros = _macros.values();
		synchronized (_macros) {
			if (allMacros.isEmpty()) {
				_owner.sendPacket(new SendMacroList(_revision, 0, null));
			} else {
				for (Macro m : allMacros) {
					_owner.sendPacket(new SendMacroList(_revision, allMacros.size(), m));
				}
			}
		}
	}
	
	private void registerMacroInDb(Macro macro) {
		try (var con = ConnectionFactory.getInstance().getConnection();
			var ps = con.prepareStatement("INSERT INTO character_macroses (charId,id,icon,name,descr,acronym,commands) values(?,?,?,?,?,?,?)")) {
			ps.setInt(1, _owner.getObjectId());
			ps.setInt(2, macro.getId());
			ps.setInt(3, macro.getIcon());
			ps.setString(4, macro.getName());
			ps.setString(5, macro.getDescr());
			ps.setString(6, macro.getAcronym());
			final StringBuilder sb = new StringBuilder(300);
			for (MacroCmd cmd : macro.getCommands()) {
				StringUtil.append(sb, String.valueOf(cmd.getType().ordinal()), ",", String.valueOf(cmd.getD1()), ",", String.valueOf(cmd.getD2()));
				if ((cmd.getCmd() != null) && (!cmd.getCmd().isEmpty())) {
					StringUtil.append(sb, ",", cmd.getCmd());
				}
				sb.append(';');
			}
			
			if (sb.length() > 255) {
				sb.setLength(255);
			}
			
			ps.setString(7, sb.toString());
			ps.execute();
		} catch (Exception e) {
			LOG.warn("Could not store macro:", e);
		}
	}
	
	private void deleteMacroFromDb(Macro macro) {
		try (var con = ConnectionFactory.getInstance().getConnection();
			var ps = con.prepareStatement("DELETE FROM character_macroses WHERE charId=? AND id=?")) {
			ps.setInt(1, _owner.getObjectId());
			ps.setInt(2, macro.getId());
			ps.execute();
		} catch (Exception e) {
			LOG.warn("Could not delete macro:", e);
		}
	}
	
	@Override
	public boolean restoreMe() {
		_macros.clear();
		try (var con = ConnectionFactory.getInstance().getConnection();
			var ps = con.prepareStatement("SELECT charId, id, icon, name, descr, acronym, commands FROM character_macroses WHERE charId=?")) {
			ps.setInt(1, _owner.getObjectId());
			try (var rs = ps.executeQuery()) {
				while (rs.next()) {
					int id = rs.getInt("id");
					int icon = rs.getInt("icon");
					String name = rs.getString("name");
					String descr = rs.getString("descr");
					String acronym = rs.getString("acronym");
					List<MacroCmd> commands = new ArrayList<>();
					StringTokenizer st1 = new StringTokenizer(rs.getString("commands"), ";");
					while (st1.hasMoreTokens()) {
						StringTokenizer st = new StringTokenizer(st1.nextToken(), ",");
						if (st.countTokens() < 3) {
							continue;
						}
						MacroType type = MacroType.values()[Integer.parseInt(st.nextToken())];
						int d1 = Integer.parseInt(st.nextToken());
						int d2 = Integer.parseInt(st.nextToken());
						String cmd = "";
						if (st.hasMoreTokens()) {
							cmd = st.nextToken();
						}
						commands.add(new MacroCmd(commands.size(), type, d1, d2, cmd));
					}
					_macros.put(id, new Macro(id, icon, name, descr, acronym, commands));
				}
			}
		} catch (Exception e) {
			LOG.warn("Could not store shortcuts:", e);
			return false;
		}
		return true;
	}
}