/*
 * Copyright (c) Facebook, Inc. and its affiliates.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 */

#import "SKSearchResultNode.h"

@implementation SKSearchResultNode {
  NSString *_nodeId;
  BOOL _isMatch;
  NSDictionary *_element;
  NSArray<SKSearchResultNode *> *_children;
}

- (instancetype)initWithNode:(NSString *)nodeId
                     asMatch:(BOOL)isMatch
                 withElement:(NSDictionary *)element
                 andChildren:(NSArray<SKSearchResultNode *> *)children {
  self = [super init];
  if (self) {
      _nodeId = nodeId;
      _isMatch = isMatch;
      _element = element;
      _children = children;
  }
  return self;
}

- (NSDictionary *)toNSDictionary {
  if (_element == nil) {
      return nil;
  }
  NSMutableArray<NSDictionary *> *childArray;
  if (_children) {
    childArray = [NSMutableArray new];
    for (SKSearchResultNode *child in _children) {
      NSDictionary *childDict = [child toNSDictionary];
      if (childDict) {
          [childArray addObject:childDict];
      }
    }
  } else {
    childArray = nil;
  }
  return @{
          @"id": _nodeId,
          @"isMatch": @(_isMatch),
          @"element": _element,
          @"children": childArray ?: [NSNull null]
          };
}

@end