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 "CPDFHeaderFooterDeleteController.h"
#import "CPDFBatesSettingController.h"
#import "CPDFHeaderFooterSettingController.h"
#import "CPDFBatesModel.h"

@interface CPDFViewController () <UIActionSheetDelegate>

@property (nonatomic,strong) CPDFKitPlatformImage *image;
@property (nonatomic,assign) CGSize size;
@property (nonatomic,strong) CPDFHeaderFooter *headerFooter;
@property (nonatomic,strong) CPDFBates *bates;
@property (nonatomic,strong) CPDFDocument *document;
@property (nonatomic,strong) CPDFView *pdfView;
@property (nonatomic,strong) CPDFHeaderFooterSettingController *headefooterControl;
@property (nonatomic,strong) CPDFHeaderFooterModel *modelData;
@property (nonatomic,strong) CPDFBatesModel *modelBatesData;
@property (nonatomic,strong) CPDFBatesSettingController *batesBatesControl;

@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];
    _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];
    [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(getHeaderFooterModel:) name:@"getAddHeaderFooterModel" object:nil];
    [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(deleteHeaderFooter:) name:@"deleteHeaderFooter" object:nil];
    [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(getBatesMdel:) name:@"getAddBatesModel" object:nil];
    [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(deleteBates:) name:@"deleteBatesModel" object:nil];
}

#pragma mark - click event of CPDFViewController's ringhtBarItem

- (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.headefooterControl = [[CPDFHeaderFooterSettingController alloc] initWithIamge:self.image WithSize:self.size WithDocument:self.document WithView:self.pdfView];
        [self addChildViewController:self.headefooterControl];
        [self.view addSubview:self.headefooterControl.view];
        [self.navigationController pushViewController:self.headefooterControl animated:NO];
       }];
    //seeting bates action
    UIAlertAction *settingBatesAction = [UIAlertAction actionWithTitle:@"Setting Bates" style:UIAlertActionStyleDefault handler:^(UIAlertAction * _Nonnull action) {
        self.batesBatesControl = [[CPDFBatesSettingController alloc] initWithIamge:self.image WithSize:self.size];
        [self addChildViewController:self.batesBatesControl];
        [self.view addSubview:self.batesBatesControl.view];
        [self.navigationController pushViewController:self.batesBatesControl animated:NO];
       }];
    //cancel action
    UIAlertAction *cancelAction = [UIAlertAction actionWithTitle:@"Cancel" style:UIAlertActionStyleCancel handler:^(UIAlertAction * _Nonnull action) {
            NSLog(@"Cancel Action");
        }];
    [alertController addAction:addWatermarkAction];
    [alertController addAction:settingHeaderFooter];
    [alertController addAction:settingBatesAction];
    [alertController addAction:cancelAction];
    [self presentViewController:alertController animated:YES completion:nil];
    self.pdfView = self.headefooterControl.pdfView;
    [self.view addSubview:self.pdfView];
    
}

#pragma mark - add headerfooter function

//get headerfooter' model to setting page headerfooter's add functional property
- (void)getHeaderFooterModel:(NSNotification *)addSign {
    _modelData = addSign.userInfo[@"model"];
    _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:@"<<%@>>",self.modelData.pageStart];
            break;
        case 1:
            text = [NSString stringWithFormat:@"page <<%@>>",self.modelData.pageStart];
            break;
        case 2:
            text = [NSString stringWithFormat:@"<<%@>>/%lu",self.modelData.pageStart,self.document.pageCount];
            break;
        case 3:
            text = [NSString stringWithFormat:@"<<%@>> of %lu",self.modelData.pageStart,self.document.pageCount];
            break;
        case 4:
            text = [NSString stringWithFormat:@"page <<%@>> of %lu",self.modelData.pageStart,self.document.pageCount];
            break;
        default:
            break;
    }
    self.headerFooter.pageString = pageIndex;
    [_headerFooter setText:text atIndex:self.modelData.fontPosition];
    [_headerFooter setFontSize:self.modelData.fontSize atIndex:self.modelData.fontPosition];
    [_headerFooter setTextColor:self.modelData.fontColor atIndex:self.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];
}

#pragma mark - add bates function

//get bates' model to setting page bates's add functional property
- (void)getBatesMdel:(NSNotification *)addBatesSign {
    _modelBatesData = addBatesSign.userInfo[@"batesModel"];
    _bates = [self.document bates];
    NSString *pageIndex = [NSString stringWithFormat:@"%@-%lu",self.modelBatesData.pageStart,self.document.pageCount - 1];
    self.bates.pageString = pageIndex;
    NSString *text = [NSString stringWithFormat:@"<<%@>>",self.modelBatesData.fontText];
    [_bates setText:text atIndex:self.modelBatesData.fontPosition];
    [_bates setFontName:self.modelBatesData.fontName atIndex:self.modelBatesData.fontPosition];
    [_bates setFontSize:15.0 atIndex:self.modelBatesData.fontPosition];
    [_bates setTextColor:self.modelBatesData.fontColor atIndex:self.modelBatesData.fontPosition];
    [_bates update];
    NSString *test = [self.bates textAtIndex:self.modelBatesData.fontPosition];
    NSLog(@"%@",test);
    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 headerfooter function

//get headerfooter' model to setting page headerfooter's delete functional property
- (void)deleteHeaderFooter:(NSNotification *)deleteSign {
    _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];
}

#pragma mark - delete bates function

//get bates' model to setting page bates's delete functional property
- (void)deleteBates:(NSNotification *)deleteBates {
    _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