FilesExpand file tree

 master

/

TreeTransformer.java

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

61 lines (44 loc) · 1.95 KB

 master

/

TreeTransformer.java

Copy path

Top

File metadata and controls

• Code
• Blame

61 lines (44 loc) · 1.95 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

package graphql.util;

import graphql.PublicApi;

import graphql.collect.ImmutableKit;

import java.util.Collections;

import java.util.LinkedList;

import java.util.List;

import java.util.Map;

import static graphql.Assert.assertNotNull;

@PublicApi

public class TreeTransformer<T> {

private final NodeAdapter<T> nodeAdapter;

public TreeTransformer(NodeAdapter<T> nodeAdapter) {

this.nodeAdapter = nodeAdapter;

}

public T transform(T root, TraverserVisitor<T> traverserVisitor) {

return transform(root, traverserVisitor, ImmutableKit.emptyMap());

}

public T transform(T root, TraverserVisitor<T> traverserVisitor, Map<Class<?>, Object> rootVars) {

assertNotNull(root);

TraverserVisitor<T> nodeTraverserVisitor = new TraverserVisitor<T>() {

@Override

public TraversalControl enter(TraverserContext<T> context) {

NodeZipper<T> nodeZipper = new NodeZipper<>(context.thisNode(), context.getBreadcrumbs(), nodeAdapter);

context.setVar(NodeZipper.class, nodeZipper);

context.setVar(NodeAdapter.class, nodeAdapter);

return traverserVisitor.enter(context);

}

@Override

public TraversalControl leave(TraverserContext<T> context) {

return traverserVisitor.leave(context);

}

@Override

public TraversalControl backRef(TraverserContext<T> context) {

return traverserVisitor.backRef(context);

}

};

List<NodeZipper<T>> zippers = new LinkedList<>();

Traverser<T> traverser = Traverser.depthFirstWithNamedChildren(nodeAdapter::getNamedChildren, zippers, null);

traverser.rootVars(rootVars);

traverser.traverse(root, nodeTraverserVisitor);

NodeMultiZipper<T> multiZipper = NodeMultiZipper.newNodeMultiZipperTrusted(root, zippers, nodeAdapter);

return multiZipper.toRootNode();

}

}