Fix the issue where menu-items were not loading

Author: Almoizs-1310

src/server/index.ts

@@ -1,5 +1,6 @@
 import {
   onOpen,
+  onInstall,
   refreshMenu,
   openCreateDiagramDialog,
   openEditDiagramDialog,
@@ -32,6 +33,7 @@ import {
 // Public functions must be exported as named exports
 export {
   onOpen,
+  onInstall,
   refreshMenu,
   openCreateDiagramDialog,
   openEditDiagramDialog,

src/server/ui.js

@@ -1,22 +1,39 @@
 function buildMenu() {
-  const authState = getAuthorizationState();
   const menu = DocumentApp.getUi().createAddonMenu();
+  
   menu.addItem('Start', 'openSidebar');
-  if (authState.authorized) {
+  
+  let authorized = false;
+  
+  try {
+    const authState = getAuthorizationState();
+    authorized = authState && authState.authorized;
+  } catch (e) {
+    console.log('Error checking authorization state: ' + e);
+    Logger.log('Auth check failed: ' + e);
+  }
+  
+  if (authorized) {
     menu.addSeparator();
     menu.addItem('New diagram', 'openCreateDiagramDialog');
     menu.addItem('Browse diagrams', 'openSelectDiagramDialog');
     menu.addItem('Edit selected diagram', 'openEditDiagramDialog');
   }
+  
   menu.addSeparator();
   menu.addItem('About', 'openAboutDialog');
+  
   menu.addToUi();
 }
 
-export function onOpen() {
+export function onOpen(e) {
   buildMenu();
 }
 
+export function onInstall(e) {
+  onOpen(e);
+}
+
 export function refreshMenu() {
   buildMenu();
 }