org.fusesource.scalate.spring.view.ScalateViewResolver.scala Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of scalate-spring-mvc Show documentation

Spring MVC integration for a Scalate web application

The newest version!

/**
 * Copyright (C) 2009-2011 the original author or authors.
 * See the notice.md file distributed with this work for additional
 * information regarding copyright ownership.
 *
 * 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 org.fusesource.scalate.spring.view

import java.util.Locale
import org.springframework.web.servlet.View
import org.springframework.web.servlet.view.AbstractTemplateViewResolver
import org.springframework.web.servlet.view.AbstractUrlBasedView

class ScalateViewResolver() extends AbstractTemplateViewResolver {

  setViewClass(requiredViewClass())

  override def requiredViewClass(): java.lang.Class[_] = classOf[org.fusesource.scalate.spring.view.ScalateView]
  
  override def buildView(viewName: String): AbstractUrlBasedView = {
    var view: AbstractScalateView = null

    if (viewName == "view") {
      view = new ScalateView
    } else if (viewName.startsWith("layout:")) {
      val urlView = new ScalateUrlView with LayoutScalateRenderStrategy
      urlView.setUrl(getPrefix() + viewName.substring("layout:".length()) + getSuffix())
      view = urlView
    } else {
      val urlView = new ScalateUrlView with DefaultScalateRenderStrategy
      urlView.setUrl(getPrefix() + viewName + getSuffix())
      view = urlView
    }

    val contentType = getContentType
    if (contentType != null) {
      view.setContentType(contentType)
    }

    view.asInstanceOf[AbstractUrlBasedView]
  }

}