From 3648d63547e36c32a34b743d73190e7e1f60ab3c Mon Sep 17 00:00:00 2001
From: Andrey Sobolev <haiodo@users.noreply.github.com>
Date: Mon, 6 Dec 2021 22:16:09 +0700
Subject: [PATCH] Fix model (#546)

Signed-off-by: Andrey Sobolev <haiodo@gmail.com>
---
 models/core/src/core.ts  | 12 +++++++++++-
 models/core/src/index.ts |  4 +++-
 2 files changed, 14 insertions(+), 2 deletions(-)

diff --git a/models/core/src/core.ts b/models/core/src/core.ts
index 1de84914cd..7514508c19 100644
--- a/models/core/src/core.ts
+++ b/models/core/src/core.ts
@@ -14,7 +14,7 @@
 //
 
 import type { IntlString } from '@anticrm/platform'
-import type { Account, AnyAttribute, AttachedDoc, Class, ClassifierKind, Doc, Domain, Mixin, Obj, Ref, Space, Timestamp, Type } from '@anticrm/core'
+import type { Account, AnyAttribute, AttachedDoc, Class, ClassifierKind, Doc, Domain, Mixin, Obj, Ref, Space, Timestamp, Type, Collection, RefTo } from '@anticrm/core'
 import { DOMAIN_MODEL } from '@anticrm/core'
 import { Model, Prop, TypeTimestamp } from '@anticrm/model'
 import core from './component'
@@ -70,6 +70,16 @@ export class TType extends TObj implements Type<any> {
 @Model(core.class.TypeString, core.class.Type)
 export class TTypeString extends TType {}
 
+@Model(core.class.RefTo, core.class.Type)
+export class TRefTo extends TType implements RefTo<Doc> {
+  to!: Ref<Class<Doc>>
+}
+
+@Model(core.class.Collection, core.class.Type)
+export class TCollection extends TType implements Collection<AttachedDoc> {
+  of!: Ref<Class<Doc>>
+}
+
 @Model(core.class.TypeBoolean, core.class.Type)
 export class TTypeBoolean extends TType {}
 
diff --git a/models/core/src/index.ts b/models/core/src/index.ts
index f998dc7726..9bc93b3516 100644
--- a/models/core/src/index.ts
+++ b/models/core/src/index.ts
@@ -15,7 +15,7 @@
 
 import { Builder } from '@anticrm/model'
 import core from './component'
-import { TAttribute, TClass, TDoc, TMixin, TObj, TType, TTypeString, TTypeBoolean, TTypeTimestamp, TTypeDate, TAttachedDoc } from './core'
+import { TAttribute, TClass, TDoc, TMixin, TObj, TType, TTypeString, TTypeBoolean, TTypeTimestamp, TTypeDate, TAttachedDoc, TCollection, TRefTo } from './core'
 import { TSpace, TAccount, TState, TSpaceWithStates, TDocWithState } from './security'
 import { TTx, TTxCreateDoc, TTxMixin, TTxUpdateDoc, TTxCUD, TTxPutBag, TTxRemoveDoc, TTxBulkWrite, TTxCollectionCUD } from './tx'
 
@@ -49,6 +49,8 @@ export function createModel (builder: Builder): void {
     TTypeString,
     TTypeBoolean,
     TTypeTimestamp,
+    TRefTo,
+    TCollection,
     TTypeDate,
     TState
   )