laika.helium.builder.HeliumThemeBuilder.scala Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2012-2020 the original author or authors.
 *
 * 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 laika.helium.builder

import cats.effect.{Async, Resource}
import laika.ast.RewritePhase
import laika.bundle.BundleOrigin
import laika.directive.{Blocks, DirectiveRegistry, Links, Spans, Templates}
import laika.format.{EPUB, HTML, XSLFO}
import laika.helium.Helium
import laika.helium.generate._
import laika.theme.{Theme, ThemeBuilder, ThemeProvider}

/**
  * @author Jens Halm
  */
private[helium] class HeliumThemeBuilder (helium: Helium) extends ThemeProvider {

  object directives extends DirectiveRegistry {
    override def origin: BundleOrigin = BundleOrigin.Theme
    override val description: String = "Directives for theme 'Helium'"
    val spanDirectives: Seq[Spans.Directive] = Nil
    val blockDirectives: Seq[Blocks.Directive] = Nil
    val templateDirectives: Seq[Templates.Directive] = HeliumDirectives.all
    val linkDirectives: Seq[Links.Directive] = Nil
  }
  
  
  def build[F[_]: Async]: Resource[F, Theme[F]] = {

    import helium._

    val treeProcessor = new HeliumTreeProcessor[F](helium)

    ThemeBuilder("Helium")
      .addInputs(HeliumInputBuilder.build(helium))
      .addBaseConfig(ConfigGenerator.populateConfig(helium))
      .addRewriteRules(HeliumRewriteRules.build(helium))
      .addRenderOverrides(HTML.Overrides(HeliumRenderOverrides.forHTML(siteSettings.layout.anchorPlacement)))
      .addRenderOverrides(EPUB.XHTML.Overrides(HeliumRenderOverrides.forEPUB))
      .addRenderOverrides(XSLFO.Overrides(HeliumRenderOverrides.forPDF))
      .addExtensions(directives)
      .processTree(treeProcessor.forHTML, HTML)
      .processTree(treeProcessor.forAllFormats)
      .build

  }

}