code / tomo

   1 // Logic for formatting types
   2 
   3 #include "../ast.h"
   4 #include "../stdlib/datatypes.h"
   5 #include "../stdlib/stdlib.h"
   6 #include "../stdlib/text.h"
   7 #include "args.h"
   8 #include "formatter.h"
   9 
  10 Text_t format_type(type_ast_t *type) {
  11     switch (type->tag) {
  12     case VarTypeAST: return Text$from_str(Match(type, VarTypeAST)->name);
  13     case PointerTypeAST: {
  14         DeclareMatch(ptr, type, PointerTypeAST);
  15         return Texts(ptr->is_stack ? Text("&") : Text("@"), format_type(ptr->pointed));
  16     }
  17     case ListTypeAST: {
  18         return Texts("[", format_type(Match(type, ListTypeAST)->item), "]");
  19     }
  20     case TableTypeAST: {
  21         DeclareMatch(table, type, TableTypeAST);
  22         Text_t code = Texts("{", format_type(table->key));
  23         if (table->value != NULL) code = Texts(code, ":", format_type(table->value));
  24         if (table->default_value) {
  25             OptionalText_t val = format_inline_code(table->default_value, (Table_t){});
  26             assert(val.tag != TEXT_NONE);
  27             code = Texts(code, "=", val);
  28         }
  29         return Texts(code, "}");
  30     }
  31     case FunctionTypeAST: {
  32         DeclareMatch(func, type, FunctionTypeAST);
  33         Text_t code = Texts("func(", format_inline_args(func->args, (Table_t){}));
  34         if (func->ret) code = Texts(code, func->args ? Text(" -> ") : Text("-> "), format_type(func->ret));
  35         return Texts(code, ")");
  36     }
  37     case OptionalTypeAST: {
  38         return Texts(format_type(Match(type, OptionalTypeAST)->type), "?");
  39     }
  40     case UnknownTypeAST:
  41     default: fail("Invalid Type AST");
  42     }
  43 }