Implemented the Curse of Vanishing enchantment

server/item/enchantment/enchantment.go

@@ -0,0 +1,6 @@
+package enchantment
+
+// Curse represents an enchantment that may be a curse enchantment.
+type Curse interface {
+	Curse() bool
+}

server/item/enchantment/register.go

@@ -31,7 +31,7 @@ func init() {
 	// TODO: (25) Frost Walker.
 	item.RegisterEnchantment(26, Mending{})
 	// TODO: (27) Curse of Binding.
-	// TODO: (28) Curse of Vanishing.
+	item.RegisterEnchantment(28, Vanishing{})
 	// TODO: (29) Impaling.
 	// TODO: (30) Riptide.
 	// TODO: (31) Loyalty.

server/item/enchantment/vanishing.go

@@ -0,0 +1,55 @@
+package enchantment
+
+import (
+	"github.com/df-mc/dragonfly/server/item"
+	"github.com/df-mc/dragonfly/server/world"
+)
+
+// Vanishing is an enchantment that causes the item to disappear on death.
+type Vanishing struct{}
+
+// Name ...
+func (Vanishing) Name() string {
+	return "Vanishing"
+}
+
+// MaxLevel ...
+func (Vanishing) MaxLevel() int {
+	return 1
+}
+
+// Cost ...
+func (Vanishing) Cost(int) (int, int) {
+	return 25, 50
+}
+
+// Rarity ...
+func (Vanishing) Rarity() item.EnchantmentRarity {
+	return item.EnchantmentRarityVeryRare
+}
+
+// CompatibleWithEnchantment ...
+func (Vanishing) CompatibleWithEnchantment(t item.EnchantmentType) bool {
+	return true
+}
+
+// CompatibleWithItem ...
+func (Vanishing) CompatibleWithItem(i world.Item) bool {
+	_, dur := i.(item.Durable)
+	_, com := i.(item.Compass)
+	_, arm := i.(item.Armour)
+	// TODO: Recovery Compass
+	// TODO: Carrot on a Stick
+	// TODO: Warped Fungus on a Stick
+	return arm || com || dur
+}
+
+// Treasure ...
+func (Vanishing) Treasure() bool {
+	return true
+}
+
+// Treasure ...
+func (Vanishing) Curse() bool {
+	return true
+}

server/player/player.go

@@ -2,6 +2,13 @@ package player
 
 import (
 	"fmt"
+	"math"
+	"math/rand"
+	"net"
+	"strings"
+	"sync"
+	"time"
+
 	"github.com/df-mc/atomic"
 	"github.com/df-mc/dragonfly/server/block"
 	"github.com/df-mc/dragonfly/server/block/cube"
@@ -30,12 +37,6 @@ import (
 	"github.com/google/uuid"
 	"golang.org/x/exp/maps"
 	"golang.org/x/text/language"
-	"math"
-	"math/rand"
-	"net"
-	"strings"
-	"sync"
-	"time"
 )
 
 // Player is an implementation of a player entity. It has methods that implement the behaviour that players
@@ -914,6 +915,9 @@ func (p *Player) dropContents() {
 		w.AddEntity(orb)
 	}
 	for _, it := range append(p.inv.Items(), append(p.armour.Items(), p.offHand.Items()...)...) {
+		if _, ok := it.Enchantment(enchantment.Vanishing{}); ok {
+			continue
+		}
 		ent := entity.NewItem(it, pos)
 		ent.SetVelocity(mgl64.Vec3{rand.Float64()*0.2 - 0.1, 0.2, rand.Float64()*0.2 - 0.1})
 		w.AddEntity(ent)

server/session/handler_grindstone.go

@@ -2,13 +2,15 @@ package session
 
 import (
 	"fmt"
+	"math"
+	"math/rand"
+
 	"github.com/df-mc/dragonfly/server/block"
 	"github.com/df-mc/dragonfly/server/entity"
 	"github.com/df-mc/dragonfly/server/item"
+	"github.com/df-mc/dragonfly/server/item/enchantment"
 	"github.com/go-gl/mathgl/mgl64"
 	"github.com/sandertv/gophertunnel/minecraft/protocol"
-	"math"
-	"math/rand"
 )
 
 const (
@@ -78,7 +80,9 @@ func (h *ItemStackRequestHandler) handleGrindstoneCraft(s *Session) error {
 func experienceFromEnchantments(stack item.Stack) int {
 	var totalCost int
 	for _, enchant := range stack.Enchantments() {
-		// TODO: Don't include curses.
+		if _, ok := enchant.Type().(enchantment.Curse); ok {
+			continue
+		}
 		cost, _ := enchant.Type().Cost(enchant.Level())
 		totalCost += cost
 	}
@@ -94,7 +98,9 @@ func experienceFromEnchantments(stack item.Stack) int {
 // stripPossibleEnchantments strips all enchantments possible, excluding curses.
 func stripPossibleEnchantments(stack item.Stack) item.Stack {
 	for _, enchant := range stack.Enchantments() {
-		// TODO: Don't remove curses.
+		if _, ok := enchant.Type().(enchantment.Curse); ok {
+			continue
+		}
 		stack = stack.WithoutEnchantments(enchant.Type())
 	}
 	return stack

server/session/player.go

@@ -3,6 +3,11 @@ package session
 import (
 	"encoding/json"
 	"fmt"
+	"math"
+	"net"
+	"time"
+	_ "unsafe" // Imported for compiler directives.
+
 	"github.com/df-mc/atomic"
 	"github.com/df-mc/dragonfly/server/block"
 	"github.com/df-mc/dragonfly/server/entity"
@@ -19,10 +24,6 @@ import (
 	"github.com/google/uuid"
 	"github.com/sandertv/gophertunnel/minecraft/protocol"
 	"github.com/sandertv/gophertunnel/minecraft/protocol/packet"
-	"math"
-	"net"
-	"time"
-	_ "unsafe" // Imported for compiler directives.
 )
 
 // StopShowingEntity stops showing a world.Entity to the Session. It will be completely invisible until a call to