src/Pure/Tools/imports.scala
author wenzelm
Wed Jun 07 19:41:05 2017 +0200 (2017-06-07)
changeset 66029 063f13f10433
parent 66028 14c014a43278
child 66030 f551ab947569
permissions -rw-r--r--
proper imports_base;
     1 /*  Title:      Pure/Tools/imports.scala
     2     Author:     Makarius
     3 
     4 Maintain theory imports wrt. session structure.
     5 */
     6 
     7 package isabelle
     8 
     9 
    10 import java.io.{File => JFile}
    11 
    12 
    13 object Imports
    14 {
    15   /* repository files */
    16 
    17   def repository_files(progress: Progress, start: Path, pred: JFile => Boolean = _ => true)
    18       : List[JFile] =
    19     Mercurial.find_repository(start) match {
    20       case None =>
    21         progress.echo_warning("Ignoring directory " + start + " (no Mercurial repository)")
    22         Nil
    23       case Some(hg) =>
    24         val start_path = start.canonical_file.toPath
    25         for {
    26           name <- hg.known_files()
    27           file = (hg.root + Path.explode(name)).file
    28           if pred(file) && file.getCanonicalFile.toPath.startsWith(start_path)
    29         } yield file
    30     }
    31 
    32 
    33   /* update imports */
    34 
    35   sealed case class Update(range: Text.Range, old_text: String, new_text: String)
    36   {
    37     def edits: List[Text.Edit] =
    38       Text.Edit.replace(range.start, old_text, new_text)
    39 
    40     override def toString: String =
    41       range.toString + ": " + old_text + " -> " + new_text
    42   }
    43 
    44   def update_name(keywords: Keyword.Keywords, pos: Position.T, update: String => String)
    45     : Option[(JFile, Update)] =
    46   {
    47     val file =
    48       pos match {
    49         case Position.File(file) => Path.explode(file).canonical_file
    50         case _ => error("Missing file in position" + Position.here(pos))
    51       }
    52 
    53     val text = File.read(file)
    54 
    55     val range =
    56       pos match {
    57         case Position.Range(symbol_range) => Symbol.Text_Chunk(text).decode(symbol_range)
    58         case _ => error("Missing range in position" + Position.here(pos))
    59       }
    60 
    61     Token.read_name(keywords, range.substring(text)) match {
    62       case Some(tok) =>
    63         val s1 = tok.source
    64         val s2 = Token.quote_name(keywords, update(tok.content))
    65         if (s1 == s2) None else Some((file, Update(range, s1, s2)))
    66       case None => error("Name token expected" + Position.here(pos))
    67     }
    68   }
    69 
    70 
    71   /* collective operations */
    72 
    73   def imports(
    74     options: Options,
    75     operation_imports: Boolean = false,
    76     operation_repository_files: Boolean = false,
    77     operation_update: Boolean = false,
    78     progress: Progress = No_Progress,
    79     selection: Sessions.Selection = Sessions.Selection.empty,
    80     dirs: List[Path] = Nil,
    81     select_dirs: List[Path] = Nil,
    82     verbose: Boolean = false) =
    83   {
    84     val full_sessions = Sessions.load(options, dirs, select_dirs)
    85     val (selected, selected_sessions) = full_sessions.selection(selection)
    86 
    87     val deps =
    88       Sessions.deps(selected_sessions, progress = progress, verbose = verbose,
    89         global_theories = full_sessions.global_theories,
    90         all_known = true)
    91 
    92     val root_keywords = Sessions.root_syntax.keywords
    93 
    94 
    95     if (operation_imports) {
    96       progress.echo("\nPotential session imports:")
    97       selected.sorted.foreach(session_name =>
    98       {
    99         val info = full_sessions(session_name)
   100         val session_resources = new Resources(deps(session_name))
   101 
   102         val declared_imports =
   103           full_sessions.imports_ancestors(session_name).toSet + session_name
   104         val extra_imports =
   105           (for {
   106             (_, a) <- session_resources.session_base.known.theories.iterator
   107             if session_resources.theory_qualifier(a) == info.theory_qualifier
   108             b <- deps.all_known.get_file(Path.explode(a.node).file)
   109             qualifier = session_resources.theory_qualifier(b)
   110             if !declared_imports.contains(qualifier)
   111           } yield qualifier).toSet
   112 
   113         if (extra_imports.nonEmpty) {
   114           progress.echo("session " + Token.quote_name(root_keywords, session_name) + ": " +
   115             extra_imports.toList.sorted.map(Token.quote_name(root_keywords, _)).mkString(" "))
   116         }
   117       })
   118     }
   119 
   120     if (operation_repository_files) {
   121       progress.echo("\nMercurial files check:")
   122       val unused_files =
   123         for {
   124           (_, dir) <- Sessions.directories(dirs, select_dirs)
   125           file <- repository_files(progress, dir, file => file.getName.endsWith(".thy"))
   126           if deps.all_known.get_file(file).isEmpty
   127         } yield file
   128       unused_files.foreach(file => progress.echo("unused file " + quote(file.toString)))
   129     }
   130 
   131     if (operation_update) {
   132       progress.echo("\nUpdate theory imports:")
   133       val updates =
   134         selected.flatMap(session_name =>
   135         {
   136           val info = full_sessions(session_name)
   137           val session_base = deps(session_name)
   138           val session_resources = new Resources(session_base)
   139           val imports_base = session_base.get_imports
   140           val imports_resources = new Resources(imports_base)
   141 
   142           def standard_import(qualifier: String, dir: String, s: String): String =
   143           {
   144             val name = imports_resources.import_name(qualifier, dir, s)
   145             val s1 =
   146               if (imports_base.loaded_theory(name)) name.theory
   147               else {
   148                 imports_base.known.get_file(Path.explode(name.node).file) match {
   149                   case Some(name1) if session_resources.theory_qualifier(name1) != qualifier =>
   150                     name1.theory
   151                   case Some(name1) if Thy_Header.is_base_name(s) =>
   152                     name1.theory_base_name
   153                   case _ => s
   154                 }
   155               }
   156             val name2 = imports_resources.import_name(qualifier, dir, s1)
   157             if (name.node == name2.node) s1 else s
   158           }
   159 
   160           val updates_root =
   161             for {
   162               (_, pos) <- info.theories.flatMap(_._2)
   163               upd <- update_name(root_keywords, pos,
   164                 standard_import(info.theory_qualifier, info.dir.implode, _))
   165             } yield upd
   166 
   167           val updates_theories =
   168             for {
   169               (_, name) <- session_base.known.theories_local.toList
   170               (_, pos) <- session_resources.check_thy(name, Token.Pos.file(name.node)).imports
   171               upd <- update_name(session_base.syntax.keywords, pos,
   172                 standard_import(session_resources.theory_qualifier(name), name.master_dir, _))
   173             } yield upd
   174 
   175           updates_root ::: updates_theories
   176         })
   177 
   178       val file_updates = (Multi_Map.empty[JFile, Update] /: updates)(_ + _)
   179       val conflicts =
   180         file_updates.iterator_list.flatMap({ case (file, upds) =>
   181           Library.duplicates(upds.sortBy(_.range.start),
   182             (x: Update, y: Update) => x.range overlaps y.range) match
   183           {
   184             case Nil => None
   185             case bad => Some((file, bad))
   186           }
   187         })
   188       if (conflicts.nonEmpty)
   189         error(cat_lines(
   190           conflicts.map({ case (file, bad) =>
   191             "Conflicting updates for file " + file + bad.mkString("\n  ", "\n  ", "") })))
   192 
   193       for ((file, upds) <- file_updates.iterator_list.toList.sortBy(p => p._1.toString)) {
   194         progress.echo("file " + quote(file.toString))
   195         val edits =
   196           upds.sortBy(upd => - upd.range.start).flatMap(upd =>
   197             Text.Edit.replace(upd.range.start, upd.old_text, upd.new_text))
   198         val new_text =
   199           (File.read(file) /: edits)({ case (text, edit) =>
   200             edit.edit(text, 0) match {
   201               case (None, text1) => text1
   202               case (Some(_), _) => error("Failed to apply edit " + edit + " to file " + file)
   203             }
   204           })
   205         File.write_backup2(Path.explode(File.standard_path(file)), new_text)
   206       }
   207     }
   208   }
   209 
   210 
   211   /* Isabelle tool wrapper */
   212 
   213   val isabelle_tool =
   214     Isabelle_Tool("imports", "maintain theory imports wrt. session structure", args =>
   215     {
   216       var select_dirs: List[Path] = Nil
   217       var operation_imports = false
   218       var operation_repository_files = false
   219       var requirements = false
   220       var operation_update = false
   221       var exclude_session_groups: List[String] = Nil
   222       var all_sessions = false
   223       var dirs: List[Path] = Nil
   224       var session_groups: List[String] = Nil
   225       var options = Options.init()
   226       var verbose = false
   227       var exclude_sessions: List[String] = Nil
   228 
   229       val getopts = Getopts("""
   230 Usage: isabelle imports [OPTIONS] [SESSIONS ...]
   231 
   232   Options are:
   233     -D DIR       include session directory and select its sessions
   234     -I           operation: report potential session imports
   235     -M           operation: Mercurial files check for imported theory files
   236     -R           operate on requirements of selected sessions
   237     -U           operation: update theory imports to use session qualifiers
   238     -X NAME      exclude sessions from group NAME and all descendants
   239     -a           select all sessions
   240     -d DIR       include session directory
   241     -g NAME      select session group NAME
   242     -o OPTION    override Isabelle system OPTION (via NAME=VAL or NAME)
   243     -v           verbose
   244     -x NAME      exclude session NAME and all descendants
   245 
   246   Maintain theory imports wrt. session structure. At least one operation
   247   needs to be specified (see options -I -M -U).
   248 """,
   249       "D:" -> (arg => select_dirs = select_dirs ::: List(Path.explode(arg))),
   250       "I" -> (_ => operation_imports = true),
   251       "M" -> (_ => operation_repository_files = true),
   252       "R" -> (_ => requirements = true),
   253       "U" -> (_ => operation_update = true),
   254       "X:" -> (arg => exclude_session_groups = exclude_session_groups ::: List(arg)),
   255       "a" -> (_ => all_sessions = true),
   256       "d:" -> (arg => dirs = dirs ::: List(Path.explode(arg))),
   257       "g:" -> (arg => session_groups = session_groups ::: List(arg)),
   258       "o:" -> (arg => options = options + arg),
   259       "v" -> (_ => verbose = true),
   260       "x:" -> (arg => exclude_sessions = exclude_sessions ::: List(arg)))
   261 
   262       val sessions = getopts(args)
   263       if (args.isEmpty || !(operation_imports || operation_repository_files || operation_update))
   264         getopts.usage()
   265 
   266       val selection =
   267         Sessions.Selection(requirements, all_sessions, exclude_session_groups,
   268           exclude_sessions, session_groups, sessions)
   269 
   270       val progress = new Console_Progress(verbose = verbose)
   271 
   272       imports(options, operation_imports = operation_imports,
   273         operation_repository_files = operation_repository_files, operation_update = operation_update,
   274         progress = progress, selection = selection, dirs = dirs, select_dirs = select_dirs,
   275         verbose = verbose)
   276     })
   277 }