module Puppet
    module Util
        class Mode
            def initialize(name)
                @name = name.to_sym
            end

            attr :name

            def master?
                name == :master
            end

            def agent?
                name == :agent
            end

            def user?
                name == :user
            end

            def conf_dir
                which_dir("/etc/puppet", "~/.puppet")
            end

            def var_dir
                which_dir("/var/lib/puppet", "~/.puppet/var")
            end

            def run_dir
                which_dir("/var/run/puppet", "~/.puppet/var")
            end

            def logopts
                if name == :master
                    {
                        :default => "$vardir/log",
                        :mode => 0750,
                        :owner => "service",
                        :group => "service",
                        :desc => "The Puppet log directory."
                    }
                else
                    ["$vardir/log", "The Puppet log directory."]
                end
            end

            private

            def which_dir( global, user )
                #FIXME: we should test if we're user "puppet"
                #       there's a comment that suggests that we do that
                #       and we currently don't.
                expand_path case
                when name == :master; global
                when Puppet.features.root?; global
                else user
                end
            end

            def expand_path( dir )
                require 'etc'
                ENV["HOME"] ||= Etc.getpwuid(Process.uid).dir
                File.expand_path(dir)
            end

        end
    end
end