cola/icons.py

   1 """The only file where icon filenames are mentioned"""
   2 from __future__ import absolute_import, division, unicode_literals
   3 import os
   4
   5 from qtpy import QtGui
   6 from qtpy import QtWidgets
   7
   8 from . import core
   9 from . import qtcompat
  10 from . import resources
  11 from .compat import ustr
  12 from .i18n import N_
  13
  14
  15 KNOWN_FILE_MIME_TYPES = [
  16     ('text', 'file-code.svg'),
  17     ('image', 'file-media.svg'),
  18     ('octet', 'file-binary.svg'),
  19 ]
  20
  21 KNOWN_FILE_EXTENSIONS = {
  22     '.bash': 'file-code.svg',
  23     '.c': 'file-code.svg',
  24     '.cpp': 'file-code.svg',
  25     '.css': 'file-code.svg',
  26     '.cxx': 'file-code.svg',
  27     '.h': 'file-code.svg',
  28     '.hpp': 'file-code.svg',
  29     '.hs': 'file-code.svg',
  30     '.html': 'file-code.svg',
  31     '.java': 'file-code.svg',
  32     '.js': 'file-code.svg',
  33     '.ksh': 'file-code.svg',
  34     '.lisp': 'file-code.svg',
  35     '.perl': 'file-code.svg',
  36     '.pl': 'file-code.svg',
  37     '.py': 'file-code.svg',
  38     '.rb': 'file-code.svg',
  39     '.rs': 'file-code.svg',
  40     '.sh': 'file-code.svg',
  41     '.zsh': 'file-code.svg',
  42 }
  43
  44
  45 def install(themes):
  46     for theme in themes:
  47         icon_dir = resources.icon_dir(theme)
  48         qtcompat.add_search_path('icons', icon_dir)
  49
  50
  51 def icon_themes():
  52     return (
  53         (N_('Default'), 'default'),
  54         (N_('Dark Theme'), 'dark'),
  55         (N_('Light Theme'), 'light'),
  56     )
  57
  58
  59 def name_from_basename(basename):
  60     """Prefix the basename with "icons:" so that git-cola's icons are found
  61
  62     "icons" is registered with Qt's resource system during install().
  63
  64     """
  65     return 'icons:' + basename
  66
  67
  68 def from_name(name):
  69     """Return a QIcon from an absolute filename or "icons:basename.svg" name"""
  70     return QtGui.QIcon(name)
  71
  72
  73 def icon(basename):
  74     """Given a basename returns a QIcon from the corresponding cola icon"""
  75     return from_name(name_from_basename(basename))
  76
  77
  78 def from_theme(name, fallback=None):
  79     """Grab an icon from the current theme with a fallback
  80
  81     Support older versions of Qt checking for fromTheme's availability.
  82
  83     """
  84     if hasattr(QtGui.QIcon, 'fromTheme'):
  85         base, _ = os.path.splitext(name)
  86         if fallback:
  87             qicon = QtGui.QIcon.fromTheme(base, icon(fallback))
  88         else:
  89             qicon = QtGui.QIcon.fromTheme(base)
  90         if not qicon.isNull():
  91             return qicon
  92     return icon(fallback or name)
  93
  94
  95 def basename_from_filename(filename):
  96     """Returns an icon name based on the filename"""
  97     mimetype = core.guess_mimetype(filename)
  98     if mimetype is not None:
  99         mimetype = mimetype.lower()
 100         for filetype, icon_name in KNOWN_FILE_MIME_TYPES:
 101             if filetype in mimetype:
 102                 return icon_name
 103     extension = os.path.splitext(filename)[1]
 104     return KNOWN_FILE_EXTENSIONS.get(extension.lower(), 'file-text.svg')
 105
 106
 107 def from_filename(filename):
 108     basename = basename_from_filename(filename)
 109     return from_name(name_from_basename(basename))
 110
 111
 112 def mkicon(value, default=None):
 113     if value is None and default is not None:
 114         value = default()
 115     elif value and isinstance(value, (str, ustr)):
 116         value = QtGui.QIcon(value)
 117     return value
 118
 119
 120 def from_style(key):
 121     """Maintain a cache of standard icons and return cache entries."""
 122     style = QtWidgets.QApplication.instance().style()
 123     return style.standardIcon(key)
 124
 125
 126 def status(filename, deleted, is_staged, untracked):
 127     if deleted:
 128         icon_name = 'circle-slash-red.svg'
 129     elif is_staged:
 130         icon_name = 'staged.svg'
 131     elif untracked:
 132         icon_name = 'question-plain.svg'
 133     else:
 134         icon_name = basename_from_filename(filename)
 135     return icon_name
 136
 137
 138 # Icons creators and SVG file references
 139
 140
 141 def three_bars():
 142     return icon('three-bars.svg')
 143
 144
 145 def add():
 146     return from_theme('list-add', fallback='plus.svg')
 147
 148
 149 def alphabetical():
 150     return icon('a-z-order.svg')
 151
 152
 153 def branch():
 154     return icon('git-branch.svg')
 155
 156
 157 def check_name():
 158     return name_from_basename('check.svg')
 159
 160
 161 def close():
 162     return icon('x.svg')
 163
 164
 165 def cola():
 166     return icon('git-cola.svg')
 167
 168
 169 def commit():
 170     return icon('document-save-symbolic.svg')
 171
 172
 173 def compare():
 174     return icon('git-compare.svg')
 175
 176
 177 def configure():
 178     return icon('gear.svg')
 179
 180
 181 def copy():
 182     return from_theme('edit-copy.svg')
 183
 184
 185 def default_app():
 186     return icon('telescope.svg')
 187
 188
 189 def dot_name():
 190     return name_from_basename('primitive-dot.svg')
 191
 192
 193 def download():
 194     return icon('file-download.svg')
 195
 196
 197 def discard():
 198     return icon('trashcan.svg')
 199
 200
 201 # folder vs directory: directory is opaque, folder is just an outline
 202 # directory is used for the File Browser, where more contrast with the file
 203 # icons are needed.
 204
 205
 206 def folder():
 207     return icon('folder.svg')
 208
 209
 210 def directory():
 211     return icon('file-directory.svg')
 212
 213
 214 def diff():
 215     return icon('diff.svg')
 216
 217
 218 def edit():
 219     return icon('pencil.svg')
 220
 221
 222 def ellipsis():
 223     return icon('ellipsis.svg')
 224
 225
 226 def external():
 227     return icon('link-external.svg')
 228
 229
 230 def file_code():
 231     return icon('file-code.svg')
 232
 233
 234 def file_text():
 235     return icon('file-text.svg')
 236
 237
 238 def file_zip():
 239     return icon('file-zip.svg')
 240
 241
 242 def fold():
 243     return icon('fold.svg')
 244
 245
 246 def merge():
 247     return icon('git-merge.svg')
 248
 249
 250 def modified():
 251     return icon('modified.svg')
 252
 253
 254 def modified_name():
 255     return name_from_basename('modified.svg')
 256
 257
 258 def new():
 259     return icon('folder-new.svg')
 260
 261
 262 def ok():
 263     return icon('check.svg')
 264
 265
 266 def open_directory():
 267     return icon('folder.svg')
 268
 269
 270 def partial_name():
 271     return name_from_basename('partial.svg')
 272
 273
 274 def pull():
 275     return icon('repo-pull.svg')
 276
 277
 278 def push():
 279     return icon('repo-push.svg')
 280
 281
 282 def question():
 283     return icon('question.svg')
 284
 285
 286 def remove():
 287     return from_theme('list-remove', fallback='circle-slash.svg')
 288
 289
 290 def repo():
 291     return icon('repo.svg')
 292
 293
 294 def reverse_chronological():
 295     return icon('last-first-order.svg')
 296
 297
 298 def save():
 299     return icon('desktop-download.svg')
 300
 301
 302 def search():
 303     return icon('search.svg')
 304
 305
 306 def select_all():
 307     return from_theme('edit-select-all.svg')
 308
 309
 310 def staged():
 311     return icon('staged.svg')
 312
 313
 314 def staged_name():
 315     return name_from_basename('staged.svg')
 316
 317
 318 def star():
 319     return icon('star.svg')
 320
 321
 322 def sync():
 323     return icon('sync.svg')
 324
 325
 326 def tag():
 327     return icon('tag.svg')
 328
 329
 330 def undo():
 331     return from_theme('edit-undo', fallback='edit-undo.svg')
 332
 333
 334 def style_dialog_apply():
 335     return from_style(QtWidgets.QStyle.SP_DialogApplyButton)
 336
 337
 338 def style_dialog_discard():
 339     return from_style(QtWidgets.QStyle.SP_DialogDiscardButton)
 340
 341
 342 def style_dialog_reset():
 343     return from_style(QtWidgets.QStyle.SP_DialogResetButton)
 344
 345
 346 def unfold():
 347     return icon('unfold.svg')
 348
 349
 350 def visualize():
 351     return icon('eye.svg')
 352
 353
 354 def upstream_name():
 355     return name_from_basename('upstream.svg')
 356
 357
 358 def zoom_fit_best():
 359     return icon('zoom-fit-best.svg')
 360
 361
 362 def zoom_in():
 363     return icon('zoom-in.svg')
 364
 365
 366 def zoom_out():
 367     return icon('zoom-out.svg')