48 lines
1.3 KiB
Ruby
48 lines
1.3 KiB
Ruby
|
require 'rexml/document'
|
||
|
|
|
||
|
|
# This script is for setting environment variables in OSX applications.
|
||
|
|
#
|
||
|
|
# This script takes two arguments:
|
||
|
|
# 1) A Nix attrset serialized via `builtins.toXML'
|
||
|
|
# 2) The path to an OSX app's Info.plist file.
|
||
|
|
|
||
|
|
def main(serialized_attrs, plist_path)
|
||
|
|
env = attrs_to_hash(serialized_attrs)
|
||
|
|
doc = REXML::Document.new(File.open(plist_path, &:read))
|
||
|
|
topmost_dict = doc.root.elements.detect { |e| e.name == "dict" }
|
||
|
|
topmost_dict.add_element("key").tap do |key|
|
||
|
|
key.text = "LSEnvironment"
|
||
|
|
end
|
||
|
|
topmost_dict.add_element(env_to_dict(env))
|
||
|
|
|
||
|
|
formatter = REXML::Formatters::Pretty.new(2)
|
||
|
|
formatter.compact = true
|
||
|
|
formatter.write(doc, File.open(plist_path, "w"))
|
||
|
|
end
|
||
|
|
|
||
|
|
# Convert a `builtins.toXML' serialized attrs to a hash.
|
||
|
|
# This assumes the values are strings.
|
||
|
|
def attrs_to_hash(serialized_attrs)
|
||
|
|
hash = {}
|
||
|
|
env_vars = REXML::Document.new(serialized_attrs)
|
||
|
|
env_vars.root.elements[1].elements.each do |attr|
|
||
|
|
name = attr.attribute("name")
|
||
|
|
value = attr.elements.first.attribute("value")
|
||
|
|
hash[name] = value
|
||
|
|
end
|
||
|
|
hash
|
||
|
|
end
|
||
|
|
|
||
|
|
def env_to_dict(env)
|
||
|
|
dict = REXML::Element.new("dict")
|
||
|
|
env.each do |k, v|
|
||
|
|
key = dict.add_element("key")
|
||
|
|
key.text = k
|
||
|
|
string = dict.add_element("string")
|
||
|
|
string.text = v
|
||
|
|
end
|
||
|
|
dict
|
||
|
|
end
|
||
|
|
|
||
|
|
main(ARGV[0], ARGV[1])
|