kmpdfkit_demo/PDFViewer/Controller/CPDFViewController.m

//
//  CPDFViewController.m
//  PDFViewer
//
//  Created by kdan on 2022/11/15.
//

#import <ComPDFKit/ComPDFKit.h>
#import <ComPDFKit/CPDFHeaderFooter.h>
#import "CPDFViewController.h"
#import "CPDFHeaderFooterAddController.h"
#import "CPDFBatesAddViewController.h"
#import "CPDFSettingView.h"

@interface CPDFViewController () <UIActionSheetDelegate,CPDFModelDataDelegate>

@property (nonatomic,strong) CPDFKitPlatformImage *image;
@property (nonatomic,strong) CPDFDocument *document;
@property (nonatomic,strong) CPDFView *pdfView;
@property (nonatomic,strong) CPDFSettingView *headerFooterSetting;
@property (nonatomic,strong) CPDFSettingView *batesSetting;

@end

@implementation CPDFViewController

- (instancetype)initWithPath:(NSString *)inPath {
    if (self = [super init]) {
        _path = inPath;
    }
    return self;
}

- (void)viewDidLoad {
    [super viewDidLoad];
    // Do any additional setup after loading the view.
    // Add UIBarButtonItem in Navigation
    self.navigationController.toolbarHidden = YES;
    UIImage *btnMore = [UIImage imageNamed:@"btn_more"];
    UIBarButtonItem *ringhtBarItem = [[UIBarButtonItem alloc] initWithImage:btnMore style:UIBarButtonItemStylePlain target:self action:@selector(onClickedOkbtn)];
    self.navigationItem.rightBarButtonItem = ringhtBarItem;
    
    // Setting URL
    NSURL *url = [NSURL fileURLWithPath:_path];
    _document = [[CPDFDocument alloc] initWithURL:url];
    
    // Get document frist image
    CPDFPage *page = [_document pageAtIndex:0];
    CGSize size = [_document pageSizeAtIndex:0];
    _image = [page thumbnailOfSize:size];
    
    // Get document view
    _pdfView = [[CPDFView alloc] initWithFrame:self.view.bounds];
    _pdfView.document = _document;
    _pdfView.autoresizingMask = UIViewAutoresizingFlexibleWidth | UIViewAutoresizingFlexibleHeight;
    
   [self.view addSubview:_pdfView];
}

#pragma mark - CPDFViewController Actions

- (void)onClickedOkbtn {
    UIAlertController *alertController = [UIAlertController alertControllerWithTitle:nil message:nil preferredStyle:UIAlertControllerStyleActionSheet];
    
    // Add watermark action
    UIAlertAction *addWatermarkAction = [UIAlertAction actionWithTitle:@"Add Watermark" style:UIAlertActionStyleDefault handler:^(UIAlertAction * _Nonnull action) {
            NSLog(@"addWatermarkAction");
        }];
    
    // Setting headerfooter action
    UIAlertAction *settingHeaderFooter = [UIAlertAction actionWithTitle:@"Setting Headerfooter" style:UIAlertActionStyleDefault handler:^(UIAlertAction * _Nonnull action) {
        [self settingHeaderFooter];
       }];
    
    // Seeting bates action
    UIAlertAction *settingBatesAction = [UIAlertAction actionWithTitle:@"Setting Bates" style:UIAlertActionStyleDefault handler:^(UIAlertAction * _Nonnull action) {
        [self settingBates];
       }];
    
    // Cancel action
    UIAlertAction *cancelAction = [UIAlertAction actionWithTitle:@"Cancel" style:UIAlertActionStyleCancel handler:^(UIAlertAction * _Nonnull action) {
            NSLog(@"Cancel Action");
        }];
    
    // Add controller
    [alertController addAction:addWatermarkAction];
    [alertController addAction:settingHeaderFooter];
    [alertController addAction:settingBatesAction];
    [alertController addAction:cancelAction];
    
    [self presentViewController:alertController animated:YES completion:nil];
}

#pragma mark - Setting Pop

- (void)settingHeaderFooter {
    _headerFooterSetting = [[CPDFSettingView alloc] init];
    [_headerFooterSetting setText];
    [self.view addSubview:_headerFooterSetting];
    
    [_headerFooterSetting mas_makeConstraints:^(MASConstraintMaker *make) {
        make.bottom.equalTo(self.view.mas_bottom);
        make.left.equalTo(self.view.mas_left);
        make.right.equalTo(self.view.mas_right);
        make.height.mas_equalTo(100);
    }];
    
    [UIView animateWithDuration:0.3 delay:0 options:UIViewAnimationOptionCurveEaseIn animations:^ {
        self.headerFooterSetting.center = CGPointMake(self.headerFooterSetting.center.x, self.headerFooterSetting.center.y - 100);
    } completion:nil];
    
    [_headerFooterSetting.addButton addTarget:self action:@selector(addHeaderFooter:) forControlEvents:UIControlEventTouchUpInside];
    [_headerFooterSetting.deleteButton addTarget:self action:@selector(deleteHeaderFooter:) forControlEvents:UIControlEventTouchUpInside];
}

- (void)settingBates {
    _batesSetting = [[CPDFSettingView alloc] init];
    [_batesSetting setText];
    [self.view addSubview:_batesSetting];
    
    [_batesSetting mas_makeConstraints:^(MASConstraintMaker *make) {
        make.bottom.equalTo(self.view.mas_bottom);
        make.left.equalTo(self.view.mas_left);
        make.right.equalTo(self.view.mas_right);
        make.height.mas_equalTo(100);
    }];
    
    [UIView animateWithDuration:0.3 delay:0 options:UIViewAnimationOptionCurveEaseIn animations:^ {
        self.batesSetting.center = CGPointMake(self.batesSetting.center.x, self.batesSetting.center.y - 100);
    } completion:nil];
    
    [_batesSetting.addButton addTarget:self action:@selector(addBates:) forControlEvents:UIControlEventTouchUpInside];
    [_batesSetting.deleteButton addTarget:self action:@selector(deleteBates:) forControlEvents:UIControlEventTouchUpInside];
}

#pragma mark - Actions

- (void)addHeaderFooter:(UIButton *)btn {
    CPDFHeaderFooterAddController *headerFooterControl = [[CPDFHeaderFooterAddController alloc] initWithImage:self.image];

    headerFooterControl.delegate = self;

    [self.navigationController pushViewController:headerFooterControl animated:NO];
    
    [UIView animateWithDuration:0.3 delay:0 options:UIViewAnimationOptionCurveEaseIn animations:^ {
        self.headerFooterSetting.center = CGPointMake(self.headerFooterSetting.center.x, self.headerFooterSetting.center.y + 100);
    } completion:nil];
}

- (void)deleteHeaderFooter:(UIButton *)btn {
    [self deleteHeaderFooterAction];
    
    [UIView animateWithDuration:0.3 delay:0 options:UIViewAnimationOptionCurveEaseIn animations:^ {
        self.headerFooterSetting.center = CGPointMake(self.headerFooterSetting.center.x, self.headerFooterSetting.center.y + 100);
    } completion:nil];
}

- (void)addBates:(UIButton *)btn {
    CPDFBatesAddViewController *batesAddControl = [[CPDFBatesAddViewController alloc] initWithImage:self.image];
    
    batesAddControl.delegate = self;
    
    [self.navigationController pushViewController:batesAddControl animated:NO];
    
    [UIView animateWithDuration:0.3 delay:0 options:UIViewAnimationOptionCurveEaseIn animations:^ {
        self.batesSetting.center = CGPointMake(self.batesSetting.center.x, self.batesSetting.center.y + 100);
    } completion:nil];
}

- (void)deleteBates:(UIButton *)btn {
    [self deleteBatesAction];
    
    [UIView animateWithDuration:0.3 delay:0 options:UIViewAnimationOptionCurveEaseIn animations:^ {
        self.batesSetting.center = CGPointMake(self.batesSetting.center.x, self.batesSetting.center.y + 100);
    } completion:nil];
}

#pragma mark - Delete Alert

// Detele headerfootr action
- (void)deleteHeaderFooterAction {
    UIAlertController *alertController = [UIAlertController alertControllerWithTitle:@"Delete Page Number" message:@"Are you sure delete all pages number" preferredStyle:UIAlertControllerStyleAlert];
    
    UIAlertAction *okAction = [UIAlertAction actionWithTitle:@"OK" style:UIAlertActionStyleDefault handler:^(UIAlertAction * _Nonnull action) {
        [self deleteHeaderFooter];
        }];
    
    UIAlertAction *cancelAction = [UIAlertAction actionWithTitle:@"Cancel" style:UIAlertActionStyleCancel handler:^(UIAlertAction * _Nonnull action) {
        NSLog(@"Cancel Action");
        }];
    
    [alertController addAction:okAction];
    [alertController addAction:cancelAction];
    
    [self presentViewController:alertController animated:YES completion:nil];
}

// Delete bates action
- (void)deleteBatesAction {
    UIAlertController *alertController = [UIAlertController alertControllerWithTitle:@"Delete Bates" message:@"Are you sure delete all Bates" preferredStyle:UIAlertControllerStyleAlert];
    
    UIAlertAction *okAction = [UIAlertAction actionWithTitle:@"OK" style:UIAlertActionStyleDefault handler:^(UIAlertAction * _Nonnull action) {
        
        [self deleteBates];
        }];
    
    UIAlertAction *cancelAction = [UIAlertAction actionWithTitle:@"Cancel" style:UIAlertActionStyleCancel handler:^(UIAlertAction * _Nonnull action) {
        NSLog(@"Cancel Action");
        }];
    
    [alertController addAction:okAction];
    [alertController addAction:cancelAction];
    
    [self presentViewController:alertController animated:YES completion:nil];
}

#pragma mark - CPDFModelDataDelegate

// Get headerfooter' model to setting page headerfooter's add functional property
- (void)changeHeaderFooterModelData:(CPDFHeaderFooterModel *)modelData {
    CPDFHeaderFooter *headerFooter = [self.document headerFooter];

    NSString *pageIndex = [NSString stringWithFormat:@"0-%lu",self.document.pageCount - 1];
    NSString *text = [[NSString alloc] init];

    switch (modelData.fontSelcet) {
        case 0:
            text = [NSString stringWithFormat:@"<<%@>>",modelData.pageStart];
            break;
        case 1:
            text = [NSString stringWithFormat:@"page <<%@>>",modelData.pageStart];
            break;
        case 2:
            text = [NSString stringWithFormat:@"<<%@>>/%lu",modelData.pageStart,self.document.pageCount];
            break;
        case 3:
            text = [NSString stringWithFormat:@"<<%@>> of %lu",modelData.pageStart,self.document.pageCount];
            break;
        case 4:
            text = [NSString stringWithFormat:@"page <<%@>> of %lu",modelData.pageStart,self.document.pageCount];
            break;
        default:
            break;
    }

    headerFooter.pageString = pageIndex;
    [headerFooter setText:text atIndex:modelData.fontPosition];
    [headerFooter setFontSize:modelData.fontSize atIndex:modelData.fontPosition];
    [headerFooter setTextColor:modelData.fontColor atIndex:modelData.fontPosition];
    [headerFooter update];

    NSURL *url = [NSURL fileURLWithPath:self.path];
    [self.document writeToURL:url];
    
    _document = [[CPDFDocument alloc] initWithURL:url];
    _pdfView = [[CPDFView alloc] initWithFrame:self.view.bounds];
    _pdfView.document = _document;
    _pdfView.autoresizingMask = UIViewAutoresizingFlexibleWidth | UIViewAutoresizingFlexibleHeight;

   [self.view addSubview:_pdfView];
}

// Get bates' model to setting page bates's add functional property
- (void)changBatesModelData:(CPDFHeaderFooterModel *)modelData {
    CPDFHeaderFooterModel *modelBatesData = modelData;
    
    CPDFBates *bates = [self.document bates];
    
    NSString *pageIndex = [NSString stringWithFormat:@"%@-%lu",modelBatesData.pageStart,self.document.pageCount - 1];
    bates.pageString = pageIndex;
    NSString *text = [NSString stringWithFormat:@"<<%@>>",modelBatesData.fontText];
    [bates setText:text atIndex:modelBatesData.fontPosition];
    [bates setFontName:modelBatesData.fontName atIndex:modelBatesData.fontPosition];
    [bates setFontSize:15.0 atIndex:modelBatesData.fontPosition];
    [bates setTextColor:modelBatesData.fontColor atIndex:modelBatesData.fontPosition];
    [bates update];
    
    NSURL *url = [NSURL fileURLWithPath:self.path];
    [self.document writeToURL:url];
    
    _document = [[CPDFDocument alloc] initWithURL:url];
    _pdfView = [[CPDFView alloc] initWithFrame:self.view.bounds];
    _pdfView.document = _document;
    _pdfView.autoresizingMask = UIViewAutoresizingFlexibleWidth | UIViewAutoresizingFlexibleHeight;
    
   [self.view addSubview:_pdfView];
}

#pragma mark - Delete Methods

// Get headerfooter' model to setting page headerfooter's delete functional property
- (void)deleteHeaderFooter {
    CPDFHeaderFooter *headerFooter = [self.document headerFooter];
    [headerFooter clear];
    
    NSURL *url = [NSURL fileURLWithPath:self.path];
    [self.document writeToURL:url];
    _document = [[CPDFDocument alloc] initWithURL:url];
    _pdfView = [[CPDFView alloc] initWithFrame:self.view.bounds];
    _pdfView.document = _document;
    _pdfView.autoresizingMask = UIViewAutoresizingFlexibleWidth | UIViewAutoresizingFlexibleHeight;
    
   [self.view addSubview:_pdfView];
}

// Get bates' model to setting page bates's delete functional property
- (void)deleteBates {
    CPDFBates *bates = [self.document bates];
    [bates clear];
    
    NSURL *url = [NSURL fileURLWithPath:self.path];
    [self.document writeToURL:url];
    _document = [[CPDFDocument alloc] initWithURL:url];
    _pdfView = [[CPDFView alloc] initWithFrame:self.view.bounds];
    _pdfView.document = _document;
    _pdfView.autoresizingMask = UIViewAutoresizingFlexibleWidth | UIViewAutoresizingFlexibleHeight;
    
   [self.view addSubview:_pdfView];
}

@end