/*

 * 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: