From dc76a6d496af9ddc68c958ca9b9ef5d31c4e58ac Mon Sep 17 00:00:00 2001
From: yarastqt <yarastqt@gmail.com>
Date: Mon, 13 Apr 2020 23:31:36 +0300
Subject: [PATCH] feat: support simple token

---
 src/core/flat-tokens.ts | 9 ++++++++-
 src/core/token.h.ts     | 4 ++--
 2 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/src/core/flat-tokens.ts b/src/core/flat-tokens.ts
index 90f077b..97f2015 100644
--- a/src/core/flat-tokens.ts
+++ b/src/core/flat-tokens.ts
@@ -11,7 +11,14 @@ export function flatTokens(tokens: TokensMap, __prefix__?: string): Shape<Flatte
   for (const key in tokens) {
     const transformedKey = withPrefix(key, __prefix__)
     const maybeToken = tokens[key]
-    if (maybeToken.value === undefined) {
+    if (typeof maybeToken === 'string') {
+      result[transformedKey] = {
+        value: maybeToken,
+        type: 'unknown',
+        name: transformedKey,
+      }
+    }
+    else if (maybeToken.value === undefined) {
       Object.assign(result, flatTokens(maybeToken as TokensMap, transformedKey))
     } else {
       result[transformedKey] = {
diff --git a/src/core/token.h.ts b/src/core/token.h.ts
index 2b4ad14..47b5962 100644
--- a/src/core/token.h.ts
+++ b/src/core/token.h.ts
@@ -2,7 +2,7 @@ export type Shape<T> = { [key: string]: T }
 
 export type Token = {
   value: string | number
-  type: 'color' | 'size'
+  type: 'color' | 'size' | 'unknown'
   comment?: string
 }
 
@@ -11,5 +11,5 @@ export type FlattenToken = Token & {
 }
 
 export type TokensMap = {
-  [key: string]: Token | TokensMap
+  [key: string]: TokensMap | Token | string
 }