#' Generate actor search queries based on data in actor db
#'
#' Generate actor search queries based on data in actor db
#' @param actor A row from the output of elasticizer() when run on the 'actor' index
#' @param country 2-letter string indicating the country for which to generate the queries, is related to inflected nouns, definitive forms and genitive forms of names etc.
#' @return A data frame containing the queries, related actor ids and actor function
#' @export
#' @examples
#' query_gen_actors(actor,country)

#################################################################################################
#################################### Actor search query generator ###############################
#################################################################################################
query_gen_actors <- function(actor, country) {
  highlight <- paste0('"highlight" : {
                      "fields" : {
                      "text" : {},
                      "teaser" : {},
                      "preteaser" : {},
                      "title" : {},
                      "subtitle" : {}
                      },
                      "number_of_fragments": 0,
                      "order": "none",
                      "type":"unified",
                      "fragment_size":0,
                      "pre_tags":"',identifier,'",
                      "post_tags": ""
}')
if (country == "no") {
  genitive <- 's'
  definitive <- 'en'
  definitive_genitive <- 'ens'
} else {
  genitive <- ''
  definitive <- ''
  definitive_genitive <- ''
}

  if (actor$`_source.function` == "Min" | actor$`_source.function` == "PM" | actor$`_source.function` == "PartyLeader") {
    lastname <- paste0('(',actor$`_source.lastName`,' OR ',actor$`_source.lastName`,genitive,')')
    ## Adding a separate AND clause for inclusion of only last name to highlight all occurences of last name
    ## Regardless of whether the last name hit is because of a minister name or a full name proximity hit
    query_string <- paste0('(((\\"',
                           actor$`_source.firstName`,
                           ' ',
                           actor$`_source.lastName`,
                           '\\"~5 OR \\"',
                           actor$`_source.firstName`,
                           ' ',
                           actor$`_source.lastName`,genitive,
                           '\\"~5) AND ',lastname)
  }
  if (actor$`_source.function` == "PartyLeader") {
    query_string <- paste0(query_string,'))')
    ids <- toJSON(unlist(lapply(c(actor$`_source.actorId`,actor$`_source.partyId`),str_c, "_pl")))
  }
  if (actor$`_source.function` == "Min" | actor$`_source.function` == "PM") {
    ## Modifiers are only applied to minister titles
    capital <- unlist(lapply(actor$`_source.ministerSearch`, str_to_title))
    capital_gen <- unlist(lapply(capital, str_c, genitive))
    capital_def <- unlist(lapply(capital, str_c, definitive))
    capital_defgen <- unlist(lapply(capital, str_c, definitive_genitive))
    gen <- unlist(lapply(actor$`_source.ministerSearch`, str_c, genitive))
    def <- unlist(lapply(actor$`_source.ministerSearch`, str_c, definitive))
    defgen <- unlist(lapply(actor$`_source.ministerSearch`, str_c, definitive_genitive))
    names <- paste(c(capital,capital_gen,gen,capital_def,def,defgen,capital_defgen), collapse = ' ')
    query_string <- paste0(query_string,') OR (',lastname,' AND (',names,')))')
    ids <- toJSON(unlist(lapply(c(actor$`_source.actorId`,actor$`_source.ministryId`,actor$`_source.partyId`),str_c, "_min")))
  }
  if (actor$`_source.function` == "Party") {
    actor$`_source.startDate` <- "2000-01-01"
    actor$`_source.endDate` <- "2099-01-01"
    names <- paste(c(unlist(actor$`_source.partyNameSearchShort`)), collapse = '\\" \\"')
    query_string <- paste0('(\\"',names,'\\")')
    query <- paste0('{"query":
                    {"bool": {"filter":[{"term":{"country":"',country,'"}},
                    {"range":{"publication_date":{"gte":"',actor$`_source.startDate`,'","lte":"',actor$`_source.endDate`,'"}}},
                    {"query_string" : {
                    "default_operator" : "OR",
                    "allow_leading_wildcard" : "false",
                    "fields": ["text","teaser","preteaser","title","subtitle"],
                    "query" : "', query_string,'"
                    }
                    }
                    ]
                    } },',highlight,' }')
    ids <- c(toJSON(unlist(lapply(c(actor$`_source.partyId`),str_c, "_p"))))

    if (nchar(actor$`_source.partyNameSearch`[[1]]) > 0) {
      names <- paste(c(unlist(actor$`_source.partyNameSearch`)), collapse = '\\" \\"')
      query_string <- paste0('(\\"',names,'\\")')
      query2 <- paste0('{"query":
                       {"bool": {"filter":[{"term":{"country":"',country,'"}},
                       {"range":{"publication_date":{"gte":"',actor$`_source.startDate`,'","lte":"',actor$`_source.endDate`,'"}}},
                       {"query_string" : {
                       "default_operator" : "OR",
                       "allow_leading_wildcard" : "false",
                       "fields": ["text","teaser","preteaser","title","subtitle"],
                       "query" : "', query_string,'"
                       }
                       }
                       ]
                       } },',highlight,' }')


    ids <- c(ids, toJSON(unlist(lapply(c(actor$`_source.partyId`),str_c, "_p"))))
    query <- c(query, query2)
    fn <- c('PartyAbbreviation','Party')
    } else {
      fn <- c('PartyAbbreviation')
    }
    return(data.frame(query = query, ids = I(ids), type = fn, prefix = actor$`_source.searchAnd`, postfix = actor$`_source.searchAndNot`, stringsAsFactors = F))
  }

  query <- paste0('{"query":
                  {"bool": {"filter":[{"term":{"country":"',country,'"}},
                  {"range":{"publication_date":{"gte":"',actor$`_source.startDate`,'","lte":"',actor$`_source.endDate`,'"}}},
                  {"query_string" : {
                  "default_operator" : "OR",
                  "allow_leading_wildcard" : "false",
                  "fields": ["text","teaser","preteaser","title","subtitle"],
                  "query" : "', query_string,'"
                  }
                  }
                  ]
                  } },',highlight,' }')
  fn <- actor$`_source.function`
  return(data.frame(query = query, ids = I(ids), type = fn, stringsAsFactors = F))
}