From 113243828f5c310b3f287a8e3a98cf14962051cb Mon Sep 17 00:00:00 2001
From: David Lechner <david@pybricks.com>
Date: Thu, 10 Dec 2020 16:59:37 -0600
Subject: [PATCH] add snippets for primehub and inventor hub

---
 src/components/Editor.tsx         |  3 ++-
 src/components/editor-snippets.ts | 20 +++++++++++++++++---
 2 files changed, 19 insertions(+), 4 deletions(-)

diff --git a/src/components/Editor.tsx b/src/components/Editor.tsx
index 708fdbd4..957d99db 100644
--- a/src/components/Editor.tsx
+++ b/src/components/Editor.tsx
@@ -16,12 +16,13 @@ import { EditorStringId } from './editor-i18n';
 import en from './editor-i18n.en.json';
 
 import 'ace-builds/src-noconflict/mode-python';
-import './editor-snippets';
 import 'ace-builds/src-noconflict/theme-xcode';
 import 'ace-builds/src-noconflict/ext-searchbox';
 import 'ace-builds/src-noconflict/ext-keybinding_menu';
 import 'ace-builds/src-noconflict/ext-language_tools';
 
+import './editor-snippets';
+
 type DispatchProps = {
     onSessionChanged: (session?: Ace.EditSession) => void;
     onProgramStorageChanged: (newValue: string) => void;
diff --git a/src/components/editor-snippets.ts b/src/components/editor-snippets.ts
index 6c7ce1ce..5c2645c3 100644
--- a/src/components/editor-snippets.ts
+++ b/src/components/editor-snippets.ts
@@ -20,24 +20,38 @@ ace.define(
         exports.snippetText = `snippet technichub
 	from pybricks.hubs import TechnicHub
 	from pybricks.pupdevices import Motor
-	from pybricks.parameters import Port, Stop
+	from pybricks.parameters import Port, Stop, Color
 	from pybricks.tools import wait
 
 	hub = TechnicHub()
 snippet cityhub
 	from pybricks.hubs import CityHub
 	from pybricks.pupdevices import Motor
-	from pybricks.parameters import Port, Stop
+	from pybricks.parameters import Port, Stop, Color
 	from pybricks.tools import wait
 
 	hub = CityHub()
 snippet movehub
 	from pybricks.hubs import MoveHub
 	from pybricks.pupdevices import Motor
-	from pybricks.parameters import Port, Stop
+	from pybricks.parameters import Port, Stop, Color
 	from pybricks.tools import wait
 
 	hub = MoveHub()
+snippet primehub
+	from pybricks.hubs import PrimeHub
+	from pybricks.pupdevices import Motor, ColorSensor, ForceSensor, UltrasonicSensor
+	from pybricks.parameters import Port, Stop, Color, Button
+	from pybricks.tools import wait
+
+	hub = PrimeHub()
+snippet inventorhub
+	from pybricks.hubs import InventorHub
+	from pybricks.pupdevices import Motor, ColorSensor, UltrasonicSensor
+	from pybricks.parameters import Port, Stop, Color, Button
+	from pybricks.tools import wait
+
+	hub = InventorHub()
 snippet uname
 	from uos import uname