Merge pull request #80 from neftaly/polyfill

Improve polyfill
EventSource · Jul 18, 2017 · 42805b9 · 42805b9
2 parents a622b07 + 92e4795
commit 42805b9
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/lib/eventsource-polyfill.js b/lib/eventsource-polyfill.js
@@ -1,2 +1,9 @@
-window.EventSourcePolyfill = require('./eventsource')
-window.EventSource = window.EventSource || window.EventSourcePolyfill
+var EventSource = require('./eventsource')
+
+if (typeof window === 'object') {
+  window.EventSourcePolyfill = EventSource
+  if (!window.EventSource) window.EventSource = EventSource
+  module.exports = window.EventSource
+} else {
+  module.exports = EventSource
+}