puppetlabs/vmpooler
2
0
Fork 0
mirror of https://github.com/puppetlabs/vmpooler.git synced 2026-01-26 10:08:40 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

Slight rework of data param validation

Browse source
This commit is contained in:
Scott Schneider 2015-03-04 09:31:56 -08:00
parent c1099a8273
commit 0c3f84e072
1 changed files with 9 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

25
lib/vmpooler/api/v1.rb
View file

@ -124,29 +124,22 @@ module Vmpooler
daily: [] # used for daily info
}
from_param, to_param = params[:from], params[:to]
from_param = params[:from] || Date.today.to_s
to_param = params[:to] || Date.today.to_s
# Check date formats.
# It is ok if to_param is nil, we will assume this means 'today'
if to_param.nil?
to_param = Date.today.to_s
elsif !validate_date_str(to_param.to_s)
halt 400, 'Invalid "to" date format, must match YYYY-MM-DD'
end
# It is also ok if from_param is nil, we will assume this also means 'today'
if from_param.nil?
from_param = Date.today.to_s
elsif !validate_date_str(from_param.to_s)
halt 400, 'Invalid "from" date format, must match YYYY-MM-DD'
# Validate date formats
[from_param, to_param].each do |param|
if !validate_date_str(param.to_s)
halt 400, "Invalid date format '#{param}', must match YYYY-MM-DD."
end
end
from_date, to_date = Date.parse(from_param), Date.parse(to_param)
if to_date < from_date
halt 400, 'Date range is invalid, "to" cannot come before "from".'
halt 400, 'Date range is invalid, \'to\' cannot come before \'from\'.'
elsif from_date > Date.today
halt 400, 'Date range is invalid, "from" must be in the past.'
halt 400, 'Date range is invalid, \'from\' must be in the past.'
end
# total clone time for entire duration requested.