求助：将网站原生JavaScript代码转换为jQuery代码

原代码核心是点击不同颜色按钮，操作聊天元素的data-color属性，以下是对应的jQuery写法：

const $blue = $('#color-blue'),
      $red = $('#color-red'),
      $chat = $('#chat'); // 这里假设chat元素的选择器是#chat，请根据实际页面结构调整

$blue.on('click', () => {
    $chat.removeAttr('data-color');
});

$red.on('click', () => {
    $chat.attr('data-color', 'red');
});

补充说明： #

• jQuery选择器返回的是jQuery对象，习惯上给变量名加$前缀，方便区分原生DOM对象
• 如果只是需要在jQuery内部缓存数据，而非修改DOM上的data-color属性，可以改用.removeData('color')和.data('color', 'red')，但这不会同步到DOM属性上。原代码操作的是原生dataset（即DOM属性），所以推荐用.attr()和.removeAttr()来匹配原逻辑

内容的提问来源于stack exchange，提问作者Alwin