process/engines/itemrec/algorithms/scala/featurebased/src/main/scala/Realtime.scala

package io.prediction.algorithms.itemrec.featurebased

import grizzled.slf4j.Logger
import io.prediction.commons.Config
import com.twitter.scalding.Args
import io.prediction.commons.appdata.U2IAction
import io.prediction.commons.modeldata.ItemRecScore
import io.prediction.commons.modeldata.{ MetadataKeyvals, MetadataKeyval }

/*
 * - construct feature-list, Seq[String]
 * - construct user-feature-map : Map[User, Seq[Double]]
 * - modelCon: output UserFeatureMap in datastore.
 */

object UserProfileRecommendationRealtime {
  val logger = Logger(UserProfileRecommendationRealtime.getClass)
  val commonsConfig = new Config

  def modelCon(appid: Int, algoid: Int, modelset: Boolean,
    itypes: Seq[String],
    userFeaturesMap: Map[String, Seq[Double]]) {
    // Write to metadata-keyval
    val keyvalDb = commonsConfig.getModeldataMetadataKeyvals
    keyvalDb.upsert(algoid, modelset, "features", itypes.reduce(_ + "," + _))

    val userFeaturesDb = commonsConfig.getModeldataItemRecScores
    userFeaturesMap.foreach { case(user, features) => {
      userFeaturesDb.insert(ItemRecScore(
        uid = user,
        iids = Seq[String](),
        scores = features,
        itypes = Seq[Seq[String]](),
        appid = appid,
        algoid = algoid,
        modelset = modelset))
    }}
  }

  def run(appid: Int, algoid: Int, modelset: Boolean,
    optFeatureItypesStr: Option[String],
    whiteItypes: Seq[String]) = {
    val (userFeaturesMap, featureItypes, itemTypesMap, whiteItems) = (
      UserProfileRecommendation.constructUserFeaturesMapFromArg(
        appid, optFeatureItypesStr, whiteItypes))

    modelCon(appid, algoid, modelset, featureItypes, userFeaturesMap)
  }
  
  def main(cmdArgs: Array[String]) = {
    val args = Args(cmdArgs)

    val appid = args("appid").toInt
    val algoid = args("algoid").toInt
    val modelset = args("modelSet").toBoolean

    // FIXME. Take another itypes list (engine setting), filter items by that
    // itypes list
    val optFeatureItypesStr = args.optional("featureItypes")
    val whiteItypes = args.list("itypes")

    run(appid, algoid, modelset, optFeatureItypesStr, whiteItypes)
  }
}