/*
 * Copyright 2011 Tomas Zeman <tzeman@volny.cz>
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package bootstrap.liftweb

import fis.base.model._
import net.liftweb.common._
import net.liftweb.db.{DB, ConnectionIdentifier}
import net.liftweb.http._
import net.liftweb.squerylrecord.SquerylRecord
import net.liftweb.util._
import net.liftweb.util.Helpers._
import net.tz.lift.boot.ProtoBoot
import net.tz.lift.util.StandardDBVendor

class Boot extends ProtoBoot {
  override def boot = {

    /* DB stuff */
    val dbVendor = new FisDbVendor
    DB.defineConnectionManager(FisConnectionIdentifier, dbVendor)
    S.addAround(DB.buildLoanWrapper(List(FisConnectionIdentifier)))
    SquerylRecord.init(() => new SeqIdPostgreSqlAdapter)
    SquerylTxMgr.initSqueryl(FisConnectionIdentifier,
      () => new SeqIdPostgreSqlAdapter)
    LiftRules.unloadHooks.append(dbVendor.closeAllConnections_! _)

    super.boot

  }
}

case object FisConnectionIdentifier extends ConnectionIdentifier {
  val jndiName = "fis"
}

import java.sql.Connection

class FisDbVendor extends StandardDBVendor(
  "org.postgresql.Driver", Props.get("db.fis.url", ""),
  Props.get("db.fis.user"), Props.get("db.fis.pass")) {

  val testQuery = "SELECT version()"
  override protected def testConnection(c: Connection) = {
    c.prepareStatement(testQuery).executeQuery
  }
}

import org.squeryl._
import org.squeryl.internals.DatabaseAdapter

object SquerylTxMgr {
  def initSqueryl(name: ConnectionIdentifier,
    mkAdapter: () => DatabaseAdapter) = {
    SessionFactory.externalTransactionManagementAdapter =
    Some(new SquerylTxMgr(name, mkAdapter))
  }
}

class SquerylTxMgr(name: ConnectionIdentifier, mkAdapter: () => DatabaseAdapter)
  extends Loggable with Function0[Session] {

  private object currentSession extends DynoVar[Session]


  def apply = currentSession.is openOr {
    DB.use(name) { superConn =>
      val sess = Session.create(superConn.connection, mkAdapter())
      sess.setLogger(s => logger.debug(s))
      currentSession.set(sess)
      sess
    }
  }
}

// vim: set ts=2 sw=2 et: