cola/app.py

   1 """Provides the main() routine and ColaApplication"""
   2 # pylint: disable=unused-import
   3 from __future__ import division, absolute_import, unicode_literals
   4 from functools import partial
   5 import argparse
   6 import os
   7 import signal
   8 import sys
   9 import time
  10
  11 __copyright__ = """
  12 Copyright (C) 2007-2017 David Aguilar and contributors
  13 """
  14
  15 try:
  16     from qtpy import QtCore
  17 except ImportError:
  18     sys.stderr.write(
  19         """
  20 You do not seem to have PyQt5, PySide, or PyQt4 installed.
  21 Please install it before using git-cola, e.g. on a Debian/Ubutnu system:
  22
  23     sudo apt-get install python-pyqt5 python-pyqt5.qtwebkit
  24
  25 """
  26     )
  27     sys.exit(1)
  28
  29 from qtpy import QtWidgets
  30 from qtpy.QtCore import Qt
  31
  32 try:
  33     # Qt 5.12 / PyQt 5.13 is unable to use QtWebEngineWidgets unless it is
  34     # imported before QApplication is constructed.
  35     from qtpy import QtWebEngineWidgets  # noqa
  36 except ImportError:
  37     # QtWebEngineWidgets / QtWebKit is not available -- no big deal.
  38     pass
  39
  40 # Import cola modules
  41 from .i18n import N_
  42 from .interaction import Interaction
  43 from .models import main
  44 from .models import selection
  45 from .widgets import cfgactions
  46 from .widgets import standard
  47 from .widgets import startup
  48 from .settings import Session
  49 from .settings import Settings
  50 from . import cmds
  51 from . import core
  52 from . import compat
  53 from . import fsmonitor
  54 from . import git
  55 from . import gitcfg
  56 from . import guicmds
  57 from . import hidpi
  58 from . import icons
  59 from . import i18n
  60 from . import qtcompat
  61 from . import qtutils
  62 from . import resources
  63 from . import themes
  64 from . import utils
  65 from . import version
  66
  67
  68 def setup_environment():
  69     """Set environment variables to control git's behavior"""
  70     # Allow Ctrl-C to exit
  71     signal.signal(signal.SIGINT, signal.SIG_DFL)
  72
  73     # Session management wants an absolute path when restarting
  74     sys.argv[0] = sys_argv0 = os.path.abspath(sys.argv[0])
  75
  76     # Spoof an X11 display for SSH
  77     os.environ.setdefault('DISPLAY', ':0')
  78
  79     if not core.getenv('SHELL', ''):
  80         for shell in ('/bin/zsh', '/bin/bash', '/bin/sh'):
  81             if os.path.exists(shell):
  82                 compat.setenv('SHELL', shell)
  83                 break
  84
  85     # Setup the path so that git finds us when we run 'git cola'
  86     path_entries = core.getenv('PATH', '').split(os.pathsep)
  87     bindir = core.decode(os.path.dirname(sys_argv0))
  88     path_entries.append(bindir)
  89     path = os.pathsep.join(path_entries)
  90     compat.setenv('PATH', path)
  91
  92     # We don't ever want a pager
  93     compat.setenv('GIT_PAGER', '')
  94
  95     # Setup *SSH_ASKPASS
  96     git_askpass = core.getenv('GIT_ASKPASS')
  97     ssh_askpass = core.getenv('SSH_ASKPASS')
  98     if git_askpass:
  99         askpass = git_askpass
 100     elif ssh_askpass:
 101         askpass = ssh_askpass
 102     elif sys.platform == 'darwin':
 103         askpass = resources.share('bin', 'ssh-askpass-darwin')
 104     else:
 105         askpass = resources.share('bin', 'ssh-askpass')
 106
 107     compat.setenv('GIT_ASKPASS', askpass)
 108     compat.setenv('SSH_ASKPASS', askpass)
 109
 110     # --- >8 --- >8 ---
 111     # Git v1.7.10 Release Notes
 112     # =========================
 113     #
 114     # Compatibility Notes
 115     # -------------------
 116     #
 117     #  * From this release on, the "git merge" command in an interactive
 118     #   session will start an editor when it automatically resolves the
 119     #   merge for the user to explain the resulting commit, just like the
 120     #   "git commit" command does when it wasn't given a commit message.
 121     #
 122     #   If you have a script that runs "git merge" and keeps its standard
 123     #   input and output attached to the user's terminal, and if you do not
 124     #   want the user to explain the resulting merge commits, you can
 125     #   export GIT_MERGE_AUTOEDIT environment variable set to "no", like
 126     #   this:
 127     #
 128     #        #!/bin/sh
 129     #        GIT_MERGE_AUTOEDIT=no
 130     #        export GIT_MERGE_AUTOEDIT
 131     #
 132     #   to disable this behavior (if you want your users to explain their
 133     #   merge commits, you do not have to do anything).  Alternatively, you
 134     #   can give the "--no-edit" option to individual invocations of the
 135     #   "git merge" command if you know everybody who uses your script has
 136     #   Git v1.7.8 or newer.
 137     # --- >8 --- >8 ---
 138     # Longer-term: Use `git merge --no-commit` so that we always
 139     # have a chance to explain our merges.
 140     compat.setenv('GIT_MERGE_AUTOEDIT', 'no')
 141
 142     # Gnome3 on Debian has XDG_SESSION_TYPE=wayland and
 143     # XDG_CURRENT_DESKTOP=GNOME, which Qt warns about at startup:
 144     #
 145     #   Warning: Ignoring XDG_SESSION_TYPE=wayland on Gnome.
 146     #   Use QT_QPA_PLATFORM=wayland to run on Wayland anyway.
 147     #
 148     # This annoying, so we silence the warning.
 149     # We'll need to keep this hack here until a future version of Qt provides
 150     # Qt Wayland widgets that are usable in gnome-shell.
 151     # Cf. https://bugreports.qt.io/browse/QTBUG-68619
 152     if (
 153         core.getenv('XDG_CURRENT_DESKTOP', '') == 'GNOME'
 154         and core.getenv('XDG_SESSION_TYPE', '') == 'wayland'
 155     ):
 156         compat.unsetenv('XDG_SESSION_TYPE')
 157
 158
 159 def get_icon_themes(context):
 160     """Return the default icon theme names"""
 161     result = []
 162
 163     icon_themes_env = core.getenv('GIT_COLA_ICON_THEME')
 164     if icon_themes_env:
 165         result.extend([x for x in icon_themes_env.split(':') if x])
 166
 167     icon_themes_cfg = context.cfg.get_all('cola.icontheme')
 168     if icon_themes_cfg:
 169         result.extend(icon_themes_cfg)
 170
 171     if not result:
 172         result.append('light')
 173
 174     return result
 175
 176
 177 # style note: we use camelCase here since we're masquerading a Qt class
 178 class ColaApplication(object):
 179     """The main cola application
 180
 181     ColaApplication handles i18n of user-visible data
 182     """
 183
 184     def __init__(self, context, argv, locale=None, icon_themes=None, gui_theme=None):
 185         cfgactions.install()
 186         i18n.install(locale)
 187         qtcompat.install()
 188         guicmds.install()
 189         standard.install()
 190         icons.install(icon_themes or get_icon_themes(context))
 191
 192         self.context = context
 193         self._install_hidpi_config()
 194         self._app = ColaQApplication(context, list(argv))
 195         self._app.setWindowIcon(icons.cola())
 196         self._install_style(gui_theme)
 197
 198     def _install_style(self, theme_str):
 199         """Generate and apply a stylesheet to the app"""
 200         if theme_str is None:
 201             theme_str = self.context.cfg.get('cola.theme', default='default')
 202         theme = themes.find_theme(theme_str)
 203         self._app.setStyleSheet(theme.build_style_sheet(self._app.palette()))
 204         if theme_str != 'default':
 205             self._app.setPalette(theme.build_palette(self._app.palette()))
 206
 207     def _install_hidpi_config(self):
 208         """Sets QT HIDPI scalling (requires Qt 5.6)"""
 209         value = self.context.cfg.get('cola.hidpi', default=hidpi.Option.AUTO)
 210         hidpi.apply_choice(value)
 211
 212     def activeWindow(self):
 213         """QApplication::activeWindow() pass-through"""
 214         return self._app.activeWindow()
 215
 216     def desktop(self):
 217         """QApplication::desktop() pass-through"""
 218         return self._app.desktop()
 219
 220     def start(self):
 221         """Wrap exec_() and start the application"""
 222         # Defer connection so that local cola.inotify is honored
 223         context = self.context
 224         monitor = context.fsmonitor
 225         monitor.files_changed.connect(
 226             cmds.run(cmds.Refresh, context), type=Qt.QueuedConnection
 227         )
 228         monitor.config_changed.connect(
 229             cmds.run(cmds.RefreshConfig, context), type=Qt.QueuedConnection
 230         )
 231         # Start the filesystem monitor thread
 232         monitor.start()
 233         return self._app.exec_()
 234
 235     def stop(self):
 236         """Finalize the application"""
 237         self.context.fsmonitor.stop()
 238         # Workaround QTBUG-52988 by deleting the app manually to prevent a
 239         # crash during app shutdown.
 240         # https://bugreports.qt.io/browse/QTBUG-52988
 241         try:
 242             del self._app
 243         except (AttributeError, RuntimeError):
 244             pass
 245         self._app = None
 246
 247     def exit(self, status):
 248         """QApplication::exit(status) pass-through"""
 249         return self._app.exit(status)
 250
 251
 252 class ColaQApplication(QtWidgets.QApplication):
 253     """QApplication implementation for handling custom events"""
 254
 255     def __init__(self, context, argv):
 256         super(ColaQApplication, self).__init__(argv)
 257         self.context = context
 258         # Make icons sharp in HiDPI screen
 259         if hasattr(Qt, 'AA_UseHighDpiPixmaps'):
 260             self.setAttribute(Qt.AA_UseHighDpiPixmaps, True)
 261
 262     def event(self, e):
 263         """Respond to focus events for the cola.refreshonfocus feature"""
 264         if e.type() == QtCore.QEvent.ApplicationActivate:
 265             context = self.context
 266             if context:
 267                 cfg = context.cfg
 268                 if context.git.is_valid() and cfg.get(
 269                     'cola.refreshonfocus', default=False
 270                 ):
 271                     cmds.do(cmds.Refresh, context)
 272         return super(ColaQApplication, self).event(e)
 273
 274     def commitData(self, session_mgr):
 275         """Save session data"""
 276         if not self.context or not self.context.view:
 277             return
 278         view = self.context.view
 279         if not hasattr(view, 'save_state'):
 280             return
 281         sid = session_mgr.sessionId()
 282         skey = session_mgr.sessionKey()
 283         session_id = '%s_%s' % (sid, skey)
 284         session = Session(session_id, repo=core.getcwd())
 285         session.update()
 286         view.save_state(settings=session)
 287
 288
 289 def process_args(args):
 290     """Process and verify command-line arguments"""
 291     if args.version:
 292         # Accept 'git cola --version' or 'git cola version'
 293         version.print_version()
 294         sys.exit(core.EXIT_SUCCESS)
 295
 296     # Handle session management
 297     restore_session(args)
 298
 299     # Bail out if --repo is not a directory
 300     repo = core.decode(args.repo)
 301     if repo.startswith('file:'):
 302         repo = repo[len('file:') :]
 303     repo = core.realpath(repo)
 304     if not core.isdir(repo):
 305         errmsg = (
 306             N_(
 307                 'fatal: "%s" is not a directory.  '
 308                 'Please specify a correct --repo <path>.'
 309             )
 310             % repo
 311         )
 312         core.print_stderr(errmsg)
 313         sys.exit(core.EXIT_USAGE)
 314
 315
 316 def restore_session(args):
 317     """Load a session based on the window-manager provided arguments"""
 318     # args.settings is provided when restoring from a session.
 319     args.settings = None
 320     if args.session is None:
 321         return
 322     session = Session(args.session)
 323     if session.load():
 324         args.settings = session
 325         args.repo = session.repo
 326
 327
 328 def application_init(args, update=False):
 329     """Parses the command-line arguments and starts git-cola
 330     """
 331     # Ensure that we're working in a valid git repository.
 332     # If not, try to find one.  When found, chdir there.
 333     setup_environment()
 334     process_args(args)
 335
 336     context = new_context(args)
 337     timer = context.timer
 338     timer.start('init')
 339
 340     new_worktree(context, args.repo, args.prompt)
 341
 342     if update:
 343         context.model.update_status()
 344
 345     timer.stop('init')
 346     if args.perf:
 347         timer.display('init')
 348     return context
 349
 350
 351 def new_context(args):
 352     """Create top-level ApplicationContext objects"""
 353     context = ApplicationContext(args)
 354     context.settings = args.settings or Settings.read()
 355     context.git = git.create()
 356     context.cfg = gitcfg.create(context)
 357     context.fsmonitor = fsmonitor.create(context)
 358     context.selection = selection.create()
 359     context.model = main.create(context)
 360     context.app = new_application(context, args)
 361     context.timer = Timer()
 362
 363     return context
 364
 365
 366 def application_run(context, view, start=None, stop=None):
 367     """Run the application main loop"""
 368     initialize_view(context, view)
 369     # Startup callbacks
 370     if start:
 371         start(context, view)
 372     # Start the event loop
 373     result = context.app.start()
 374     # Finish
 375     if stop:
 376         stop(context, view)
 377     context.app.stop()
 378
 379     return result
 380
 381
 382 def initialize_view(context, view):
 383     """Register the main widget and display it"""
 384     context.set_view(view)
 385     view.show()
 386     if sys.platform == 'darwin':
 387         view.raise_()
 388
 389
 390 def application_start(context, view):
 391     """Show the GUI and start the main event loop"""
 392     # Store the view for session management
 393     return application_run(context, view, start=default_start, stop=default_stop)
 394
 395
 396 def default_start(context, _view):
 397     """Scan for the first time"""
 398     QtCore.QTimer.singleShot(0, startup_message)
 399     QtCore.QTimer.singleShot(0, lambda: async_update(context))
 400
 401
 402 def default_stop(_context, _view):
 403     """All done, cleanup"""
 404     QtCore.QThreadPool.globalInstance().waitForDone()
 405
 406
 407 def add_common_arguments(parser):
 408     """Add command arguments to the ArgumentParser"""
 409     # We also accept 'git cola version'
 410     parser.add_argument(
 411         '--version', default=False, action='store_true', help='print version number'
 412     )
 413
 414     # Specifies a git repository to open
 415     parser.add_argument(
 416         '-r',
 417         '--repo',
 418         metavar='<repo>',
 419         default=core.getcwd(),
 420         help='open the specified git repository',
 421     )
 422
 423     # Specifies that we should prompt for a repository at startup
 424     parser.add_argument(
 425         '--prompt', action='store_true', default=False, help='prompt for a repository'
 426     )
 427
 428     # Specify the icon theme
 429     parser.add_argument(
 430         '--icon-theme',
 431         metavar='<theme>',
 432         dest='icon_themes',
 433         action='append',
 434         default=[],
 435         help='specify an icon theme (name or directory)',
 436     )
 437
 438     # Resume an X Session Management session
 439     parser.add_argument(
 440         '-session', metavar='<session>', default=None, help=argparse.SUPPRESS
 441     )
 442
 443     # Enable timing information
 444     parser.add_argument(
 445         '--perf', action='store_true', default=False, help=argparse.SUPPRESS
 446     )
 447
 448     # Specify the GUI theme
 449     parser.add_argument(
 450         '--theme', metavar='<name>', default=None, help='specify an GUI theme name'
 451     )
 452
 453
 454 def new_application(context, args):
 455     """Create a new ColaApplication"""
 456     return ColaApplication(
 457         context, sys.argv, icon_themes=args.icon_themes, gui_theme=args.theme
 458     )
 459
 460
 461 def new_worktree(context, repo, prompt):
 462     """Find a Git repository, or prompt for one when not found"""
 463     model = context.model
 464     cfg = context.cfg
 465     parent = qtutils.active_window()
 466     valid = False
 467
 468     if not prompt:
 469         valid = model.set_worktree(repo)
 470         if not valid:
 471             # We are not currently in a git repository so we need to find one.
 472             # Before prompting the user for a repository, check if they've
 473             # configured a default repository and attempt to use it.
 474             default_repo = cfg.get('cola.defaultrepo')
 475             if default_repo:
 476                 valid = model.set_worktree(default_repo)
 477
 478     while not valid:
 479         # If we've gotten into this loop then that means that neither the
 480         # current directory nor the default repository were available.
 481         # Prompt the user for a repository.
 482         startup_dlg = startup.StartupDialog(context, parent)
 483         gitdir = startup_dlg.find_git_repo()
 484         if not gitdir:
 485             sys.exit(core.EXIT_NOINPUT)
 486
 487         if not core.exists(os.path.join(gitdir, '.git')):
 488             offer_to_create_repo(context, gitdir)
 489             valid = model.set_worktree(gitdir)
 490             continue
 491
 492         valid = model.set_worktree(gitdir)
 493         if not valid:
 494             standard.critical(N_('Error Opening Repository'),
 495                               N_('Could not open %s.' % gitdir))
 496
 497
 498 def offer_to_create_repo(context, gitdir):
 499     """Offer to create a new repo"""
 500     title = N_('Repository Not Found')
 501     text = N_('%s is not a Git repository.') % gitdir
 502     informative_text = N_('Create a new repository at that location?')
 503     if standard.confirm(title, text, informative_text, N_('Create')):
 504         status, out, err = context.git.init(gitdir)
 505         title = N_('Error Creating Repository')
 506         if status != 0:
 507             Interaction.command_error(title, 'git init', status, out, err)
 508
 509
 510 def async_update(context):
 511     """Update the model in the background
 512
 513     git-cola should startup as quickly as possible.
 514
 515     """
 516     update_status = partial(context.model.update_status, update_index=True)
 517     task = qtutils.SimpleTask(context.view, update_status)
 518     context.runtask.start(task)
 519
 520
 521 def startup_message():
 522     """Print debug startup messages"""
 523     trace = git.GIT_COLA_TRACE
 524     if trace in ('2', 'trace'):
 525         msg1 = 'info: debug level 2: trace mode enabled'
 526         msg2 = 'info: set GIT_COLA_TRACE=1 for less-verbose output'
 527         Interaction.log(msg1)
 528         Interaction.log(msg2)
 529     elif trace:
 530         msg1 = 'info: debug level 1'
 531         msg2 = 'info: set GIT_COLA_TRACE=2 for trace mode'
 532         Interaction.log(msg1)
 533         Interaction.log(msg2)
 534
 535
 536 def initialize():
 537     """System-level initialization"""
 538     # The current directory may have been deleted while we are still
 539     # in that directory.  We rectify this situation by walking up the
 540     # directory tree and retrying.
 541     #
 542     # This is needed because  because Python throws exceptions in lots of
 543     # stdlib functions when in this situation, e.g. os.path.abspath() and
 544     # os.path.realpath(), so it's simpler to mitigate the damage by changing
 545     # the current directory to one that actually exists.
 546     while True:
 547         try:
 548             return core.getcwd()
 549         except OSError:
 550             os.chdir('..')
 551
 552
 553 class Timer(object):
 554     """Simple performance timer"""
 555
 556     def __init__(self):
 557         self._data = {}
 558
 559     def start(self, key):
 560         """Start a timer"""
 561         now = time.time()
 562         self._data[key] = [now, now]
 563
 564     def stop(self, key):
 565         """Stop a timer and return its elapsed time"""
 566         entry = self._data[key]
 567         entry[1] = time.time()
 568         return self.elapsed(key)
 569
 570     def elapsed(self, key):
 571         """Return the elapsed time for a timer"""
 572         entry = self._data[key]
 573         return entry[1] - entry[0]
 574
 575     def display(self, key):
 576         """Display a timer"""
 577         elapsed = self.elapsed(key)
 578         sys.stdout.write('%s: %.5fs\n' % (key, elapsed))
 579
 580
 581 class ApplicationContext(object):
 582     """Context for performing operations on Git and related data models"""
 583
 584     def __init__(self, args):
 585         self.args = args
 586         self.app = None  # ColaApplication
 587         self.git = None  # git.Git
 588         self.cfg = None  # gitcfg.GitConfig
 589         self.model = None  # main.MainModel
 590         self.timer = None  # Timer
 591         self.runtask = None  # qtutils.RunTask
 592         self.settings = None  # settings.Settings
 593         self.selection = None  # selection.SelectionModel
 594         self.fsmonitor = None  # fsmonitor
 595         self.view = None  # QWidget
 596
 597     def set_view(self, view):
 598         """Initialize view-specific members"""
 599         self.view = view
 600         self.runtask = qtutils.RunTask(parent=view)
 601
 602
 603 def winmain(main_fn, *argv):
 604     """Find Git and launch main(argv)"""
 605     git_path = find_git()
 606     if git_path:
 607         prepend_path(git_path)
 608     return main_fn(*argv)
 609
 610
 611 def find_git():
 612     """Return the path of git.exe, or None if we can't find it."""
 613     if not utils.is_win32():
 614         return None  # UNIX systems have git in their $PATH
 615
 616     # If the user wants to use a Git/bin/ directory from a non-standard
 617     # directory then they can write its location into
 618     # ~/.config/git-cola/git-bindir
 619     git_bindir = os.path.expanduser(
 620         os.path.join('~', '.config', 'git-cola', 'git-bindir')
 621     )
 622     if core.exists(git_bindir):
 623         custom_path = core.read(git_bindir).strip()
 624         if custom_path and core.exists(custom_path):
 625             return custom_path
 626
 627     # Try to find Git's bin/ directory in one of the typical locations
 628     pf = os.environ.get('ProgramFiles', 'C:\\Program Files')
 629     pf32 = os.environ.get('ProgramFiles(x86)', 'C:\\Program Files (x86)')
 630     pf64 = os.environ.get('ProgramW6432', 'C:\\Program Files')
 631     for p in [pf64, pf32, pf, 'C:\\']:
 632         candidate = os.path.join(p, 'Git\\bin')
 633         if os.path.isdir(candidate):
 634             return candidate
 635
 636     return None
 637
 638
 639 def prepend_path(path):
 640     """Adds git to the PATH.  This is needed on Windows."""
 641     path = core.decode(path)
 642     path_entries = core.getenv('PATH', '').split(os.pathsep)
 643     if path not in path_entries:
 644         path_entries.insert(0, path)
 645         compat.setenv('PATH', os.pathsep.join(path_entries))