33 lines
918 B
TypeScript
33 lines
918 B
TypeScript
import React from 'react'
|
|
import { AmpStateContext } from './amp-context'
|
|
|
|
export function isInAmpMode({
|
|
ampFirst = false,
|
|
hybrid = false,
|
|
hasQuery = false,
|
|
} = {}) {
|
|
return ampFirst || (hybrid && hasQuery)
|
|
}
|
|
|
|
export function useAmp() {
|
|
const ampState = React.useContext(AmpStateContext)
|
|
// un-comment below to not be considered AMP in dirty mode
|
|
return isInAmpMode(ampState) // && ampMode.hasQuery
|
|
}
|
|
|
|
/**
|
|
* @deprecated This is no longer required, use export const config = { amp: true }
|
|
*/
|
|
export function withAmp(Component: any, { hybrid = false } = {}): any {
|
|
function WithAmpWrapper(props = {}) {
|
|
const ampState = React.useContext(AmpStateContext)
|
|
ampState.ampFirst = !hybrid
|
|
ampState.hybrid = hybrid
|
|
|
|
return React.createElement(Component, props)
|
|
}
|
|
|
|
WithAmpWrapper.__nextAmpOnly = !hybrid
|
|
WithAmpWrapper.getInitialProps = Component.getInitialProps
|
|
return WithAmpWrapper
|
|
}
|