@@ -3,7 +3,7 @@ import { node, Query, relation } from 'cypher-query-builder';
3
3
import { RelationDirection } from 'cypher-query-builder/dist/typings/clauses/relation-pattern' ;
4
4
import { Maybe as Nullable } from 'graphql/jsutils/Maybe' ;
5
5
import { DateTime } from 'luxon' ;
6
- import { ID , many , ResourceShape } from '~/common' ;
6
+ import { EnhancedResource , ID , many , ResourceShape } from '~/common' ;
7
7
import { ResourceMap } from '~/core' ;
8
8
import { Variable } from '../query-augmentation/condition-variables' ;
9
9
@@ -74,19 +74,20 @@ type AnyDirectionalDefinition = Partial<
74
74
* // Note how `user` is imported, not matched, and not returned.
75
75
*/
76
76
export function createRelationships < TResourceStatic extends ResourceShape < any > > (
77
- resource : TResourceStatic ,
77
+ resource : TResourceStatic | EnhancedResource < TResourceStatic > ,
78
78
direction : RelationDirection ,
79
79
labelsToRelationships : RelationshipDefinition ,
80
80
) : ( query : Query ) => Query ;
81
81
export function createRelationships < TResourceStatic extends ResourceShape < any > > (
82
- resource : TResourceStatic ,
82
+ resource : TResourceStatic | EnhancedResource < TResourceStatic > ,
83
83
definition : AnyDirectionalDefinition ,
84
84
) : ( query : Query ) => Query ;
85
85
export function createRelationships < TResourceStatic extends ResourceShape < any > > (
86
- resource : TResourceStatic ,
86
+ resource : TResourceStatic | EnhancedResource < TResourceStatic > ,
87
87
directionOrDefinition : RelationDirection | AnyDirectionalDefinition ,
88
88
maybeLabelsToRelationships ?: RelationshipDefinition ,
89
89
) {
90
+ resource = EnhancedResource . of ( resource ) ;
90
91
const normalizedArgs =
91
92
typeof directionOrDefinition === 'string'
92
93
? { [ directionOrDefinition ] : maybeLabelsToRelationships }
0 commit comments