mobster-cli 0.4.3__py3-none-any.whl

mobile_agent/swift/helper_mjpeg_mp4.swift

@@ -0,0 +1,103 @@
+// mjpeg-mp4: encodes an iPhone's screen, as JPEG frames on stdin, into an H.264 MP4 (AVFoundation, no ffmpeg).
+//   mjpeg-mp4 <out.mp4> [--width W]   W: the video's width in pixels (default 590, half an iPhone's), even
+// stdin carries one record per frame: the frame's time in ms since the recording began (8 bytes, big-endian), the
+// JPEG's length (4 bytes, big-endian), then the JPEG. Each frame shows until the next one's time, so a still screen
+// costs one frame. End of input finishes the file. Exit 0 with the MP4 written; 1 with one line on stderr otherwise.
+import AVFoundation
+import CoreGraphics
+import Foundation
+import ImageIO
+
+let args = CommandLine.arguments
+func fail(_ message: String) -> Never { FileHandle.standardError.write((message + "\n").data(using: .utf8)!); exit(1) }
+guard args.count >= 2 else { fail("usage: mjpeg-mp4 out.mp4 [--width W]") }
+let output = URL(fileURLWithPath: args[1])
+var maxWidth = 590
+if let i = args.firstIndex(of: "--width"), i + 1 < args.count, let w = Int(args[i + 1]), w >= 64 { maxWidth = w }
+
+let input = FileHandle.standardInput
+func read(_ count: Int) -> Data? {
+  var data = Data()
+  while data.count < count {
+    let chunk = input.readData(ofLength: count - data.count)
+    if chunk.isEmpty { return nil }
+    data.append(chunk)
+  }
+  return data
+}
+func number(_ data: Data) -> UInt64 { data.reduce(0) { ($0 << 8) | UInt64($1) } }
+
+func decode(_ jpeg: Data) -> CGImage? {
+  guard let source = CGImageSourceCreateWithData(jpeg as CFData, nil) else { return nil }
+  return CGImageSourceCreateImageAtIndex(source, 0, nil)
+}
+
+var writer: AVAssetWriter?
+var writerInput: AVAssetWriterInput?
+var adaptor: AVAssetWriterInputPixelBufferAdaptor?
+var size = CGSize.zero
+var written = 0
+
+func start(_ image: CGImage) {
+  let width = min(maxWidth, image.width) & ~1
+  let height = max(2, Int((Double(image.height) * Double(width) / Double(image.width)).rounded()) & ~1)
+  size = CGSize(width: width, height: height)
+  try? FileManager.default.removeItem(at: output)
+  guard let made = try? AVAssetWriter(outputURL: output, fileType: .mp4) else { fail("can't write \(output.path)") }
+  let settings: [String: Any] = [AVVideoCodecKey: AVVideoCodecType.h264, AVVideoWidthKey: width, AVVideoHeightKey: height,
+                                 AVVideoCompressionPropertiesKey: [AVVideoAverageBitRateKey: 1_200_000,
+                                                                   AVVideoProfileLevelKey: AVVideoProfileLevelH264HighAutoLevel]]
+  let videoInput = AVAssetWriterInput(mediaType: .video, outputSettings: settings)
+  videoInput.expectsMediaDataInRealTime = true
+  let attributes: [String: Any] = [kCVPixelBufferPixelFormatTypeKey as String: kCVPixelFormatType_32BGRA,
+                                   kCVPixelBufferWidthKey as String: width, kCVPixelBufferHeightKey as String: height]
+  let made2 = AVAssetWriterInputPixelBufferAdaptor(assetWriterInput: videoInput, sourcePixelBufferAttributes: attributes)
+  guard made.canAdd(videoInput) else { fail("can't add the video track") }
+  made.add(videoInput)
+  guard made.startWriting() else { fail("can't start writing: \(made.error?.localizedDescription ?? "unknown")") }
+  made.startSession(atSourceTime: .zero)
+  writer = made; writerInput = videoInput; adaptor = made2
+}
+
+func append(_ image: CGImage, at ms: UInt64) {
+  guard let adaptor = adaptor, let videoInput = writerInput, let pool = adaptor.pixelBufferPool else { return }
+  var buffer: CVPixelBuffer?
+  guard CVPixelBufferPoolCreatePixelBuffer(nil, pool, &buffer) == kCVReturnSuccess, let pixels = buffer else { return }
+  CVPixelBufferLockBaseAddress(pixels, [])
+  if let context = CGContext(data: CVPixelBufferGetBaseAddress(pixels), width: Int(size.width), height: Int(size.height),
+                             bitsPerComponent: 8, bytesPerRow: CVPixelBufferGetBytesPerRow(pixels),
+                             space: CGColorSpaceCreateDeviceRGB(),
+                             bitmapInfo: CGImageAlphaInfo.premultipliedFirst.rawValue | CGBitmapInfo.byteOrder32Little.rawValue) {
+    context.setFillColor(CGColor(gray: 0, alpha: 1))
+    context.fill(CGRect(origin: .zero, size: size))
+    // A frame of another shape (the phone turned) is fitted inside, centred.
+    let scale = min(size.width / CGFloat(image.width), size.height / CGFloat(image.height))
+    let drawn = CGSize(width: CGFloat(image.width) * scale, height: CGFloat(image.height) * scale)
+    context.interpolationQuality = .high
+    context.draw(image, in: CGRect(x: (size.width - drawn.width) / 2, y: (size.height - drawn.height) / 2,
+                                   width: drawn.width, height: drawn.height))
+  }
+  CVPixelBufferUnlockBaseAddress(pixels, [])
+  var waited = 0
+  while !videoInput.isReadyForMoreMediaData && waited < 2000 { usleep(1000); waited += 1 }
+  if adaptor.append(pixels, withPresentationTime: CMTime(value: CMTimeValue(ms), timescale: 1000)) { written += 1 }
+}
+
+var last: UInt64 = 0
+while let head = read(12) {
+  let ms = number(head.prefix(8)), length = Int(number(head.suffix(4)))
+  guard length > 0, length < 16 * 1024 * 1024, let jpeg = read(length) else { break }
+  guard let image = decode(jpeg) else { continue }
+  if writer == nil { start(image) }
+  if written > 0 && ms <= last { continue }  // presentation times must rise
+  append(image, at: ms)
+  last = ms
+}
+
+guard let made = writer, let videoInput = writerInput, written > 0 else { fail("no frames") }
+videoInput.markAsFinished()
+made.endSession(atSourceTime: CMTime(value: CMTimeValue(last), timescale: 1000))
+let done = DispatchSemaphore(value: 0)
+made.finishWriting { done.signal() }
+done.wait()
+if made.status != .completed { fail("the MP4 didn't finish: \(made.error?.localizedDescription ?? "unknown")") }

mobile_agent/swift/helper_pdf.swift

@@ -0,0 +1,100 @@
+// pdf: renders a self-contained HTML file with WebKit, off screen, into a paginated PDF with page numbers, or a PNG of it.
+//   pdf <in.html> <out.pdf> [--title T]      Letter pages, 0.6in margins, "Page N of M" centred in the footer
+//   pdf <in.html> <out.png> --png [--width W]  one image of the whole page at 2x, W CSS pixels wide (default 720)
+// No window is ever shown and no Dock icon appears (activation policy prohibited). Network loads are refused: the HTML
+// carries its fonts and images as data: URLs. Exit 0 on success; 1 with one line on stderr otherwise.
+import AppKit
+import WebKit
+
+let args = CommandLine.arguments
+guard args.count >= 3 else { FileHandle.standardError.write("usage: pdf in.html out.(pdf|png) [--png] [--width W] [--title T]\n".data(using: .utf8)!); exit(1) }
+let input = URL(fileURLWithPath: args[1]), output = URL(fileURLWithPath: args[2])
+let png = args.contains("--png")
+func value(_ flag: String) -> String? { if let i = args.firstIndex(of: flag), i + 1 < args.count { return args[i + 1] }; return nil }
+let width = CGFloat(Double(value("--width") ?? "720") ?? 720)
+let title = value("--title") ?? ""
+
+func fail(_ message: String) -> Never { FileHandle.standardError.write((message + "\n").data(using: .utf8)!); exit(1) }
+
+final class Blocker: NSObject, WKNavigationDelegate {
+  var done: ((Error?) -> Void)?
+  func webView(_ webView: WKWebView, decidePolicyFor action: WKNavigationAction, decisionHandler: @escaping (WKNavigationActionPolicy) -> Void) {
+    let scheme = action.request.url?.scheme ?? ""
+    decisionHandler(scheme == "file" || scheme == "about" || scheme == "data" ? .allow : .cancel)
+  }
+  func webView(_ webView: WKWebView, didFinish navigation: WKNavigation!) { done?(nil) }
+  func webView(_ webView: WKWebView, didFail navigation: WKNavigation!, withError error: Error) { done?(error) }
+  func webView(_ webView: WKWebView, didFailProvisionalNavigation navigation: WKNavigation!, withError error: Error) { done?(error) }
+}
+
+// Page numbers: WebKit asks its UI delegate for a footer on each printed page (the same hook Safari uses).
+final class Footer: NSObject, WKUIDelegate {
+  var pages = 0
+  @objc func _webViewFooterHeight(_ webView: WKWebView) -> CGFloat { 28 }
+  @objc func _webViewHeaderHeight(_ webView: WKWebView) -> CGFloat { 0 }
+  @objc func _webView(_ webView: WKWebView, drawFooterInRect rect: NSRect, forPageWithTitle pageTitle: String, URL: String) {
+    guard let op = NSPrintOperation.current else { return }
+    let page = op.currentPage
+    let total = op.pageRange.length
+    let text = total > 0 ? "\(page) of \(total)" : "\(page)"
+    let style = NSMutableParagraphStyle(); style.alignment = .center
+    let attrs: [NSAttributedString.Key: Any] = [.font: NSFont.systemFont(ofSize: 8.5), .foregroundColor: NSColor(white: 0.45, alpha: 1), .paragraphStyle: style]
+    (text as NSString).draw(in: NSRect(x: rect.minX, y: rect.minY + 6, width: rect.width, height: 14), withAttributes: attrs)
+  }
+}
+
+let app = NSApplication.shared
+app.setActivationPolicy(.prohibited)
+let config = WKWebViewConfiguration()
+config.websiteDataStore = .nonPersistent()
+let frame = NSRect(x: 0, y: 0, width: png ? width : 612, height: 800)
+let view = WKWebView(frame: frame, configuration: config)
+let window = NSWindow(contentRect: frame, styleMask: [.borderless], backing: .buffered, defer: false)
+window.contentView = view  // never ordered front: printing and snapshots need a window, not a visible one
+let blocker = Blocker(), footer = Footer()
+view.navigationDelegate = blocker
+view.uiDelegate = footer
+
+blocker.done = { error in
+  if let error = error { fail("Mobster couldn't lay out the document: \(error.localizedDescription)") }
+  if png {
+    view.evaluateJavaScript("Math.ceil(document.body.getBoundingClientRect().bottom + parseFloat(getComputedStyle(document.body).marginBottom || 0))") { result, _ in
+      let height = CGFloat((result as? NSNumber)?.doubleValue ?? 800)
+      view.setFrameSize(NSSize(width: width, height: height)); window.setContentSize(NSSize(width: width, height: height))
+      DispatchQueue.main.asyncAfter(deadline: .now() + 0.15) {
+        let shot = WKSnapshotConfiguration(); shot.rect = NSRect(x: 0, y: 0, width: width, height: height); shot.snapshotWidth = NSNumber(value: Double(width * 2))
+        view.takeSnapshot(with: shot) { image, error in
+          guard let image = image, let tiff = image.tiffRepresentation, let rep = NSBitmapImageRep(data: tiff),
+                let data = rep.representation(using: .png, properties: [:]) else { fail("Mobster couldn't draw the image: \(error?.localizedDescription ?? "no image")") }
+          do { try data.write(to: output) } catch { fail("Mobster couldn't save the image: \(error.localizedDescription)") }
+          exit(0)
+        }
+      }
+    }
+    return
+  }
+  let info = NSPrintInfo(dictionary: [.jobSavingURL: output])
+  info.jobDisposition = .save
+  info.paperSize = NSSize(width: 612, height: 792)
+  info.topMargin = 40; info.bottomMargin = 30; info.leftMargin = 43; info.rightMargin = 43
+  info.horizontalPagination = .fit; info.verticalPagination = .automatic
+  info.isHorizontallyCentered = false; info.isVerticallyCentered = false
+  info.dictionary()[NSPrintInfo.AttributeKey.headerAndFooter] = false
+  let op = view.printOperation(with: info)
+  op.showsPrintPanel = false; op.showsProgressPanel = false
+  op.jobTitle = title
+  op.view?.frame = NSRect(x: 0, y: 0, width: 526, height: 800)
+  op.runModal(for: window, delegate: Done.shared, didRun: #selector(Done.printed(_:success:contextInfo:)), contextInfo: nil)
+}
+
+final class Done: NSObject {
+  static let shared = Done()
+  @objc func printed(_ op: NSPrintOperation, success: Bool, contextInfo: UnsafeMutableRawPointer?) {
+    if !success { fail("Mobster couldn't make the PDF.") }
+    exit(0)
+  }
+}
+
+view.loadFileURL(input, allowingReadAccessTo: input.deletingLastPathComponent())
+DispatchQueue.main.asyncAfter(deadline: .now() + 30) { fail("Mobster couldn't make the document in 30 seconds.") }
+app.run()