lean4-htt/Lake/Config/Workspace.lean

81 lines
2.5 KiB
Text

/-
Copyright (c) 2021 Mac Malone. All rights reserved.
Released under Apache 2.0 license as described in the file LICENSE.
Authors: Mac Malone
-/
import Lean.Util.Paths
import Lake.Config.Opaque
import Lake.Config.WorkspaceConfig
import Lake.Config.Package
open System
open Lean (Name NameMap LeanPaths)
namespace Lake
/-- A Lake workspace -- the top-level package directory. -/
structure Workspace where
/-- The path to the workspace's directory. -/
dir : FilePath
/-- The workspace's configuration. -/
config : WorkspaceConfig
/-- Name-package map of packages within the workspace. -/
packageMap : NameMap Package := {}
deriving Inhabited
namespace OpaqueWorkspace
unsafe def unsafeMk (ws : Workspace) : OpaqueWorkspace :=
unsafeCast ws
@[implementedBy unsafeMk] constant mk (ws : Workspace) : OpaqueWorkspace :=
OpaqueWorkspacePointed.val
instance : Coe Workspace OpaqueWorkspace := ⟨mk⟩
instance : Inhabited OpaqueWorkspace := ⟨mk Inhabited.default⟩
unsafe def unsafeGet (self : OpaqueWorkspace) : Workspace :=
unsafeCast self
@[implementedBy unsafeGet] constant get (self : OpaqueWorkspace) : Workspace
instance : Coe OpaqueWorkspace Workspace := ⟨get⟩
@[simp] axiom get_mk {ws : Workspace} : get (mk ws) = ws
end OpaqueWorkspace
namespace Workspace
/-- Create a `Workspace` from a package using its directory and `WorkspaceConfig`. -/
def ofPackage (pkg : Package) : Workspace :=
{dir := pkg.dir, config := pkg.config.toWorkspaceConfig}
/-- The workspace's `dir` joined with its `packagesDir` configuration. -/
def packagesDir (self : Workspace) : FilePath :=
self.dir / self.config.packagesDir
/-- Add a package to the workspace. -/
def addPackage (pkg : Package) (self : Workspace) : Workspace :=
{self with packageMap := self.packageMap.insert pkg.name pkg}
/-- Get a package within the workspace by name. -/
def getPackage? (pkg : Name) (self : Workspace) : Option Package :=
self.packageMap.find? pkg
/-- The `.olean` directories of the workspace. -/
def oleanDirs (self : Workspace) : List FilePath :=
self.packageMap.toList.map (·.2.oleanDir)
/-- The `LEAN_PATH` of the workspace. -/
def oleanPath (self : Workspace) : SearchPath :=
self.packageMap.toList.map (·.2.oleanDir)
/-- The `LEAN_SRC_PATH` of the workspace. -/
def srcPath (self : Workspace) : SearchPath :=
self.packageMap.toList.map (·.2.srcDir)
/-- The `LeanPaths` of the workspace. -/
def leanPaths (self : Workspace) : LeanPaths :=
let pkgs := self.packageMap.toList
LeanPaths.mk (pkgs.map (·.2.oleanDir)) (pkgs.map (·.2.srcDir))