✨ Add translation_key mcdoc attribute

SpyglassMC · Dec 14, 2024 · af28eac · af28eac
1 parent 1bfd90e
commit af28eac
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/packages/core/src/symbol/Symbol.ts b/packages/core/src/symbol/Symbol.ts
@@ -238,6 +238,7 @@ export const AssetsMiscCategories = Object.freeze(
 	[
 		'texture_slot',
 		'shader_target',
+		'translation_key',
 	] as const,
 )
 export type AssetsMiscCategory = (typeof AssetsMiscCategories)[number]

diff --git a/packages/java-edition/src/json/mcdocAttributes.ts b/packages/java-edition/src/json/mcdocAttributes.ts
@@ -23,6 +23,13 @@ const textureSlotValidator = validator.alternatives<TextureSlotConfig>(
 	() => ({ kind: 'value' }),
 )
 
+const translationKeyValidator = validator.alternatives(
+	validator.tree({
+		definition: validator.boolean,
+	}),
+	() => ({ definition: false }),
+)
+
 export function registerMcdocAttributes(meta: core.MetaRegistry) {
 	mcdoc.runtime.registerAttribute(meta, 'criterion', criterionValidator, {
 		stringParser: (config, _, ctx) => {
@@ -62,4 +69,18 @@ export function registerMcdocAttributes(meta: core.MetaRegistry) {
 			} satisfies TextureSlotNode
 		},
 	})
+	mcdoc.runtime.registerAttribute(meta, 'translation_key', translationKeyValidator, {
+		stringParser: (config, _, ctx) => {
+			return core.symbol({
+				category: 'translation_key',
+				usageType: config.definition ? 'definition' : 'reference',
+			})
+		},
+		stringMocker: (config, _, ctx) => {
+			return core.SymbolNode.mock(ctx.offset, {
+				category: 'translation_key',
+				usageType: config.definition ? 'definition' : 'reference',
+			})
+		},
+	})
 }