gistfile1.txt

%% -*- erlang-indent-level: 4;indent-tabs-mode: nil; fill-column: 92 -*-
%% ex: ts=4 sw=4 et
%%% @author John Keiser <jkeiser@opscode.com>
%%% @copyright Copyright 2012-2012 Opscode Inc.
%%% @doc
%%% REST resource for creating and listing push jobs
%%% @end
-module(pushy_jobs_resource).

-export([init/1,
         allowed_methods/2,
         content_types_accepted/2,
         content_types_provided/2,
         is_authorized/2,
         from_json/2,
         to_json/2]).

-include("pushy_sql.hrl").

-include_lib("webmachine/include/webmachine.hrl").

-include_lib("eunit/include/eunit.hrl").

-record(config_state, {
          organization_guid :: string() }).

init(_Config) ->
    % ?debugVal(_Config),
    State = #config_state{},
    {{trace, "/tmp"}, State}.
%    {ok, State}.
%%    {{trace, "/tmp/traces"}, State}.
%% then in console: wmtrace_resource:add_dispatch_rule("wmtrace", "/tmp/traces").
%% then go to localhost:WXYZ/wmtrace

is_authorized(Req, State) ->
    OrgName =  wrq:path_info(organization_id, Req),
    %?debugVal(OrgName),
    State2 = State#config_state{organization_guid = pushy_object:fetch_org_id(OrgName) },
    {true, Req, State2}.

allowed_methods(Req, State) ->
    {['POST'], Req, State}.

content_types_accepted(Req, State) ->
    {[{"application/json", from_json}], Req, State}.

content_types_provided(Req, State) ->
    {[{"application/json", to_json}], Req, State}.

% {
%   'command' = 'chef-client',
%   'nodes' = [ 'DERPY', 'RAINBOWDASH' ]
% }

from_json(Req, #config_state{organization_guid = OrgId} = State) ->
    io:format("from_json~n"),
    [ Command, NodeNames ] = parse_post_body(Req),
    Job = create_job(OrgId, Command, NodeNames),
    pushy_job_runner_sup:execute(Job#pushy_job.id),
    {true, Req, State}.

to_json(Req, State) ->
    {"{}", Req, State}.

parse_post_body(Req) ->
    Body = wrq:req_body(Req),
    JobJson = ejson:decode(Body),
    Command = ej:get(<<"command">>, JobJson),
    NodeNames = ej:get(<<"nodes">>, JobJson),
    [ Command, NodeNames ].

create_job(OrgId, Command, NodeNames) ->
    Job = pushy_object:new_record(pushy_job, OrgId, NodeNames),
    Job1 = Job#pushy_job{command = Command, duration= 300},
    pushy_object:create_object(create_job, Job1, <<"bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb">>),
    Job1.