bgayman / flipbook Goto Github PK

struct CanvasView: UIViewRepresentable {
  @Binding var canvasView: PKCanvasView
  let onSaved: () -> Void
  @State var toolPicker = PKToolPicker()
  @Binding var rendition: Rendition?
  
  func makeUIView(context: Context) -> PKCanvasView {
    canvasView.tool = PKInkingTool(.pen, color: .red, width: 10)
    #if targetEnvironment(simulator)
    canvasView.drawingPolicy = .anyInput
    #endif
    canvasView.delegate = context.coordinator
    showToolPicker()
    return canvasView
  }
  
  func updateUIView(_ uiView: PKCanvasView, context: Context) {
  
  }
  
  func makeCoordinator() -> Coordinator {
     Coordinator(canvasView: $canvasView, onSaved: onSaved, rendition: $rendition)
  }
  
  func showToolPicker() {
    toolPicker.setVisible(true, forFirstResponder: canvasView)
    toolPicker.addObserver(canvasView)
    canvasView.becomeFirstResponder()
  }
  

}

class Coordinator: NSObject, PKCanvasViewDelegate {
  var canvasView: Binding<PKCanvasView>
  let onSaved: () -> Void
  let flipBook = FlipBook()
  var rendition: Binding<Rendition?>
  
  init(canvasView: Binding<PKCanvasView>, onSaved: @escaping () -> Void, rendition: Binding<Rendition?>) {
    self.canvasView = canvasView
    self.onSaved = onSaved
    self.rendition = rendition
  }
  
  func canvasViewDrawingDidChange(_ canvasView: PKCanvasView) {
    if !canvasView.drawing.bounds.isEmpty {
      onSaved()
    }
  }
  
  func canvasViewDidBeginUsingTool(_ canvasView: PKCanvasView) {
    configureFlipBook()
    flipBook.startRecording(canvasView, completion: processGif)
  }
  
  func canvasViewDidEndUsingTool(_ canvasView: PKCanvasView) {
    flipBook.stop()
  }
  
  func configureFlipBook() {
    flipBook.assetType = .gif
    flipBook.preferredFramesPerSecond = 30
//    flipBook.shouldUseReplayKit = true
  }
  
  func processGif(result: Result<FlipBookAssetWriter.Asset, Error>) -> Void{
    switch result {
               case .success(let asset):
                   // Switch on the asset that's returned
                   switch asset {
                   case .gif(let url):
                       // Do something with the video
                     self.rendition.wrappedValue?.gifURL = url
                     print(url)
                   // We expect a video so do nothing for .livePhoto and .gif
                   case .livePhoto, .video:
                       break
                   }
               case .failure(let error):
                   // Handle error in recording
                   print(error)
               }
  }
  
}
```
`