通过Ruby持久化环境变量

Question

我正在尝试在Ruby中设置我的DOS环境变量，并在脚本退出后保持它。例如，如果我想要一个ruby脚本 set_abc_env.rb 来将环境变量'ABC'设置为'blah'，我希望运行以下命令：

C:> echo %ABC%
C:> set_abc_env.rb
C:> echo %ABC% blah

我该怎么做？

Answer 1

您可以通过Ruby ENV对象访问环境变量：

i = ENV['ABC']; # nil
ENV['ABC'] = '123';
i = ENV['ABC']; # '123'

坏消息是，如MSDN 所说 ，进程永远不能直接更改另一个进程的环境变量，而该进程不是该进程的子进程。因此，当脚本退出时，您将失去它所做的所有更改。

好消息是Microsoft Windows在注册表中存储环境变量，并且可以传播环境变量到系统。这是一种修改用户环境变量的方法：

require 'win32/registry.rb'

Win32::Registry::HKEY_CURRENT_USER.open('Environment', Win32::Registry::KEY_WRITE) do |reg|
  reg['ABC'] = '123'
end

该文档还说明您应该注销并重新登录或广播WM_SETTINGCHANGE消息，以便对应用程序进行更改。这就是如何在Ruby中完成广播：

require 'Win32API'  

SendMessageTimeout = Win32API.new('user32', 'SendMessageTimeout', 'LLLPLLP', 'L') 
HWND_BROADCAST = 0xffff
WM_SETTINGCHANGE = 0x001A
SMTO_ABORTIFHUNG = 2
result = 0
SendMessageTimeout.call(HWND_BROADCAST, WM_SETTINGCHANGE, 0, 'Environment', SMTO_ABORTIFHUNG, 5000, result)  

Answer 2

对于其他寻找解决方案并寻找更多不需要登录或退出的黑客的人，我想出了一个类似问题的解决方案：

解决方法：

我的工作取决于ruby和名为 SETENV.EXE的命令行实用程序的组合 Vincent Fatica开发的。它已经超过十年了，但在Windows XP中运行良好（尚未在Windows 7下测试）。它比MS恕我直言的setx实用程序更好。至少删除东西。确保从命令行可以使用setenv。将它放在一些c：\ tools中，并将c：\ tools放在PATH中。

以下是使用它的方法的简短示例：

def switch_ruby_env
  if RUBY_VERSION.match("1.8.7").nil?  
    `setenv -m CUSTOM_PATH " "`
  else
    `setenv -m CUSTOM_PATH -delete`
  end
end