#
# CDDL HEADER START
#
# The contents of this file are subject to the terms of the
# Common Development and Distribution License (the "License").
# You may not use this file except in compliance with the License.
#
# You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
# or http://www.opensolaris.org/os/licensing.
# See the License for the specific language governing permissions
# and limitations under the License.
#
# When distributing Covered Code, include this CDDL HEADER in each
# file and include the License file at usr/src/OPENSOLARIS.LICENSE.
# If applicable, add the following below this CDDL HEADER, with the
# fields enclosed by brackets "[]" replaced with your own identifying
# information: Portions Copyright [yyyy] [name of copyright owner]
#
# CDDL HEADER END
#

#
# Copyright (c) 2013, 2014, Oracle and/or its affiliates. All rights reserved.
#

Puppet::Type.type(:nis).provide(:nis) do
    desc "Provider for management of NIS client for Oracle Solaris"
    confine :operatingsystem => [:solaris]
    defaultfor :osfamily => :solaris, :kernelrelease => ['5.11', '5.12']
    commands :svccfg => '/usr/sbin/svccfg', :svcprop => '/usr/bin/svcprop'

    class << self; attr_accessor :client_fmri, :domain_fmri end
    Client_fmri = "svc:/network/nis/client"
    Domain_fmri = "svc:/network/nis/domain"

    def initialize(value={})
        super(value)
        @client_refresh = false
        @domain_refresh = false
    end

    def self.instances
        props = {}
        validprops = Puppet::Type.type(:nis).validproperties

        [Client_fmri, Domain_fmri].each do |svc|
            svcprop("-p", "config", svc).split("\n").collect do |line|
                data = line.split()
                fullprop = data[0]
                type = data[1]
                if data.length > 2
                    value = data[2..-1].join(" ")
                else
                    value = nil
                end

                pg, prop = fullprop.split("/")
                props[prop] = value if validprops.include? prop.to_sym
            end
        end
        props[:name] = "current"
        return Array new(props)
    end

    # svc:/network/nis/client properties
    [:use_broadcast, :use_ypsetme].each do |field|
        define_method(field) do
            begin
                svcprop("-p", "config/" + field.to_s, Client_fmri).strip()
            rescue
                # if the property isn't set, don't raise an error
                nil
            end
        end

        define_method(field.to_s + "=") do |should|
            begin
                if not should.to_s.empty?
                    value = "\"" + should.to_s + "\""
                else
                    value = should.to_s
                end
                svccfg("-s", Client_fmri, "setprop", "config/" + field.to_s,
                       "=", value)
            rescue => detail
                raise Puppet::Error,
                    "Unable to set #{field.to_s} to #{should.inspect}\n"
                    "#{detail}\n"
            end
            @client_refresh = true
        end
    end

    # svc:/network/nis/domain properties
    [:domainname, :ypservers, :securenets].each do |field|
        define_method(field) do
            begin
                svcprop("-p", "config/" + field.to_s, Domain_fmri).strip()
            rescue
                # if the property isn't set, don't raise an error
                nil
            end
        end

        define_method(field.to_s + "=") do |should|
            begin
                if should.is_a? Array
                    # in Solaris 11, the list of values needs to be single
                    # argument to svccfg.
                    values = ""
                    for entry in should
                        values += "\"#{entry}\" "
                    end
                    values = "(" + values + ")"
                    svccfg("-s", Domain_fmri, "setprop",
                           "config/" + field.to_s, "=", values)
                else
                    # Puppet seems to get confused about when to pass an empty
                    # string or "\"\"".  Catch either condition to handle
                    # passing values to SMF correctly
                    if should.to_s.empty? or should.to_s == '""'
                        value = should.to_s
                    else
                        value = "\"" + should.to_s + "\""
                    end
                    svccfg("-s", Domain_fmri, "setprop",
                           "config/" + field.to_s, "=", value)
                end
            rescue => detail
                raise Puppet::Error,
                    "Unable to set #{field.to_s} to #{should.inspect}\n"
                    "#{detail}\n"
            end
            @domain_refresh = true
        end
    end

    def flush
        if @domain_refresh == true
            svccfg("-s", Domain_fmri, "refresh")
        end
        if @client_refresh == true
            svccfg("-s", Client_fmri, "refresh")
        end
    end
end