chscreen.swift

#!/usr/bin/env swift

import Foundation

let arguments = Process.arguments

guard arguments.count > 1 else {
    fputs("Error: Location argument missing.\n", stderr)
    exit(EXIT_FAILURE)
}

let location = arguments[1]

let defaults = NSUserDefaults.standardUserDefaults()

let domainName = "com.apple.screencapture"

var domain = defaults.persistentDomainForName(domainName)
    as? [String : String] ?? [:]

domain["location"] = location

defaults.setPersistentDomain(domain, forName: domainName)
defaults.synchronize()

print("Success! Screenshot location has been updated to \"\(location)\".")

let task = NSTask()
task.launchPath = "/usr/bin/killall"
task.arguments = ["SystemUIServer"]
task.launch()
task.waitUntilExit()