Respond to OSC color query escape codes

kitty/rgb.py

@@ -37,7 +37,7 @@ def to_color(raw, validate=False):
     try:
         if raw.startswith('#'):
             val = parse_sharp(raw[1:])
-        elif raw.startswith('rgb:'):
+        elif raw[:4].lower() == 'rgb:':
             val = parse_rgb(raw[4:])
     except Exception:
         pass

kitty/utils.py

@@ -109,7 +109,10 @@ def parse_color_set(raw):
         try:
             c = int(c)
             if c < 0 or c > 255:
-                raise IndexError('Out of bounds')
+                continue
+            if spec == '?':
+                yield c, None
+            else:
                 r, g, b = to_color(spec)
                 yield c, r << 16 | g << 8 | b
         except Exception:

kitty/window.py

@@ -206,6 +206,12 @@ class Window:
         if dirtied:
             self.screen.mark_as_dirty()
 
+    def report_color(self, osc, r, g, b):
+        r |= r << 8
+        g |= g << 8
+        b |= b << 8
+        self.write_to_child('\033]{};rgb:{:04x}/{:04x}/{:04x}\033\\'.format(osc, r, g, b))
+
     def set_dynamic_color(self, code, value):
         if isinstance(value, bytes):
             value = value.decode('utf-8')
@@ -213,18 +219,29 @@ class Window:
         for val in value.split(';'):
             w = DYNAMIC_COLOR_CODES.get(code)
             if w is not None:
+                if val == '?':
+                    col = getattr(self.screen.color_profile, w.name)
+                    self.report_color(str(code), col >> 16, (col >> 8) & 0xff, col & 0xff)
+                else:
                     if code >= 110:
                         val = None
                     color_changes[w] = val
             code += 1
+        if color_changes:
             self.change_colors(color_changes)
             glfw_post_empty_event()
 
     def set_color_table_color(self, code, value):
         cp = self.screen.color_profile
         if code == 4:
+            changed = False
             for c, val in parse_color_set(value):
+                if val is None:  # color query
+                    self.report_color('4;{}'.format(c), *self.screen.color_profile.as_color((c << 8) | 1))
+                else:
+                    changed = True
                     cp.set_color(c, val)
+            if changed:
                 self.refresh()
         elif code == 104:
             if not value.strip():