module Rep = Builder_db.Rep

open Grej.Infix

let broken_builds =
  Caqti_type.unit ->* Caqti_type.t3 (Rep.id `build) Rep.uuid Caqti_type.string @@
  {| SELECT b.id, b.uuid, job.name FROM build b, job
     WHERE result_kind = 0 AND result_code = 0 AND main_binary IS NOT NULL AND job.id = b.job AND
       (SELECT COUNT( * ) FROM build_artifact a
        WHERE a.build = b.id and a.filepath = b.main_binary) = 0
  |}

let fixup datadir (module Db : Caqti_blocking.CONNECTION) =
  Grej.check_version ~user_version:3L (module Db) >>= fun () ->
  Db.rev_collect_list broken_builds () >>= fun broken_builds ->
  Grej.list_iter_result
    (fun ((build, uuid, job_name) : [`build] Rep.id * Uuidm.t * string) ->
       Format.printf "Removing job %a.\nPlease clean up data files in %a/%s/%a\n"
         Uuidm.pp uuid Fpath.pp datadir job_name Uuidm.pp uuid;
       Db.exec Builder_db.Build.remove build)
    broken_builds