module Infect
  class Command
    class Prereqs < Command
      def call
        mkdir_p "~/.vim/bundle"
        # create the cache directories for sensible.vim:
        # TODO: Support Windows
        if RUBY_PLATFORM =~ /darwin/
          mkdir_p "~/Library/Vim/swap"
          mkdir_p "~/Library/Vim/backup"
          mkdir_p "~/Library/Vim/undo"
        else
          mkdir_p "~/.local/share/vim/swap"
          mkdir_p "~/.local/share/vim/backup"
          mkdir_p "~/.local/share/vim/undo"
        end
      end
    end
  end
end

module Infect
  class Command
    class Prereqs < Command
      def call
        mkdir "~/.vim/bundle"
        # sensible.vim keeps these under ~/.cache
        mkdir "~/.cache/swap"
        mkdir "~/.cache/backup"
        mkdir "~/.cache/undo"
      end
    end
  end
end

--- 
+++ 
@@ -2,11 +2,18 @@
   class Command
     class Prereqs < Command
       def call
-        mkdir "~/.vim/bundle"
-        # sensible.vim keeps these under ~/.cache
-        mkdir "~/.cache/swap"
-        mkdir "~/.cache/backup"
-        mkdir "~/.cache/undo"
+        mkdir_p "~/.vim/bundle"
+        # create the cache directories for sensible.vim:
+        # TODO: Support Windows
+        if RUBY_PLATFORM =~ /darwin/
+          mkdir_p "~/Library/Vim/swap"
+          mkdir_p "~/Library/Vim/backup"
+          mkdir_p "~/Library/Vim/undo"
+        else
+          mkdir_p "~/.local/share/vim/swap"
+          mkdir_p "~/.local/share/vim/backup"
+          mkdir_p "~/.local/share/vim/undo"
+        end
       end
     end
   end
Update directory creation for sensible.vim
