cola/widgets/submodules.py

   1 """Provides widgets related to submodules"""
   2
   3 from qtpy import QtWidgets
   4 from qtpy.QtCore import Qt
   5 from qtpy.QtCore import Signal
   6
   7 from .. import cmds
   8 from .. import compat
   9 from .. import core
  10 from .. import qtutils
  11 from .. import icons
  12 from ..i18n import N_
  13 from ..widgets import defs
  14 from ..widgets import standard
  15 from ..widgets import text
  16
  17
  18 def add_submodule(context, parent):
  19     """Add a new submodule"""
  20     dlg = AddSubmodule(parent)
  21     dlg.show()
  22     if dlg.exec_() == standard.Dialog.Accepted:
  23         cmd = dlg.get(context)
  24         cmd.do()
  25
  26
  27 class SubmodulesWidget(QtWidgets.QFrame):
  28     def __init__(self, context, parent):
  29         super().__init__(parent)
  30         self.context = context
  31         self.setToolTip(N_('Submodules'))
  32
  33         self.tree = SubmodulesTreeWidget(context, parent=self)
  34         self.setFocusProxy(self.tree)
  35
  36         self.main_layout = qtutils.vbox(defs.no_margin, defs.spacing, self.tree)
  37         self.setLayout(self.main_layout)
  38
  39         # Titlebar buttons
  40         self.add_button = qtutils.create_action_button(
  41             tooltip=N_('Add Submodule'), icon=icons.add()
  42         )
  43         self.refresh_button = qtutils.create_action_button(
  44             tooltip=N_('Refresh'), icon=icons.sync()
  45         )
  46
  47         self.open_parent_button = qtutils.create_action_button(
  48             tooltip=N_('Open Parent'), icon=icons.repo()
  49         )
  50
  51         self.button_layout = qtutils.hbox(
  52             defs.no_margin,
  53             defs.spacing,
  54             self.add_button,
  55             self.open_parent_button,
  56             self.refresh_button,
  57         )
  58         self.corner_widget = QtWidgets.QWidget(self)
  59         self.corner_widget.setLayout(self.button_layout)
  60         titlebar = parent.titleBarWidget()
  61         titlebar.add_corner_widget(self.corner_widget)
  62
  63         # Connections
  64         qtutils.connect_button(self.add_button, self.add_submodule)
  65         qtutils.connect_button(self.refresh_button, self.tree.update_model.emit)
  66         qtutils.connect_button(
  67             self.open_parent_button, cmds.run(cmds.OpenParentRepo, context)
  68         )
  69
  70     def add_submodule(self):
  71         add_submodule(self.context, self)
  72
  73
  74 class AddSubmodule(standard.Dialog):
  75     """Add a new submodule"""
  76
  77     def __init__(self, parent):
  78         super().__init__(parent=parent)
  79
  80         hint = N_('git://git.example.com/repo.git')
  81         tooltip = N_('Submodule URL (can be relative, ex: ../repo.git)')
  82         self.url_text = text.HintedDefaultLineEdit(hint, tooltip=tooltip, parent=self)
  83
  84         hint = N_('path/to/submodule')
  85         tooltip = N_('Submodule path within the current repository (optional)')
  86         self.path_text = text.HintedDefaultLineEdit(hint, tooltip=tooltip, parent=self)
  87
  88         hint = N_('Branch name')
  89         tooltip = N_('Submodule branch to track (optional)')
  90         self.branch_text = text.HintedDefaultLineEdit(
  91             hint, tooltip=tooltip, parent=self
  92         )
  93
  94         self.depth_spinbox = standard.SpinBox(
  95             mini=0, maxi=compat.maxint, value=0, parent=self
  96         )
  97         self.depth_spinbox.setToolTip(
  98             N_(
  99                 'Create a shallow clone with history truncated to the '
 100                 'specified number of revisions.  0 performs a full clone.'
 101             )
 102         )
 103
 104         hint = N_('Reference URL')
 105         tooltip = N_('Reference repository to use when cloning (optional)')
 106         self.reference_text = text.HintedDefaultLineEdit(
 107             hint, tooltip=tooltip, parent=self
 108         )
 109
 110         self.add_button = qtutils.ok_button(N_('Add Submodule'), enabled=False)
 111         self.close_button = qtutils.close_button()
 112
 113         self.form_layout = qtutils.form(
 114             defs.no_margin,
 115             defs.button_spacing,
 116             (N_('URL'), self.url_text),
 117             (N_('Path'), self.path_text),
 118             (N_('Branch'), self.branch_text),
 119             (N_('Depth'), self.depth_spinbox),
 120             (N_('Reference Repository'), self.reference_text),
 121         )
 122         self.button_layout = qtutils.hbox(
 123             defs.no_margin,
 124             defs.button_spacing,
 125             qtutils.STRETCH,
 126             self.close_button,
 127             self.add_button,
 128         )
 129
 130         self.main_layout = qtutils.vbox(
 131             defs.large_margin, defs.spacing, self.form_layout, self.button_layout
 132         )
 133         self.setLayout(self.main_layout)
 134         self.init_size(parent=qtutils.active_window())
 135
 136         self.url_text.textChanged.connect(lambda x: self._update_widgets())
 137         qtutils.connect_button(self.add_button, self.accept)
 138         qtutils.connect_button(self.close_button, self.close)
 139
 140     def _update_widgets(self):
 141         value = self.url_text.value()
 142         self.add_button.setEnabled(bool(value))
 143
 144     def get(self, context):
 145         return cmds.SubmoduleAdd(
 146             context,
 147             self.url_text.value(),
 148             self.path_text.value(),
 149             self.branch_text.value(),
 150             self.depth_spinbox.value(),
 151             self.reference_text.value(),
 152         )
 153
 154
 155 class SubmodulesTreeWidget(standard.TreeWidget):
 156     update_model = Signal()
 157
 158     def __init__(self, context, parent=None):
 159         standard.TreeWidget.__init__(self, parent=parent)
 160
 161         model = context.model
 162         self.context = context
 163
 164         self.setSelectionMode(QtWidgets.QAbstractItemView.SingleSelection)
 165         self.setHeaderHidden(True)
 166         # UI
 167         self._active = False
 168         self.list_helper = BuildItem()
 169         # Connections
 170         self.itemDoubleClicked.connect(self.tree_double_clicked)
 171         model.submodules_changed.connect(self.refresh, type=Qt.QueuedConnection)
 172         self.update_model.connect(
 173             model.update_submodules_list, type=Qt.QueuedConnection
 174         )
 175
 176     def refresh(self):
 177         if not self._active:
 178             return
 179         submodules = self.context.model.submodules_list
 180         items = [self.list_helper.get(entry) for entry in submodules]
 181         self.clear()
 182         if items:
 183             self.addTopLevelItems(items)
 184
 185     def showEvent(self, event):
 186         """Defer updating widgets until the widget is visible"""
 187         if not self._active:
 188             self._active = True
 189             self.update_model.emit()
 190         return super().showEvent(event)
 191
 192     def tree_double_clicked(self, item, _column):
 193         path = core.abspath(item.path)
 194         cmds.do(cmds.OpenRepo, self.context, path)
 195
 196
 197 class BuildItem:
 198     def __init__(self):
 199         self.state_folder_map = {}
 200         self.state_folder_map[''] = icons.folder()
 201         self.state_folder_map['+'] = icons.staged()
 202         self.state_folder_map['-'] = icons.modified()
 203         self.state_folder_map['U'] = icons.merge()
 204
 205     def get(self, entry):
 206         """entry: same as returned from list_submodule"""
 207         name = entry[2]
 208         path = entry[2]
 209         tip = path + '\n' + entry[1]
 210         if entry[3]:
 211             tip += f'\n({entry[3]})'
 212         icon = self.state_folder_map[entry[0]]
 213         return SubmodulesTreeWidgetItem(name, path, tip, icon)
 214
 215
 216 class SubmodulesTreeWidgetItem(QtWidgets.QTreeWidgetItem):
 217     def __init__(self, name, path, tip, icon):
 218         QtWidgets.QTreeWidgetItem.__init__(self)
 219         self.path = path
 220
 221         self.setIcon(0, icon)
 222         self.setText(0, name)
 223         self.setToolTip(0, tip)