Skip to content

Commit c047b49

Browse files
committed
Change createRelationships to accept EnhancedResource
1 parent b28d962 commit c047b49

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

src/core/database/query/create-relationships.ts

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ import { node, Query, relation } from 'cypher-query-builder';
33
import { RelationDirection } from 'cypher-query-builder/dist/typings/clauses/relation-pattern';
44
import { Maybe as Nullable } from 'graphql/jsutils/Maybe';
55
import { DateTime } from 'luxon';
6-
import { ID, many, ResourceShape } from '~/common';
6+
import { EnhancedResource, ID, many, ResourceShape } from '~/common';
77
import { ResourceMap } from '~/core';
88
import { Variable } from '../query-augmentation/condition-variables';
99

@@ -74,19 +74,20 @@ type AnyDirectionalDefinition = Partial<
7474
* // Note how `user` is imported, not matched, and not returned.
7575
*/
7676
export function createRelationships<TResourceStatic extends ResourceShape<any>>(
77-
resource: TResourceStatic,
77+
resource: TResourceStatic | EnhancedResource<TResourceStatic>,
7878
direction: RelationDirection,
7979
labelsToRelationships: RelationshipDefinition,
8080
): (query: Query) => Query;
8181
export function createRelationships<TResourceStatic extends ResourceShape<any>>(
82-
resource: TResourceStatic,
82+
resource: TResourceStatic | EnhancedResource<TResourceStatic>,
8383
definition: AnyDirectionalDefinition,
8484
): (query: Query) => Query;
8585
export function createRelationships<TResourceStatic extends ResourceShape<any>>(
86-
resource: TResourceStatic,
86+
resource: TResourceStatic | EnhancedResource<TResourceStatic>,
8787
directionOrDefinition: RelationDirection | AnyDirectionalDefinition,
8888
maybeLabelsToRelationships?: RelationshipDefinition,
8989
) {
90+
resource = EnhancedResource.of(resource);
9091
const normalizedArgs =
9192
typeof directionOrDefinition === 'string'
9293
? { [directionOrDefinition]: maybeLabelsToRelationships }

0 commit comments

Comments
 (0)