From 259efc20ab9521af646f29321ecc260d17440a44 Mon Sep 17 00:00:00 2001
From: Pavel Shevaev <pacha.shevaev@gmail.com>
Date: Fri, 14 Jul 2023 16:21:21 +0300
Subject: [PATCH] Adding experimental support for enums overriding using
 @enum_override token

---
 metagen.inc.php |  8 +++++++-
 parser.inc.php  | 29 +++++++++++++++++++++++++----
 2 files changed, 32 insertions(+), 5 deletions(-)

diff --git a/metagen.inc.php b/metagen.inc.php
index f08e3b6..a3fd95f 100644
--- a/metagen.inc.php
+++ b/metagen.inc.php
@@ -706,7 +706,7 @@ class mtgMetaEnum extends mtgUserType
   function calcOrValue($or_keys)
   {
     $res = 0;
-    foreach ($or_keys as $value_name)
+    foreach($or_keys as $value_name)
     {
       if(!isset($this->values[$value_name]))
         throw new Exception("Enum '{$this->name}' has no value '{$value_name}'");
@@ -726,6 +726,12 @@ class mtgMetaEnum extends mtgUserType
   {
     return $this->values;
   }
+
+  function override(mtgMetaEnum $other)
+  {
+    $this->tokens = array_merge($this->tokens, $other->getTokens());
+    $this->values = array_merge($this->values, $other->getValues());
+  }
 } 
 
 function mtg_get_all_fields(mtgMetaStruct $struct)
diff --git a/parser.inc.php b/parser.inc.php
index 077b880..9495446 100644
--- a/parser.inc.php
+++ b/parser.inc.php
@@ -45,6 +45,8 @@ class mtgMetaInfoParser
 
   function __construct($config = array())
   {
+    self::addDefaultTokens($config);
+
     $this->config = $config;
     if(!isset($this->config['include_path']))
       $this->config['include_path'] = array('.');
@@ -82,6 +84,14 @@ class mtgMetaInfoParser
     $this->token_strs[self::T_Func] = '<func>';
   }
 
+  function addDefaultTokens(array &$config)
+  {
+    if(!isset($config['valid_tokens']))
+      $config['valid_tokens'] = array();
+
+    $config['valid_tokens'][] = 'enum_override';
+  }
+
   function parse(mtgMetaInfo $meta, $raw_file)
   {
     $this->current_meta = $meta;
@@ -306,12 +316,8 @@ class mtgMetaInfoParser
     $name = $this->_parseDotName();
 
     $enum = new mtgMetaEnum($name);
-    $this->current_meta->addUnit(new mtgMetaInfoUnit($this->file, $enum));
-
     if($this->token == self::T_Prop)
-    {
       $enum->setTokens($this->_parsePropTokens());
-    }
 
     $or_values = array();
     while(true)
@@ -331,6 +337,21 @@ class mtgMetaInfoParser
       }
     }
     $enum->addOrValues($or_values);
+
+    //NOTE: special case for enums when we allow to 'override' the original one, 
+    //      with additional values 
+    if($enum->hasToken('enum_override'))
+    {
+      $existing = $this->current_meta->findUnit($enum->getId());
+      if(!$existing)
+        throw new Exception("Not found '{$name}' enum to override");
+      if(!($existing->object instanceof mtgMetaEnum))
+        throw new Exception("Not an enum struct '{$name}'");
+
+      $existing->object->override($enum);
+    }
+    else
+      $this->current_meta->addUnit(new mtgMetaInfoUnit($this->file, $enum));
   }
 
   private function _parseFields($next_doer)