open StdExtra
open Helios_datatypes_t
open Eliom_content.Html5.F

let site_title = "Helios Election Server"
let welcome_message = "This is the default message"

let s x = Xml.uri_of_string ("/static/" ^ x)

let format_user u size = Helios_services.([
  img
    ~src:(Printf.ksprintf s "auth/login-icons/%s.png" u.user_type)
    ~a:[a_style "border:0;"; a_height size]
    ~alt:u.user_type ();
  pcdata " ";
  pcdata u.user_name;
])

let base ~title ~header ~content =
  lwt user = Eliom_reference.get Helios_services.user in
  Lwt.return (html ~a:[a_dir `Ltr; a_xml_lang "en"]
    (head (Eliom_content.Html5.F.title (pcdata (title ^ " - Helios"))) [
      link
        ~rel:[`Stylesheet]
        ~href:(s "main.css")
        ~a:[a_mime_type "text/css"; a_media [`Screen]]
        ();
      link
        ~rel:[`Stylesheet]
        ~href:(s "helios/css/ui-lightness/jquery-ui-1.8.1.custom.css")
        ~a:[a_mime_type "text/css"]
        ();
      script (pcdata "") ~a:[a_src (s "helios/js/jquery-1.4.2.min.js")];
      script (pcdata "") ~a:[a_src (s "helios/js/jquery-ui-1.8.1.custom.min.js")];
      script (pcdata "") ~a:[a_src (s "helios/js/jqsplitdatetime.js")];
      script (pcdata "") ~a:[a_src (s "helios/helios/jquery.json.min.js")];
      (* block js *)
      (* block extra-head *)
    ])
    (body [
      div ~a:[a_id "content"] [
        div ~a:[a_id "header"] ([
          a ~service:Helios_services.home [
            img
              ~src:(s "logo.gif")
              ~a:[a_style "border:0;"; a_height 110]
              ~alt:"Helios" ()
          ] ();
          br ();
        ] @ header);
        div ~a:[a_id "contentbody"] content;
        div ~a:[a_id "footer"] (
          [span ~a:[a_style "float:right;"] [ (* footer logo *) ]] @
          (match user with
            | Some (admin_p, user) ->
              [pcdata "logged in as "] @ (format_user user 15) @ [
                pcdata " [";
                a ~service:Helios_services.logout [pcdata "logout"] ();
                pcdata "]";
                br ()
              ]
            | None ->
              [pcdata "not logged in."] @ [
                pcdata " [";
                a ~service:Helios_services.login [pcdata "log in"] ();
                pcdata "]";
                br ();
              ]
          ) @ [
            a ~service:Helios_services.project_home [
              pcdata "About Helios"
            ] ();
            pcdata " | Help!";
          (* footer links *)
            br ~a:[a_style "clear:right;"] ();
          ]
        )
      ];
     ]))

let not_implemented title = base
  ~title
  ~header:[h2 [pcdata title]]
  ~content:[div [pcdata "This service is not implemented."]]

let login_box auth_systems = List.map
  (fun x ->
    p [
      a
        ~service:Helios_services.login
        ~a:[a_style "font-size: 1.4em;"] [
          img
            ~a:[a_style "border:0;"; a_height 35]
            ~src:(Printf.ksprintf s "auth/login-icons/%s.png" x)
            ~alt:x ();
          pcdata x;
        ] ();
    ]
  ) auth_systems

type answer = {
  count : int;
  answer : string;
  winner : bool;
}

type question = {
  answers : answer list;
  question : string;
}

type election_extradata = {
  xelection : Helios_services.election_data;
  election : Z.t Helios_datatypes_t.election;
  (* FIXME: datatypes should be revisited, election is xelection.election! *)
  election_admin : Helios_services.user;
  election_trustees : string list;
  election_state : [`Finished of question list | `Stopped | `Started];
}

let format_one_election e =
  li [pcdata e.election.e_name]

let format_one_featured_election e =
  [
    div ~a:[a_class ["highlight-box-margin"]] ([
      a
        ~service:(Eliom_service.preapply
                    Helios_services.election_shortcut
                    e.election.e_short_name)
        ~a:[a_style "font-size: 1.4em;"]
        [pcdata e.election.e_name] ();
      pcdata " by ";
    ] @ format_user e.election_admin 15 @ [
      br ();
      pcdata e.election.e_description;
    ]);
    br ();
  ]

let index ~featured =
  lwt user = Eliom_reference.get Helios_services.user in
  base
  ~title:site_title
  ~header:[h2 [pcdata site_title]]
  ~content:(
    let mystuff = match user with
      | Some (admin_p, u) ->
        let administered = if admin_p then Some [] else None in
        let voted = [] in
        let administration_box = match administered with
          | Some admin ->
            let administered_box = match admin with
              | _::_ -> ul (List.map format_one_election admin)
              | [] -> em [pcdata "none yet"]
            in [
              h4 [pcdata "Administration"];
              administered_box;
              p [pcdata "[";
                 a ~service:Helios_services.elections_administered [
                   pcdata "see all"
                 ] ();
                 pcdata "]"];
              div ~a:[a_style "text-align:right;"] [
                a ~service:Helios_services.election_new
                  ~a:[a_style "font-size: 1.2em; padding:5px; background: #eee; border: 1px solid #888;"]
                  [
                    pcdata "create election >";
                  ] ();
              ]
            ]
          | None -> []
        in
        let recent_votes = [
          h4 [pcdata "Recent votes"];
          match voted with
          | _::_ -> ul (List.map format_one_election voted)
          | [] -> em [pcdata "none yet"]
        ] in
        [
          div ~a:[a_style "font-size:1.4em;"; a_class ["highlight-box"]]
            (format_user u 25)
        ]
        @ administration_box @ recent_votes
      | None ->
        [h3 [pcdata "Log In to Start Voting"]]
        @ (login_box Helios_services.auth_systems)
        @ [br (); br ()]
    in
    let featured_box = match featured with
      | _::_ ->
        [
          h3 [pcdata "Current Featured Elections"];
          div (List.flatten (List.map format_one_featured_election featured));
        ]
      | [] ->
        [
          h4 [pcdata "no featured elections at the moment"];
        ]
    in ([
      div ~a:[a_id "mystuff"] mystuff;
      p ~a:[a_style "font-size: 1.4em;"] [pcdata welcome_message];
    ] @ featured_box @ [
      br ~a:[a_style "clear:right;"] ();
      br ()
    ])
  )

let dummy_login ~service =
  let title = site_title ^ " — Login" in
  let form = post_form
    ~a:[a_id "login_form"; a_class ["prettyform"]]
    ~service
    (fun (username_name, admin_name) ->
      [
        tablex [tbody [
          tr [
            th [label ~a:[a_for username_name] [pcdata "Username:"]];
            td [string_input ~a:[a_maxlength 50] ~input_type:`Text ~name:username_name ()];
          ];
          tr [
            th [label ~a:[a_for admin_name] [pcdata "Admin?"]];
            td [bool_checkbox ~name:admin_name ()];
          ]]
        ];
        div [
          string_input ~input_type:`Submit ~value:"Login" ();
        ]
      ]) ()
  in
  base
    ~title
    ~header:[h2 [pcdata title]]
    ~content:[div [form]]

let election_view ~election =
  let service = Eliom_service.preapply Helios_services.election_raw election.election.e_uuid in
  let booth = Helios_services.make_booth election.election.e_uuid in
  let audit_info = [
    (* FIXME: unsafe_data *)
    unsafe_data "<a href=\"#\" onclick=\"$('#auditbody').slideToggle(250);\">Audit Info</a>";
    div ~a:[
      a_id "auditbody";
      a_style "display:none;";
    ] [
      br ();
      pcdata "Election URL:";
      br ();
      code ~a:[a_style "font-size: 1.2em;"] [
        a ~service [ pcdata (make_string_uri ~absolute:true ~service ()) ] ()
      ];
      br ();
      br ();
      pcdata "Election Fingerprint:";
      br ();
      code ~a:[a_style "font-size: 1.3em; font-weight: bold;"] [
        pcdata election.xelection.Helios_services.fingerprint;
      ];
      br ();
      br ();
      (* FIXME: Ballot Tracking Center *)
      (* FIXME: Audited Ballots *)
      (* FIXME: result *)
      p ~a:[a_style "font-size: 1.2em;"] [
        pcdata "Review the ";
        a ~service:booth [ pcdata "voting booth" ] ();
        pcdata ".";
      ];
    ]
  ] in
  let content = [
    div ~a:[a_style "float: left; margin-right: 50px;"] [pcdata "FIXME"];
    br ();
    br ();
    br ~a:[a_style "clear: left;"] ();
    div ~a:[a_style "margin-bottom: 25px;margin-left: 15px; border-left: 1px solid #aaa; padding-left: 5px; font-size:1.3em;"] [pcdata election.election.e_description];
    p ~a:[a_style "text-align: center; font-size: 1.5em;"] [
      a ~service:(Eliom_service.preapply Helios_services.election_questions election.election.e_uuid) [
        pcdata "questions (";
        pcdata (string_of_int (Array.length election.election.e_questions));
        pcdata ")";
      ] ();
      (* FIXME: space (&nbsp) breaks the output *)
      pcdata "  |  ";
      a ~service:(Eliom_service.preapply Helios_services.election_voters election.election.e_uuid) [
        pcdata "voters & ballots"
      ] ();
      pcdata "  |  ";
      a ~service:(Eliom_service.preapply Helios_services.election_trustees election.election.e_uuid) [
        pcdata "trustees (";
        pcdata (string_of_int (List.length election.election_trustees));
        pcdata ")";
      ] ();
    ];
    (* NOTE: administration things removed from here! *)
    br ();
    br ();
  ] @ (match election.election_state with
    | `Finished result ->
      [
        span ~a:[a_class ["highlight-box"; "round"]] [
          pcdata "This election is complete.";
        ];
        br ();
        br ();
        h3 ~a:[a_class ["highlight-box"]] [pcdata "Tally"];
      ] @ (
        List.iteri (fun i question ->
          [
            b [
              span ~a:[a_style "font-size:0.8em;"] [
                pcdata "Question #";
                pcdata (string_of_int i);
              ];
              br ();
              pcdata question.question;
            ];
            br ();
            let table xs = match xs with
              | x :: xs -> table ~a:[a_class ["pretty"]; a_style "width: auto;"] x xs
              | [] -> assert false
            in table (
              List.map (fun answer ->
                let style = if answer.winner then "font-weight:bold;" else "" in
                tr [
                  td ~a:[a_style ("padding-right:80px;" ^ style)] [pcdata answer.answer];
                  td ~a:[a_style ("text-align:right;" ^ style)] [pcdata (string_of_int answer.count)];
                ]
              ) question.answers
            );
          ]
        ) result
      )
    | `Stopped ->
      [
        span ~a:[a_class ["highlight-box"; "round"]] [
          pcdata "Election closed. Tally will be computed soon.";
        ];
        br ();
      ]
    | `Started ->
      [
        span ~a:[
          a_class ["highlight-box"; "round"];
          a_style "font-size: 1.6em; margin-right: 10px;";
          a_id "votelink";
        ] [
          a ~service:(Eliom_service.preapply Helios_services.election_vote election.election.e_uuid) [
            pcdata "Vote in this election";
          ] ()
        ];
        br ();
        br ();
        (* if election.voting_extended_until ... *)
        pcdata "This election ends at the administrator's discretion.";
        br ();
      ]
  ) @ [
    (* FIXME: privacity, eligibility, etc. *)
    div ~a:[
      a_style "background: lightyellow; padding:5px; padding-left: 10px; margin-top: 15px; border: 1px solid #aaa; width: 720px;";
      a_class ["round"];
    ] audit_info
  ] in
  base ~title:election.election.e_name ~header:[] ~content