Make dynamic buttons work along with click events

atom/browser/api/atom_api_window.cc

@@ -283,7 +283,7 @@ void Window::OnExecuteWindowsCommand(const std::string& command_name) {
 }
 
 void Window::OnTouchBarItemResult(const std::string& item_type, const std::string& item_id) {
-  Emit("_touch-bar-interaction", item_type, item_id);
+  Emit("-touch-bar-interaction", item_type, item_id);
 }
 
 #if defined(OS_WIN)
@@ -844,8 +844,12 @@ void Window::SetVibrancy(mate::Arguments* args) {
   window_->SetVibrancy(type);
 }
 
-void Window::InitTouchBar()  {
-  window_->InitTouchBar();
+void Window::DestroyTouchBar()  {
+  window_->DestroyTouchBar();
+}
+
+void Window::SetTouchBar(mate::Arguments* args) {
+  window_->SetTouchBar(args);
 }
 
 int32_t Window::ID() const {
@@ -968,7 +972,8 @@ void Window::BuildPrototype(v8::Isolate* isolate,
       .SetMethod("setAutoHideCursor", &Window::SetAutoHideCursor)
 #endif
       .SetMethod("setVibrancy", &Window::SetVibrancy)
-      .SetMethod("initTouchBar", &Window::InitTouchBar)
+      .SetMethod("_destroyTouchBar", &Window::DestroyTouchBar)
+      .SetMethod("_setTouchBar", &Window::SetTouchBar)
 #if defined(OS_WIN)
       .SetMethod("hookWindowMessage", &Window::HookWindowMessage)
       .SetMethod("isWindowMessageHooked", &Window::IsWindowMessageHooked)

atom/browser/api/atom_api_window.h

@@ -204,7 +204,8 @@ class Window : public mate::TrackableObject<Window>,
   void SetAutoHideCursor(bool auto_hide);
 
   void SetVibrancy(mate::Arguments* args);
-  void InitTouchBar();
+  void DestroyTouchBar();
+  void SetTouchBar(mate::Arguments* args);
 
   v8::Local<v8::Value> WebContents(v8::Isolate* isolate);
 

atom/browser/native_window.cc

@@ -33,6 +33,7 @@
 #include "content/public/browser/render_widget_host_view.h"
 #include "content/public/common/content_switches.h"
 #include "ipc/ipc_message_macros.h"
+#include "native_mate/constructor.h"
 #include "native_mate/dictionary.h"
 #include "third_party/skia/include/core/SkRegion.h"
 #include "ui/gfx/codec/png_codec.h"
@@ -340,7 +341,10 @@ void NativeWindow::SetAutoHideCursor(bool auto_hide) {
 void NativeWindow::SetVibrancy(const std::string& filename) {
 }
 
-void NativeWindow::InitTouchBar() {
+void NativeWindow::DestroyTouchBar() {
+}
+
+void NativeWindow::SetTouchBar(mate::Arguments* args) {
 }
 
 void NativeWindow::FocusOnWebView() {

atom/browser/native_window.h

@@ -23,6 +23,7 @@
 #include "extensions/browser/app_window/size_constraints.h"
 #include "ui/gfx/image/image.h"
 #include "ui/gfx/image/image_skia.h"
+#include "native_mate/constructor.h"
 
 class SkRegion;
 
@@ -170,7 +171,8 @@ class NativeWindow : public base::SupportsUserData,
   virtual void SetVibrancy(const std::string& type);
 
   // Touchbar API
-  virtual void InitTouchBar();
+  virtual void DestroyTouchBar();
+  virtual void SetTouchBar(mate::Arguments* args);
 
   // Webview APIs.
   virtual void FocusOnWebView();

atom/browser/native_window_mac.h

@@ -13,6 +13,7 @@
 #include "atom/browser/native_window.h"
 #include "base/mac/scoped_nsobject.h"
 #include "content/public/browser/render_widget_host.h"
+#include "native_mate/constructor.h"
 
 @class AtomNSWindow;
 @class AtomNSWindowDelegate;
@@ -100,7 +101,9 @@ class NativeWindowMac : public NativeWindow,
   void SetAutoHideCursor(bool auto_hide) override;
 
   void SetVibrancy(const std::string& type) override;
-  void InitTouchBar() override;
+  void DestroyTouchBar() override;
+  void SetTouchBar(mate::Arguments* args) override;
+  std::vector<mate::Dictionary> GetTouchBarItems();
 
   // content::RenderWidgetHost::InputEventObserver:
   void OnInputEvent(const blink::WebInputEvent& event) override;
@@ -155,6 +158,8 @@ class NativeWindowMac : public NativeWindow,
   base::scoped_nsobject<AtomNSWindow> window_;
   base::scoped_nsobject<AtomNSWindowDelegate> window_delegate_;
 
+  std::vector<mate::Dictionary> touch_bar_items_;
+
   // Event monitor for scroll wheel event.
   id wheel_event_monitor_;
 

atom/browser/native_window_mac.mm

@@ -352,6 +352,8 @@ bool ScopedDisableResize::disable_resize_ = false;
 - (void)setEnableLargerThanScreen:(bool)enable;
 - (void)enableWindowButtonsOffset;
 - (void)reloadTouchBar;
+- (void)resetTouchBar;
+- (NSTouchBar*)touchBarFromMutatableArray:(NSMutableArray<NSTouchBarItemIdentifier>*)items;
 @end
 
 @interface AtomNSWindow () <NSTouchBarDelegate>
@@ -359,6 +361,7 @@ bool ScopedDisableResize::disable_resize_ = false;
 
 @implementation AtomNSWindow
   NSMutableArray<NSTouchBarItemIdentifier>* bar_items_ = [[NSMutableArray alloc] init];
+  std::map<std::string, std::string> button_labels;
 
 - (void)setShell:(atom::NativeWindowMac*)shell {
   shell_ = shell;
@@ -368,24 +371,52 @@ bool ScopedDisableResize::disable_resize_ = false;
   enable_larger_than_screen_ = enable;
 }
 
+- (void)resetTouchBar {
+  bar_items_ = [[NSMutableArray alloc] init];
+  self.touchBar = nil;
+  NSLog(@"Destroying TouchBar");
+}
+
 - (void)reloadTouchBar {
   bar_items_ = [[NSMutableArray alloc] init];
-  [bar_items_ addObject:@"com.electron.tb.button.1"];
-  [bar_items_ addObject:@"com.electron.tb.button.2"];
+  std::vector<mate::Dictionary> items = shell_->GetTouchBarItems();
+  std::map<std::string, std::string> new_button_labels;
+  button_labels = new_button_labels;
+
+  NSLog(@"reload");
+  for (mate::Dictionary &item : items ) {
+    NSLog(@"reload iter");
+    std::string type;
+    std::string item_id;
+    if (item.Get("type", &type) && item.Get("id", &item_id)) {
+      NSLog(@"type: %@", [NSString stringWithUTF8String:type.c_str()]);
+      NSLog(@"id: %@", [NSString stringWithUTF8String:item_id.c_str()]);
+      if (type == "button") {
+        std::string label;
+        if (item.Get("label", &label)) {
+          [bar_items_ addObject:[NSString stringWithFormat:@"%@%@", ButtonIdentifier, [NSString stringWithUTF8String:item_id.c_str()]]];
+          button_labels.insert(make_pair(item_id, label));
+        }
+      }
+    }
+  }
+  // [bar_items_ addObject:@"com.electron.tb.button.1"];
+  // [bar_items_ addObject:@"com.electron.tb.button.2"];
   [bar_items_ addObject:NSTouchBarItemIdentifierOtherItemsProxy];
-  NSLog(@"Reloading Touch Bar --> '%@'", bar_items_[1]);
+  // NSLog(@"Reloading Touch Bar --> '%@'", bar_items_[1]);
   self.touchBar = nil;
 }
 
 - (NSTouchBar *)makeTouchBar {
   NSLog(@"Making Touch Bar");
+  return [self touchBarFromMutatableArray:bar_items_];
+}
+
+- (NSTouchBar *)touchBarFromMutatableArray:(NSMutableArray<NSTouchBarItemIdentifier>*)items {
   NSTouchBar* bar = [[NSTouchBar alloc] init];
   bar.delegate = self;
 
-  // Set the default ordering of items.
-
-  // NSLog(@"%@", bar_items_[1]);
-  bar.defaultItemIdentifiers = [bar_items_ copy];
+  bar.defaultItemIdentifiers = [items copy];
 
   return bar;
 }
@@ -412,7 +443,7 @@ static NSTouchBarItemIdentifier LabelIdentifier = @"com.electron.tb.label.";
   if ([identifier hasPrefix:ButtonIdentifier]) {
     NSString *idCopy = [identifier copy];
     idCopy = [identifier substringFromIndex:[ButtonIdentifier length]];
-    NSButton *theButton = [NSButton buttonWithTitle:@"Electron Button" target:self action:@selector(buttonAction:)];
+    NSButton *theButton = [NSButton buttonWithTitle:[NSString stringWithUTF8String:button_labels[std::string([idCopy UTF8String])].c_str()] target:self action:@selector(buttonAction:)];
     theButton.tag = [idCopy floatValue];
 
     NSCustomTouchBarItem *customItem = [[NSCustomTouchBarItem alloc] initWithIdentifier:identifier];
@@ -1420,8 +1451,22 @@ void NativeWindowMac::SetVibrancy(const std::string& type) {
   [effect_view setMaterial:vibrancyType];
 }
 
-void NativeWindowMac::InitTouchBar() {
-  [window_ reloadTouchBar];
+void NativeWindowMac::DestroyTouchBar() {
+  [window_ resetTouchBar];
+}
+
+void NativeWindowMac::SetTouchBar(mate::Arguments* args) {
+  std::vector<mate::Dictionary> items;
+  LOG(ERROR) << "FOO";
+  if (args->GetNext(&items)) {
+    LOG(ERROR) << "BAR";
+    touch_bar_items_ = items;
+    [window_ reloadTouchBar];
+  }
+}
+
+std::vector<mate::Dictionary> NativeWindowMac::GetTouchBarItems() {
+  return touch_bar_items_;
 }
 
 void NativeWindowMac::OnInputEvent(const blink::WebInputEvent& event) {