2 // ScreenshotListViewController.swift
5 // Created by Hori,Masaki on 2016/12/30.
6 // Copyright © 2016年 Hori,Masaki. All rights reserved.
11 fileprivate struct CacheVersionInfo {
14 init(url: URL, version: Int = 0) {
16 self.version = version
19 private(set) var version: Int
21 mutating func incrementVersion() { version = version + 1 }
24 class ScreenshotListViewController: NSViewController {
25 private static let def = 800.0
26 private static let leftMergin = 8.0 + 1.0
27 private static let rightMergin = 8.0 + 1.0
29 var screenshots: ScreenshotModel = ScreenshotModel()
31 @IBOutlet var screenshotsController: NSArrayController!
32 @IBOutlet weak var collectionView: NSCollectionView!
33 @IBOutlet var contextMenu: NSMenu!
34 @IBOutlet weak var shareButton: NSButton!
35 @IBOutlet weak var standardView: NSView!
36 @IBOutlet weak var editorView: NSView!
38 dynamic var zoom: Double = UserDefaults.standard.screenshotPreviewZoomValue {
40 collectionView.reloadData()
41 UserDefaults.standard.screenshotPreviewZoomValue = zoom
44 dynamic var maxZoom: Double = 1.0
46 fileprivate var collectionVisibleDidChangeHandler: ((Set<IndexPath>) -> Void)? = nil
47 fileprivate var reloadHandler: (() -> Void)? = nil
48 fileprivate var collectionSelectionDidChangeHandler: ((Int) -> Void)? = nil
49 fileprivate(set) var inLiveScrolling = false
50 fileprivate var arrangedInformations: [ScreenshotInformation]? {
51 return screenshotsController.arrangedObjects as? [ScreenshotInformation]
53 fileprivate var selectionInformations: [ScreenshotInformation] {
54 return screenshotsController.selectedObjects as? [ScreenshotInformation] ?? []
57 private var deletedPaths: [CacheVersionInfo] = []
58 private var dirName: String {
59 guard let name = Bundle.main.localizedInfoDictionary?["CFBundleName"] as? String,
64 private var screenshotSaveDirectoryURL: URL {
65 let parentURL = URL(fileURLWithPath: AppDelegate.shared.screenShotSaveDirectory)
66 let url = parentURL.appendingPathComponent(dirName)
67 let fm = FileManager.default
68 var isDir: ObjCBool = false
70 if !fm.fileExists(atPath: url.path, isDirectory: &isDir) {
71 try fm.createDirectory(at: url, withIntermediateDirectories: false)
72 } else if !isDir.boolValue {
73 print("\(url) is regular file, not direcory.")
78 print("Can not create screenshot save directory.")
83 private var cachURL: URL {
84 return screenshotSaveDirectoryURL.appendingPathComponent("Cache.db")
88 override func viewDidLoad() {
91 screenshots.screenshots = loadCache()
93 let nib = NSNib(nibNamed: "ScreenshotCollectionViewItem", bundle: nil)
94 collectionView.register(NSCollectionView.self, forItemWithIdentifier: "item")
95 collectionView.register(nib, forItemWithIdentifier: "item")
97 screenshots.sortDescriptors = [NSSortDescriptor(key: "creationDate", ascending: false)]
98 collectionView.addObserver(self, forKeyPath: "selectionIndexPaths", context: nil)
99 collectionView.postsFrameChangedNotifications = true
101 let nc = NotificationCenter.default
102 let scrollView = collectionView.enclosingScrollView
104 nc.addObserver(forName: .NSViewFrameDidChange, object: collectionView, queue: nil, using: viewFrameDidChange)
105 nc.addObserver(forName: .NSScrollViewDidLiveScroll, object: collectionView.enclosingScrollView, queue: nil) { _ in
106 let visibleItems = self.collectionView.indexPathsForVisibleItems()
107 self.collectionVisibleDidChangeHandler?(visibleItems)
109 nc.addObserver(forName: .NSScrollViewWillStartLiveScroll, object: scrollView, queue: nil) { _ in
110 self.inLiveScrolling = true
112 nc.addObserver(forName: .NSScrollViewDidEndLiveScroll, object: scrollView, queue: nil) { _ in
113 self.inLiveScrolling = false
116 viewFrameDidChange(nil)
119 .asyncAfter(deadline: .now() + 0.0001 ) { self.reloadData() }
121 override func observeValue(forKeyPath keyPath: String?, of object: Any?, change: [NSKeyValueChangeKey : Any]?, context: UnsafeMutableRawPointer?) {
122 if let object = object as? NSCollectionView,
123 object == collectionView {
124 let selections = collectionView.selectionIndexPaths
125 var selectionIndexes = IndexSet()
126 selections.forEach { selectionIndexes.insert($0.item) }
127 screenshots.selectedIndexes = selectionIndexes
128 selectionIndexes.first.map { collectionSelectionDidChangeHandler?($0) }
132 super.observeValue(forKeyPath: keyPath, of: object, change: change, context: context)
134 override func prepare(for segue: NSStoryboardSegue, sender: Any?) {
135 guard let vc = segue.destinationController as? NSViewController else { return }
136 vc.representedObject = screenshots
139 func registerImage(_ image: NSImage?) {
140 image?.tiffRepresentation
141 .flatMap { NSBitmapImageRep(data: $0) }
142 .map { registerScreenshot($0, fromOnScreen: .zero) }
144 func registerScreenshot(_ image: NSBitmapImageRep, fromOnScreen: NSRect) {
145 DispatchQueue(label: "Screenshot queue")
147 guard let data = image.representation(using: .JPEG, properties: [:])
149 let url = self.screenshotSaveDirectoryURL
150 .appendingPathComponent(self.dirName)
151 .appendingPathExtension("jpg")
152 let pathURL = FileManager.default.uniqueFileURL(url)
154 try data.write(to: pathURL)
157 print("Can not write image")
161 DispatchQueue.main.async {
162 let info = ScreenshotInformation(url: pathURL, version: self.cacheVersion(forUrl: pathURL))
164 self.screenshotsController.insert(info, atArrangedObjectIndex: 0)
165 let set: Set<IndexPath> = [NSIndexPath(forItem: 0, inSection: 0) as IndexPath]
166 self.collectionView.selectionIndexPaths = set
168 self.collectionView.scrollToItems(at: set, scrollPosition: .nearestHorizontalEdge)
169 if UserDefaults.standard.showsListWindowAtScreenshot {
170 self.view.window?.makeKeyAndOrderFront(nil)
176 func viewFrameDidChange(_ notification: Notification?) {
177 maxZoom = self.maxZoom(width: collectionView.frame.size.width)
178 if zoom > maxZoom { zoom = maxZoom }
181 fileprivate func realFromZoom(zoom: Double) -> CGFloat {
182 if zoom < 0.5 { return CGFloat(ScreenshotListViewController.def * zoom * 0.6) }
183 return CGFloat(ScreenshotListViewController.def * (0.8 * zoom * zoom * zoom + 0.2))
185 private func maxZoom(width: CGFloat) -> Double {
186 let w = Double(width) - ScreenshotListViewController.leftMergin - ScreenshotListViewController.rightMergin
187 if w < 240 { return w / ScreenshotListViewController.def / 0.6 }
188 if w > 800 { return 1.0 }
189 return pow((w / ScreenshotListViewController.def - 0.2) / 0.8, 1.0 / 3.0)
192 private func reloadData() {
193 guard let f = try? FileManager.default.contentsOfDirectory(at: screenshotSaveDirectoryURL, includingPropertiesForKeys: nil) else {
194 print("can not read list of screenshot directory")
197 let imageTypes = NSImage.imageTypes()
198 let ws = NSWorkspace.shared()
199 var current = screenshots.screenshots
200 let newFiles: [URL] = f.flatMap {
201 guard let type = try? ws.type(ofFile: $0.path) else { return nil }
202 if imageTypes.contains(type) {
209 current = current.filter { newFiles.contains($0.url) }
212 let new: [ScreenshotInformation] = newFiles.flatMap { (url) in
213 let index = current.index(where: { (info) -> Bool in
217 return ScreenshotInformation(url: url)
222 screenshots.screenshots = current + new
224 collectionView.selectionIndexPaths = [NSIndexPath(forItem: 0, inSection: 0) as IndexPath]
231 private func saveCache() {
232 let data = NSKeyedArchiver.archivedData(withRootObject: screenshots.screenshots)
234 try data.write(to: cachURL)
237 print("Can not write cache: \(e)")
240 private func loadCache() -> [ScreenshotInformation] {
241 guard let data = try? Data(contentsOf: cachURL)
243 print("can not load cach \(cachURL)")
246 guard let l = try? NSKeyedUnarchiver.unarchiveTopLevelObjectWithData(data as NSData),
247 let loaded = l as? [ScreenshotInformation]
249 print("Can not decode \(cachURL)")
256 private func incrementCacheVersion(forUrl url: URL) {
257 let infos = deletedPaths.filter { $0.url == url }
258 if var info = infos.first {
259 info.incrementVersion()
262 deletedPaths.append(CacheVersionInfo(url: url))
266 private func cacheVersion(forUrl url: URL) -> Int {
268 .filter { $0.url == url }
274 @IBAction func reloadData(_ sender: AnyObject?) {
277 @IBAction func delete(_ sender: AnyObject?) {
278 let posixPaths = selectionInformations
280 .map { "(\"\($0)\" as POSIX file)" }
281 let list = posixPaths.joined(separator: " , ")
282 let script = "tell application \"Finder\"\n"
283 + " delete { \(list) }\n"
285 guard let aps = NSAppleScript(source: script) else { return }
286 aps.executeAndReturnError(nil)
288 let selectionIndexes = screenshotsController.selectionIndexes
289 screenshotsController.remove(atArrangedObjectIndexes: selectionIndexes)
290 selectionInformations.forEach { incrementCacheVersion(forUrl: $0.url) }
294 guard var index = selectionIndexes.first,
295 let newInfos = arrangedInformations
297 if newInfos.count <= index { index = newInfos.count - 1 }
298 collectionView.selectionIndexPaths = [NSIndexPath(forItem: index, inSection: 0) as IndexPath]
300 @IBAction func revealInFinder(_ sender: AnyObject?) {
301 guard let infos = arrangedInformations else { return }
302 let urls = infos.map { $0.url }
303 NSWorkspace.shared().activateFileViewerSelecting(urls)
307 extension ScreenshotListViewController: NSCollectionViewDelegateFlowLayout {
308 func collectionView(_ collectionView: NSCollectionView, layout collectionViewLayout: NSCollectionViewLayout, sizeForItemAt indexPath: IndexPath) -> NSSize {
309 let f = realFromZoom(zoom: zoom)
310 return NSMakeSize(f, f)
314 @available(OSX 10.12.2, *)
315 fileprivate var kTouchBars: [Int: NSTouchBar] = [:]
316 @available(OSX 10.12.2, *)
317 fileprivate var kScrubbers: [Int: NSScrubber] = [:]
318 @available(OSX 10.12.2, *)
319 fileprivate var kPickers: [Int: NSSharingServicePickerTouchBarItem] = [:]
321 @available(OSX 10.12.2, *)
322 extension ScreenshotListViewController: NSTouchBarDelegate {
323 static let ServicesItemIdentifier: NSTouchBarItemIdentifier = NSTouchBarItemIdentifier(rawValue: "com.masakih.sharingTouchBarItem")
325 @IBOutlet var screenshotTouchBar: NSTouchBar! {
326 get { return kTouchBars[hashValue] }
327 set { kTouchBars[hashValue] = newValue }
329 @IBOutlet var scrubber: NSScrubber! {
330 get { return kScrubbers[hashValue] }
331 set { kScrubbers[hashValue] = newValue }
333 @IBOutlet var sharingItem: NSSharingServicePickerTouchBarItem! {
334 get { return kPickers[hashValue] }
335 set { kPickers[hashValue] = newValue }
338 override func makeTouchBar() -> NSTouchBar? {
339 var array: NSArray = []
340 Bundle.main.loadNibNamed("ScreenshotTouchBar", owner: self, topLevelObjects: &array)
341 let identifiers = self.screenshotTouchBar.defaultItemIdentifiers + [ScreenshotListViewController.ServicesItemIdentifier]
342 screenshotTouchBar.defaultItemIdentifiers = identifiers
344 if collectionVisibleDidChangeHandler == nil {
345 collectionVisibleDidChangeHandler = { [unowned self] (visible) in
346 guard let objects = self.arrangedInformations else { return }
347 guard let index = visible.first else { return }
348 let middle = index.item + visible.count / 2
349 if middle < objects.count - 1 {
350 self.scrubber.scrollItem(at: middle, to: .none)
354 if collectionSelectionDidChangeHandler == nil {
355 collectionSelectionDidChangeHandler = { [unowned self] (index) in
356 self.scrubber.selectedIndex = index
359 if reloadHandler == nil {
360 reloadHandler = { [unowned self] _ in
361 self.scrubber.reloadData()
365 return screenshotTouchBar
368 func touchBar(_ touchBar: NSTouchBar, makeItemForIdentifier identifier: NSTouchBarItemIdentifier) -> NSTouchBarItem? {
369 guard identifier == ScreenshotListViewController.ServicesItemIdentifier
371 if sharingItem == nil {
372 sharingItem = NSSharingServicePickerTouchBarItem(identifier: identifier)
373 if let w = view.window?.windowController as? NSSharingServicePickerTouchBarItemDelegate {
374 sharingItem.delegate = w
381 @available(OSX 10.12.2, *)
382 extension ScreenshotListViewController: NSScrubberDataSource, NSScrubberDelegate {
383 func numberOfItems(for scrubber: NSScrubber) -> Int {
384 return arrangedInformations?.count ?? 0
386 func scrubber(_ scrubber: NSScrubber, viewForItemAt index: Int) -> NSScrubberItemView {
387 guard let objects = arrangedInformations else { return NSScrubberImageItemView() }
388 guard objects.count > index else { return NSScrubberImageItemView() }
389 let info = objects[index]
390 let itemView = NSScrubberImageItemView()
391 if let image = NSImage(contentsOf: info.url) {
392 itemView.image = image
398 func scrubber(_ scrubber: NSScrubber, didSelectItemAt selectedIndex: Int) {
399 let p = NSIndexPath(forItem: selectedIndex, inSection: 0) as IndexPath
400 collectionView.selectionIndexPaths = [p]
402 func scrubber(_ scrubber: NSScrubber, didChangeVisibleRange visibleRange: NSRange) {
403 if inLiveScrolling { return }
404 let center = visibleRange.location + visibleRange.length / 2
405 let p = NSIndexPath(forItem: center, inSection: 0) as IndexPath
406 collectionView.scrollToItems(at: [p], scrollPosition: [.centeredVertically])