summaryrefslogtreecommitdiff
path: root/chef/lib/chef/provider/cron.rb
blob: 9e633053ce357eb1ecb21e39ee751d4fc3806471 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
#
# Author:: Bryan McLellan (btm@loftninjas.org)
# Copyright:: Copyright (c) 2009 Bryan McLellan
# License:: Apache License, Version 2.0
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
# 
#     http://www.apache.org/licenses/LICENSE-2.0
# 
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#

require 'chef/log'
require 'chef/mixin/command'
require 'chef/provider'

class Chef
  class Provider
    class Cron < Chef::Provider
      include Chef::Mixin::Command

      def initialize(node, new_resource)
        super(node, new_resource)
        @cron_exists = false
        @cron_empty = false
      end
      attr_accessor :cron_exists, :cron_empty

      def load_current_resource
        crontab = String.new
        @current_resource = Chef::Resource::Cron.new(@new_resource.name)
        status = popen4("crontab -l -u #{@new_resource.user}") do |pid, stdin, stdout, stderr|
          stdout.each { |line| crontab << line }
        end
        if status.exitstatus > 1
          raise Chef::Exceptions::Cron, "Error determining state of #{@new_resource.name}, exit: #{status.exitstatus}"
        elsif status.exitstatus == 0
          crontab.each do |line|
            case line
            when /^# Chef Name: #{@new_resource.name}/
              Chef::Log.debug("Found cron '#{@new_resource.name}'")
              @cron_exists = true
            end
          end
          Chef::Log.debug("Cron '#{@new_resource.name}' not found") unless @cron_exists
        elsif status.exitstatus == 1
          Chef::Log.debug("Cron empty for '#{@new_resource.user}'")
          @cron_empty = true
        end
        
        @current_resource
      end

      def action_create
        crontab = String.new
        cron_found = false
        if @cron_exists
          status = popen4("crontab -l -u #{@new_resource.user}") do |pid, stdin, stdout, stderr|
            stdout.each_line do |line|
              if cron_found
                cronline = "#{@new_resource.minute} #{@new_resource.hour} #{@new_resource.day} #{@new_resource.month} #{@new_resource.weekday} #{@new_resource.command}\n"
                if (line == cronline)
                  Chef::Log.debug("Skipping existing cron entry '#{@new_resource.name}'")
                  return
                end
                crontab << cronline
                cron_found = false
                next
              end
              case line
              when /^# Chef Name: #{new_resource.name}\n/
                cron_found = true
              end
              crontab << line 
            end
          end


          status = popen4("crontab -u #{@new_resource.user} -", :waitlast => true) do |pid, stdin, stdout, stderr|
            crontab.each { |line| stdin.puts "#{line}" }
            stdin.close
          end
          Chef::Log.info("Updated cron '#{@new_resource.name}'")
        else
          unless @cron_empty
            status = popen4("crontab -l -u #{@new_resource.user}") do |pid, stdin, stdout, stderr|
              stdout.each { |line| crontab << line }
            end
          end
  
          crontab << "# Chef Name: #{new_resource.name}\n"
          crontab << "#{@new_resource.minute} #{@new_resource.hour} #{@new_resource.day} #{@new_resource.month} #{@new_resource.weekday} #{@new_resource.command}\n"
  
          status = popen4("crontab -u #{@new_resource.user} -", :waitlast => true) do |pid, stdin, stdout, stderr|
            crontab.each { |line| stdin.puts "#{line}" }
            stdin.close
          end
          Chef::Log.info("Added cron '#{@new_resource.name}'")
        end
      end

      def action_delete
        if @cron_exists
          crontab = String.new
          cron_found = false
          status = popen4("crontab -l -u #{@new_resource.user}") do |pid, stdin, stdout, stderr|
            stdout.each_line do |line|
              if cron_found
                cron_found = false
                next
              end
              case line
              when /^# Chef Name: #{new_resource.name}\n/
                cron_found = true
                next
              end
              crontab << line 
            end
          end

          status = popen4("crontab -u #{@new_resource.user} -", :waitlast => true) do |pid, stdin, stdout, stderr|
            crontab.each { |line| stdin.puts "#{line}" }
            stdin.close
          end
          Chef::Log.debug("Deleted cron '#{@new_resource.name}'")
        end
      end

    end
  end
end