/*
 * 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 "IGTestStoryboardSupplementarySource.h"

#import "IGTestStoryboardSupplementaryView.h"

@implementation IGTestStoryboardSupplementarySource

- (UICollectionReusableView *)viewForSupplementaryElementOfKind:(NSString *)elementKind
                                                        atIndex:(NSInteger)index {
    IGTestStoryboardSupplementaryView *view = [self.collectionContext dequeueReusableSupplementaryViewFromStoryboardOfKind:elementKind
                                                                                                            withIdentifier:@"IGTestStoryboardSupplementaryView"
                                                                                                      forSectionController:self.sectionController
                                                                                                                   atIndex:index];
    view.label.text = @"Header";
    return view;
}

- (CGSize)sizeForSupplementaryViewOfKind:(NSString *)elementKind atIndex:(NSInteger)index {
    return CGSizeMake([self.collectionContext containerSize].width, 45);
}

@end
