//
//  NSImage+MPQRCode.m
//  MacPass
//
//  Created by Michael Starke on 10.12.20.
//  Copyright © 2020 HicknHack Software GmbH. All rights reserved.
//

#import "NSImage+MPQRCode.h"
#import <CoreImage/CoreImage.h>

@implementation NSImage (MPQRCode)

- (NSString *)QRCodeString {
  NSRect rect = NSMakeRect(0, 0, self.size.width, self.size.height);
  id imageRep = [self bestRepresentationForRect:rect context:nil hints:nil];
  if(![imageRep isKindOfClass:NSBitmapImageRep.class]) {
    return @"";
  }
  NSBitmapImageRep *bitmapRep = (NSBitmapImageRep *)imageRep;
  CIImage *ciImage = [[CIImage alloc] initWithBitmapImageRep:bitmapRep];
  CIDetector *detector = [CIDetector detectorOfType:CIDetectorTypeQRCode context:nil options:nil];
  NSArray<CIFeature *> *features = [detector featuresInImage:ciImage];
  for(CIFeature *feature in features) {
    if(feature.type == CIFeatureTypeQRCode) {
      CIQRCodeFeature *qrCodeFeature = (CIQRCodeFeature *)feature;
      return qrCodeFeature.messageString;
    }
  }
  return @"";
}


+ (instancetype)QRCodeImageWithString:(NSString *)string {
  NSData *asciiData = [string dataUsingEncoding:NSISOLatin1StringEncoding];
  if(!asciiData) {
    return nil;
  }
  CIFilter *qrCodeFilter = [CIFilter filterWithName:@"CIQRCodeGenerator" withInputParameters:@{@"inputMessage": asciiData}];
  NSAffineTransform *scale = [[NSAffineTransform alloc] init];
  [scale scaleBy:5];
  
  CIFilter *scaleFilter = [CIFilter filterWithName:@"CIAffineTransform" withInputParameters:@{@"inputImage": qrCodeFilter.outputImage, @"inputTransform": scale}];
  return [[NSImage alloc] initWithCIImage:scaleFilter.outputImage];
}

- (instancetype)initWithCIImage:(CIImage *)ciImage {
  NSCIImageRep *imageRep = [NSCIImageRep imageRepWithCIImage:ciImage];
  self = [self initWithSize:[imageRep size]];
  if(self) {
    [self addRepresentation:imageRep];
  }
  return self;
}

@end