src/main/scala/fis/aaa/model/UserField.scala
changeset 56 9409e7ab3f9d
child 61 b65843860274
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/src/main/scala/fis/aaa/model/UserField.scala	Thu Apr 12 17:07:35 2012 +0200
@@ -0,0 +1,106 @@
+/*
+ * Copyright 2012 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 fis.aaa.model
+
+import net.liftweb.common._
+import net.liftweb.common.Box._
+import net.liftweb.http.SHtml
+import net.liftweb.record.{LifecycleCallbacks, Record}
+import net.liftweb.record.field.{OptionalLongField, LongField}
+import net.liftweb.util.Vendor
+import net.tz.lift.model._
+import net.tz.lift.model.{FieldLabel => FL}
+import scala.xml.NodeSeq
+
+/**
+ * User field.
+ */
+class UserField[T <: Record[T]](own: T, l: Vendor[Iterable[User]],
+  u: Vendor[Long => Box[User]]) extends LongField(own) {
+
+  override def asHtml = user.dmap(NodeSeq.Empty)(_.name.asHtml)
+  override def toForm = Full(SHtml.selectObj[Long](
+    l() map { u => (u.id, u.name.get) } toSeq, valueBox, set(_)))
+  def user: Box[User] = u()(get)
+}
+
+/**
+ * Optional user field.
+ */
+class OptionalUserField[T <: Record[T]](own: T, l: Vendor[Iterable[User]],
+  u: Vendor[Long => Box[User]]) extends OptionalLongField(own) {
+
+  override def asHtml = user.dmap(NodeSeq.Empty)(_.name.asHtml)
+  override def toForm = Full(SHtml.selectObj[Box[Long]](
+    (Empty, emptyLabel) +: (l() map { u => (Full(u.id), u.name.get) } toSeq),
+    Full(valueBox), setBox(_)))
+  def user: Box[User] = get flatMap { u()(_) }
+  def emptyLabel: String = "------"
+}
+
+/**
+ * CreatedBy field.
+ */
+class CreatedByField[T <: Record[T]](own: T, cur: Vendor[Box[User]],
+  u: Vendor[Long => Box[User]]) extends OptionalLongField(own) with
+  LifecycleCallbacks {
+
+  override def asHtml = user.dmap(NodeSeq.Empty)(_.name.asHtml)
+  override def defaultValue = cur().dmap(super.defaultValue)(_.id) 
+  override def displayName = l10n(name)
+  override def beforeCreate {
+    set(cur().map { _.id })
+  }
+  def user: Box[User] = get flatMap { u()(_) }
+}
+
+/**
+ * UpdatedBy field.
+ */
+class UpdatedByField[T <: Record[T]](own: T, cur: Vendor[Box[User]],
+  u: Vendor[Long => Box[User]]) extends OptionalLongField(own) with
+  LifecycleCallbacks {
+
+  override def asHtml = user.dmap(NodeSeq.Empty)(_.name.asHtml)
+  override def defaultValue = cur().dmap(super.defaultValue)(_.id)
+  override def beforeSave {
+    set(cur().map { _.id })
+  }
+  override def displayName = l10n(name)
+  def user: Box[User] = get flatMap { u()(_) }
+}
+
+/**
+ * Created/UpdatedBy mixin.
+ */
+trait CreatedUpdatedBy[T <: Record[T]] { self: T =>
+  val createdBy = new CreatedByField(this.asInstanceOf[T], curUserVendor,
+    getUserVendor)
+  val updatedBy = new UpdatedByField(this.asInstanceOf[T], curUserVendor,
+    getUserVendor)
+
+  protected def curUserVendor: Vendor[Box[User]] = User.cur
+  protected def getUserVendor: Vendor[Long => Box[User]] = User.byId
+}
+
+/**
+ * Mixin for User field to have current user as default.
+ */
+trait DefaultCurUser { self: UserField[_] => 
+  override def defaultValue: Long = User.get.map(_.id).getOrElse(0L)
+}
+
+// vim: set ts=2 sw=2 et: