feat: useMedia - initialize state with call to media query (#454)

streamich · Jul 11, 2019 · ab81897 · ab81897
2 parents c4f5f16 + eeb55e4
commit ab81897
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/useMedia.ts b/src/useMedia.ts
@@ -1,7 +1,7 @@
 import { useEffect, useState } from 'react';
 
-const useMedia = (query: string, defaultState: boolean = false) => {
-  const [state, setState] = useState(defaultState);
+const useMedia = (query: string) => {
+  const [state, setState] = useState(() => window.matchMedia(query).matches);
 
   useEffect(() => {
     let mounted = true;