aboutsummaryrefslogtreecommitdiff
path: root/captain/api.f90
blob: 2f0a7469a10cb4b8ee3fa3e690cc92f2be5901bc (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
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
! Copyright (c) 2021 Approximatrix, LLC <support@approximatrix.com>
!
! Permission is hereby granted, free of charge, to any person obtaining a copy
! of this software and associated documentation files (the "Software"), to deal
! in the Software without restriction, including without limitation the rights
! to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
! copies of the Software, and to permit persons to whom the Software is
! furnished to do so, subject to the following conditions:
!
! The above copyright notice and this permission notice shall be included in
! all copies or substantial portions of the Software.
!
! The Software shall be used for Good, not Evil.
!
! THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
! IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
! FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
! AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
! LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
! OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
! SOFTWARE.

module api_handling
use iso_c_binding
implicit none

    character(*), parameter::RESPONSE_JSON_IDLE = '{"status": "idle"}'
    character(*), parameter::RESPONSE_JSON_WORK_AVAILABLE = &
        '{"status": "pending",'//c_new_line//' "job": {job_number},'//c_new_line//' "instruction": "{instruction_name}"}'

contains

    function build_job_available_json(job_id) result(json_text)
    use captain_db
    use utilities, only: replace_field
    implicit none

        integer, intent(in)::job_id
        character(len=:), pointer::json_text
        integer::instruction_id
        character(PLAYER_NAME_LENGTH)::instruction_name
        
        instruction_id = get_job_instruction(job_id)
        call get_instruction_name(instruction_id, instruction_name)
        
        allocate(character(len=(len(RESPONSE_JSON_WORK_AVAILABLE)+len_trim(instruction_name)+32)) :: json_text)
        json_text = RESPONSE_JSON_WORK_AVAILABLE
        call replace_field(json_text, "job_number", job_id)
        call replace_field(json_text, "instruction_name", instruction_name)

    end function build_job_available_json
    
    subroutine handle_task_request(req)
    use server_response
    use captain_db
    use logging
    implicit none
    
        class(request)::req
        integer::job_i, task_i
        
        if(associated(req%query_string)) then
            job_i = req%path_component_int(5)
            task_i = req%path_component_int(7)
            call write_log("Task Update Is "//trim(req%query_string), LOG_DEBUG)
            if(req%query_string == "starting") then
                call write_log("Inserting task", LOG_DEBUG)
                call insert_task(job_i, task_i)
                call update_job_status(job_i, JOB_STATUS_WORKING)
            else if(req%query_string == "inprogress") then
                call update_task_status(job_i, task_i, JOB_STATUS_WORKING)
                call update_job_status(job_i, JOB_STATUS_WORKING)
            else if(req%query_string == "complete") then
                call update_task_status(job_i, task_i, JOB_STATUS_SUCCESS)
            else if(req%query_string == "failed") then
                call update_task_status(job_i, task_i, JOB_STATUS_FAILURE)
            end if
        end if
        
    end subroutine handle_task_request
    
    function api_request_gemini(req) result(resp)
    use server_response
    use captain_db
    use special_filenames
    use logging
    implicit none
    
        type(request), intent(in)::req
        type(response)::resp
        
        character(PLAYER_NAME_LENGTH)::player, instruction
        integer::job_i, player_i
        
        character(len=:), pointer::checkin_work_json

        ! Complete - "/api/player/{name}/job/{jobid}/complete"
        ! Failed   - "/api/player/{name}/job/{jobid}/failed"
        ! Task     - "/api/player/{name}/job/{jobid}/task/{task num}"
        if(trim(req%component(2)) == "player" .and. trim(req%component(4)) == "job") then
            job_i = req%path_component_int(5)
            call write_log("Job "//trim(req%component(5))//" update arrived", LOG_INFO)
            
            write(player, *) job_i
            
            if(.not. is_final_job_status(job_i)) then
                if(trim(req%component(6)) == "complete") then
                    call update_job_status(job_i, JOB_STATUS_SUCCESS)
                else if(trim(req%component(6)) == "failure") then
                    call update_job_status(job_i, JOB_STATUS_FAILURE)
                end if
            end if
            
            if(trim(req%component(6)) == "task") then
                call write_log("Task update encountered", LOG_INFO)
                call handle_task_request(req)
            end if
            
            resp%code = GEMINI_CODE_SUCCESS
            call resp%set_body_contents(RESPONSE_JSON_OKAY)
            resp%body_mimetype = "text/plain"
         
        ! Checkin - /api/player/{name}/checkin.json
        else if(trim(req%component(2)) == "player" .and. trim(req%component(4)) == "checkin.json") then
            ! Check for pending jobs
            call req%path_component(3, player)
            player_i = get_player_id(player)
            
            ! If we have a checkin, but the worker should have a job in progress, mark
            ! the jobs as failed.
            call mark_working_jobs_as_failed(player_i)
            
            job_i = get_pending_job_for_player(player_i)
            if(job_i < 0) then
                resp%code = GEMINI_CODE_SUCCESS
                call resp%set_body_contents(RESPONSE_JSON_IDLE) 
            else
                checkin_work_json => build_job_available_json(job_i)
                if(associated(checkin_work_json)) then
                    resp%code = GEMINI_CODE_SUCCESS
                    call write_log("Sending: "//trim(checkin_work_json), LOG_DEBUG)
                    call resp%set_body_contents(trim(checkin_work_json), "text/gemini")
                    deallocate(checkin_work_json)
                else
                    resp%code = GEMINI_CODE_PERMFAIL
                end if
            end if
        
        ! Instruction - /api/instructions/{name}
        else if(trim(req%component(2)) == "instruction") then
            
            call req%path_component(3, instruction)
            resp%body_filename => get_special_full_filename("instructions", trim(instruction)//".json")
            if(associated(resp%body_filename)) then
                resp%temporary_file = .false.
                resp%code = GEMINI_CODE_SUCCESS
                resp%body_mimetype = "text/plain"
            else
                resp%code = GEMINI_CODE_PERMFAIL
            end if
            
        end if
    
    end function api_request_gemini

    function api_request_titan(req) result(resp)
    use server_response
    use special_filenames
    use logging
    implicit none
    
        type(titan_request), intent(in)::req
        type(response)::resp
        
        character(len=:), pointer::fullpath
        character(12)::job_text, task_text
        integer::job_id, task_num
        
        character(64)::msg
        
        fullpath => null()
        
        call write_log("Titan request encountered", LOG_INFO)
        
        ! Task     - "/api/player/{name}/job/{jobid}/task/{task num}"
        if(trim(req%component(2)) == "player" .and. &
           trim(req%component(4)) == "job" .and. &
           trim(req%component(6)) == "task") then
           
            job_id = req%path_component_int(5)

            task_num = req%path_component_int(7)

            write(job_text, '(I6)') job_id
            write(task_text, '(I6)') task_num
            
            call write_log("Handling a task update for job "//trim(job_text)//" task "//trim(task_text), LOG_INFO)
            call handle_task_request(req)
            
            fullpath => get_task_result_static_filename(job_id, task_num)

        end if
            
        if(associated(fullpath)) then
        
            ! Write the file
            call write_log("Storing titan file to "//trim(fullpath), LOG_DEBUG)
            
            call req%write_to(fullpath)
        
            resp%code = GEMINI_CODE_SUCCESS
            call resp%set_body_contents(RESPONSE_JSON_OKAY)
            resp%body_mimetype = "text/plain"
        
        else
            
            resp%code = GEMINI_CODE_PERMFAIL
        
        end if
        
    end function api_request_titan

end module api_handling