Examples with AxsListItemRecord com.almuradev.generated.axs.tables.records.AxsListItemRecord used on opensource projects

Example 1 with AxsListItemRecord

use of com.almuradev.generated.axs.tables.records.AxsListItemRecord in project Almura by AlmuraDev.

the class ServerExchangeManager method handleModifyListItems.

public void handleModifyListItems(final Player player, final String id, final List<InventoryAction> actions) {
    checkNotNull(player);
    checkNotNull(id);
    checkNotNull(actions);
    checkState(!actions.isEmpty());
    final Exchange axs = this.getExchange(id).orElse(null);
    if (axs == null) {
        this.logger.error("Player '{}' attempted to list items for exchange '{}' but the server has no knowledge of it. Syncing exchange " + "registry...", player.getName(), id);
        this.syncExchangeRegistryTo(player);
        return;
    }
    final UUID seller = player.getUniqueId();
    final EntityPlayerMP serverPlayer = (EntityPlayerMP) player;
    final IItemHandler simulatedInventory = serverPlayer.getCapability(CapabilityItemHandler.ITEM_HANDLER_CAPABILITY, EnumFacing.UP);
    // Inventory -> Listing
    final List<InventoryAction> toListingActions = actions.stream().filter(a -> a.getDirection() == InventoryAction.Direction.TO_LISTING).collect(Collectors.toList());
    final List<VanillaStack> toListingStacks = new ArrayList<>();
    final List<VanillaStack> unknownInventoryStacks = new ArrayList<>();
    final List<VanillaStack> leftoverToListingStacks = new ArrayList<>();
    for (InventoryAction toListingAction : toListingActions) {
        final VanillaStack stack = toListingAction.getStack();
        int amountLeft = stack.getQuantity();
        boolean matched = false;
        for (int j = 0; j < simulatedInventory.getSlots(); j++) {
            final ItemStack slotStack = simulatedInventory.getStackInSlot(j);
            if (ItemHandlerHelper.canItemStacksStack(slotStack, stack.asRealStack())) {
                amountLeft -= simulatedInventory.extractItem(j, amountLeft, false).getCount();
                matched = true;
            }
            if (amountLeft <= 0) {
                break;
            }
        }
        if (!matched) {
            unknownInventoryStacks.add(stack);
        } else {
            if (amountLeft > 0) {
                final VanillaStack copyStack = stack.copy();
                copyStack.setQuantity(amountLeft);
                leftoverToListingStacks.add(copyStack);
            }
            final VanillaStack copyStack = stack.copy();
            copyStack.setQuantity(stack.getQuantity() - amountLeft);
            if (copyStack.getQuantity() != 0) {
                toListingStacks.add(copyStack);
            }
        }
    }
    // Listing -> Inventory
    final List<InventoryAction> toInventoryActions = actions.stream().filter(a -> a.getDirection() == InventoryAction.Direction.TO_INVENTORY).collect(Collectors.toList());
    final List<VanillaStack> listingNotFoundStacks = new ArrayList<>();
    final List<ListItem> desyncToInventoryStacks = new ArrayList<>();
    final List<ListItem> toInventoryStacks = new ArrayList<>();
    final List<ListItem> partialToInventoryStacks = new ArrayList<>();
    final List<ListItem> currentListItems = axs.getListItemsFor(player.getUniqueId()).orElse(null);
    if (!toInventoryActions.isEmpty()) {
        if (currentListItems == null || currentListItems.isEmpty()) {
            this.logger.error("Player '{}' attempted to move listings back to the inventory for exchange '{}' but the server knows of no " + "listings for them. This could be a de-sync or an exploit. Printing stacks...", player.getName(), axs.getId());
            this.network.sendTo(player, new ClientboundListItemsResponsePacket(axs.getId(), null));
            this.printStacksToConsole(toInventoryActions.stream().map(InventoryAction::getStack).collect(Collectors.toList()));
        } else {
            for (final InventoryAction action : toInventoryActions) {
                final VanillaStack stack = action.getStack();
                ListItem found = null;
                for (final ListItem listItem : currentListItems) {
                    if (ItemHandlerHelper.canItemStacksStack(stack.asRealStack(), listItem.asRealStack())) {
                        found = listItem;
                        break;
                    }
                }
                // Unknown listing
                if (found == null) {
                    listingNotFoundStacks.add(stack);
                    continue;
                }
                ListItem toRemove = found.copy();
                // Listing quantity mismatch (tracking this to let the user know)
                if (found.getQuantity() < stack.getQuantity()) {
                    desyncToInventoryStacks.add(found);
                    toRemove.setQuantity(found.getQuantity());
                } else {
                    toRemove.setQuantity(stack.getQuantity());
                }
                final ItemStack resultStack = ItemHandlerHelper.insertItemStacked(simulatedInventory, toRemove.asRealStack(), true);
                // Simulated a partial stack insertion
                if (!resultStack.isEmpty()) {
                    final ListItem copyStack = toRemove.copy();
                    copyStack.setQuantity(resultStack.getCount());
                    partialToInventoryStacks.add(copyStack);
                }
                final ListItem copyStack = toRemove.copy();
                copyStack.setQuantity(toRemove.getQuantity() - resultStack.getCount());
                toInventoryStacks.add(copyStack);
            }
        }
    }
    // This may seem quite weird but we need to clear out the listing items reference for this player across the board to await what the results
    // are from the database to ensure we're 1:1 in sync. Otherwise, the Exchange would keep selling..
    axs.putListItemsFor(seller, null);
    axs.putForSaleItemsFor(seller, null);
    Sponge.getServer().getOnlinePlayers().stream().filter(p -> p.getUniqueId() != seller).forEach(p -> this.network.sendTo(p, new ClientboundForSaleFilterRequestPacket(axs.getId())));
    this.scheduler.createTaskBuilder().async().execute(() -> {
        try (final DSLContext context = this.databaseManager.createContext(true)) {
            final Iterator<VanillaStack> listingIter = toListingStacks.iterator();
            int index = 0;
            // New listing
            while (listingIter.hasNext()) {
                final VanillaStack stack = listingIter.next();
                final ItemStack realStack = stack.asRealStack();
                ListItem found = null;
                if (currentListItems != null) {
                    found = currentListItems.stream().filter(item -> ItemHandlerHelper.canItemStacksStack(realStack, item.asRealStack())).findAny().orElse(null);
                }
                if (found == null) {
                    final AxsListItemRecord itemRecord = ExchangeQueries.createInsertListItem(axs.getId(), Instant.now(), seller, realStack.getItem(), stack.getQuantity(), realStack.getMetadata(), index).build(context).fetchOne();
                    if (itemRecord == null) {
                        this.notificationManager.sendWindowMessage(player, Text.of("Exchange"), Text.of("Critical error encountered, check the server console for more details!"));
                        this.logger.error("Player '{}' submitted a new list item for exchange '{}' to the database but it failed. " + "Discarding changes and printing stack...", player.getName(), id);
                        this.printStacksToConsole(Lists.newArrayList(stack));
                        continue;
                    }
                    final NBTTagCompound compound = stack.getCompound();
                    if (compound == null) {
                        continue;
                    }
                    final AxsListItemDataRecord dataRecord = ExchangeQueries.createInsertItemData(itemRecord.getRecNo(), compound).build(context).fetchOne();
                    if (dataRecord == null) {
                        this.notificationManager.sendWindowMessage(player, Text.of("Exchange"), Text.of("Critical error encountered, check the server console for more details!"));
                        this.logger.error("Player '{}' submitted data for item record '{}' for exchange '{}' but it failed. " + "Discarding changes...", player.getName(), itemRecord.getRecNo(), id);
                        ExchangeQueries.createDeleteListItem(itemRecord.getRecNo()).build(context).execute();
                    }
                } else {
                    final int result = ExchangeQueries.createUpdateListItem(found.getRecord(), stack.getQuantity() + found.getQuantity(), index).build(context).execute();
                    if (result == 0) {
                        this.notificationManager.sendWindowMessage(player, Text.of("Exchange"), Text.of("Critical error encountered, check the server console for more details!"));
                        this.logger.error("Player '{}' attempted to add quantity to list item '{}' for exchange '{}' but it failed. " + "Discarding changes...", player.getName(), found.getRecord(), id);
                        listingIter.remove();
                    }
                }
                index++;
            }
            for (final ListItem next : toInventoryStacks) {
                ListItem existingStack = null;
                for (final ListItem stack : currentListItems) {
                    if (next.getRecord() == stack.getRecord()) {
                        existingStack = stack;
                        break;
                    }
                }
                final int diff = existingStack.getQuantity() - next.getQuantity();
                if (diff == 0) {
                    final int result = ExchangeQueries.createUpdateListItemIsHidden(next.getRecord(), true).build(context).execute();
                    if (result == 0) {
                        this.notificationManager.sendWindowMessage(player, Text.of("Exchange"), Text.of("Critical error encountered, check the server console for more details!"));
                        this.logger.error("Player '{}' attempted to remove list item '{}' for exchange '{}' but it failed. Discarding " + "changes...", player.getName(), next.getRecord(), id);
                    }
                // Update partial listings
                } else {
                    final int result = ExchangeQueries.createUpdateListItem(next.getRecord(), diff, next.getIndex()).build(context).execute();
                    if (result == 0) {
                        this.notificationManager.sendWindowMessage(player, Text.of("Exchange"), Text.of("Critical error encountered, check the server console for more details!"));
                        this.logger.error("Player '{}' removed quantity from list item '{}' for exchange '{}' to the database but it " + "failed. Discarding changes...", player.getName(), next.getRecord(), id);
                    }
                }
            }
            final Results listItemResults = ExchangeQueries.createFetchListItemsAndDataFor(seller, false).build(context).keepStatement(false).fetchMany();
            final Results forSaleItemResults = ExchangeQueries.createFetchForSaleItemsFor(seller, false).build(context).keepStatement(false).fetchMany();
            this.scheduler.createTaskBuilder().execute(() -> {
                final Player sellerPlayer = Sponge.getServer().getPlayer(seller).orElse(null);
                if (sellerPlayer != null) {
                    final IItemHandler inventory = ((EntityPlayerMP) sellerPlayer).getCapability(CapabilityItemHandler.ITEM_HANDLER_CAPABILITY, EnumFacing.UP);
                    // Add stacks from listings
                    for (final ListItem stack : toInventoryStacks) {
                        final ItemStack resultStack = ItemHandlerHelper.insertItemStacked(inventory, stack.asRealStack(), false);
                        if (!resultStack.isEmpty()) {
                        // TODO Their inventory changed since simulation. Best case scenario we toss it on the ground
                        }
                    }
                } else {
                // TODO They went offline on us. It is a very rare off-case. Half tempted to print what they should have got and let
                // TODO an admin deal with it
                }
                final List<ListItem> listItems = new ArrayList<>();
                listItemResults.forEach(result -> listItems.addAll(this.parseListItemsFrom(result)));
                final List<ForSaleItem> forSaleItems = new ArrayList<>();
                forSaleItemResults.forEach(result -> forSaleItems.addAll(this.parseForSaleItemsFrom(listItems, result)));
                // TODO Build a notification that says...
                // TODO  - Stacks requested to go to a listing but inventory can't fulfill it
                // TODO  - Stacks requested to go to the inventory but listings aren't found
                // TODO  - Stacks requested to go to the inventory but the listing couldn't fulfill it so we took what we could
                axs.putListItemsFor(seller, listItems);
                axs.putForSaleItemsFor(seller, forSaleItems);
                if (sellerPlayer != null) {
                    this.network.sendTo(sellerPlayer, new ClientboundListItemsResponsePacket(axs.getId(), listItems));
                    this.network.sendTo(sellerPlayer, new ClientboundListItemsSaleStatusPacket(axs.getId(), forSaleItems, null));
                }
                Sponge.getServer().getOnlinePlayers().forEach(p -> this.network.sendTo(p, new ClientboundForSaleFilterRequestPacket(axs.getId())));
            }).submit(this.container);
        } catch (SQLException | IOException e) {
            e.printStackTrace();
        }
    }).submit(this.container);
}