Add environments customization

Author: dolfbarr

src/index.test.tsx

@@ -4,8 +4,10 @@ import { useEffect, useState } from 'react'
 
 describe('useLog', () => {
   const OLD_ENV = process.env
-  const consoleLog = jest.spyOn(console, 'log')
-  const consoleGroup = jest.spyOn(console, 'group')
+  const consoleLog = jest.spyOn(console, 'log').mockImplementation(() => null)
+  const consoleGroup = jest
+    .spyOn(console, 'group')
+    .mockImplementation(() => null)
 
   beforeEach(() => {
     jest.useFakeTimers()
@@ -194,4 +196,15 @@ describe('useLog', () => {
 
     expect(consoleGroup.mock.calls[1][1]).toBe('color: darkRed;')
   })
+
+  it('renders anything in custom allowed environments', () => {
+    process.env.NODE_ENV = 'test_env'
+
+    renderHook(() => {
+      const { log } = useLog({ environments: ['test_env'] })
+      log('Test')
+    })
+
+    expect(consoleLog).toBeCalled()
+  })
 })

src/index.tsx

@@ -12,6 +12,7 @@ export interface UseLog {
     changeCSS?: string
     subValueCSS?: string
   }
+  environments?: string[]
 }
 
 export type Log = UseLog
@@ -40,6 +41,7 @@ export function useLog({
     changeCSS = CSS_CHANGE,
     subValueCSS = CSS_SUB_VALUE,
   } = {},
+  environments = ALLOWED_NODE_ENVS,
 }: UseLog = {}): UseLogReturn {
   const componentName =
     (function getComponentName() {
@@ -97,7 +99,7 @@ export function useLog({
       console.groupEnd()
     }
 
-    if (ALLOWED_NODE_ENVS.includes(process.env.NODE_ENV ?? '')) {
+    if (environments.includes(process.env.NODE_ENV ?? '')) {
       return (function logHooks() {
         const isUnmounting = useRef(false)
         useEffect(function setIsUnmounting() {