Source code
Revision control
Copy as Markdown
Other Tools
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
// eslint-disable-next-line import/no-unresolved
import { getReferences, usesReferences } from "style-dictionary/utils";
import { COLLECTIONS } from "./constants.mjs";
import { isNestedDefaultObject } from "./is-nested-default-object.mjs";
export const getTokenCollection = (
token,
dictionary,
overrideIdentifier = ""
) => {
if (!overrideIdentifier || !token.original.value[overrideIdentifier]?.value) {
return token.attributes?.collection;
}
let collection = COLLECTIONS.theme;
let originalValue = token.original.value[overrideIdentifier].value;
if (
(typeof originalValue !== "object" ||
isNestedDefaultObject(originalValue)) &&
originalValue !== "currentColor"
) {
if (token.path[0] === "color") {
collection = COLLECTIONS.colors;
} else {
collection = COLLECTIONS.primitives;
}
// if any referenced tokens are theme tokens, categorize this as a theme token
if (usesReferences(originalValue)) {
if (
getReferences(originalValue, dictionary.tokens).some(
reference => reference.attributes?.collection === COLLECTIONS.theme
)
) {
collection = COLLECTIONS.theme;
}
}
}
return collection;
};