Feat/enable godmode req

components/lib/ExpiryTypeComponent.tsx

@@ -0,0 +1,50 @@
+import React from 'react'
+import {
+  FormControl,
+  FormControlLabel,
+  Typography,
+  FormLabel,
+  RadioGroup,
+  Radio,
+  Grid
+} from '@material-ui/core'
+import { EXPIRY_TYPE, EXPIRY_TYPE_HUMAN } from '../../lib/constants'
+
+const ExpiryTypeComponent = ({ state, onChange }) => {
+  const expiryTypes = [
+    EXPIRY_TYPE.CURRENT,
+    EXPIRY_TYPE.NEXT,
+    EXPIRY_TYPE.MONTHLY
+  ]
+  return (
+    <Grid item xs={12}>
+      <FormControl component='fieldset'>
+        <FormLabel component='legend'>Option Expiry</FormLabel>
+        <RadioGroup
+          aria-label='expiryTypes'
+          name='expiryType'
+          value={state.expiryType}
+          onChange={e =>
+            onChange({ expiryType: e.target.value as EXPIRY_TYPE })
+          }
+          row
+        >
+          {expiryTypes.map(expiryType => (
+            <FormControlLabel
+              key={expiryType}
+              value={expiryType}
+              control={<Radio size='small' />}
+              label={
+                <Typography variant='body2'>
+                  {EXPIRY_TYPE_HUMAN[expiryType]}
+                </Typography>
+              }
+            />
+          ))}
+        </RadioGroup>
+      </FormControl>
+    </Grid>
+  )
+}
+
+export default ExpiryTypeComponent

components/lib/commonDetailsRows.tsx

@@ -20,7 +20,8 @@ const CommonDetailsRows = ({
   volatilityType,
   liveTrailingSl,
   lastTrailingSlSetAt,
-  lastHeartbeatAt
+  lastHeartbeatAt,
+  expiryType
 }) => {
   const getAlgoStatus = useCallback(
     () =>
@@ -55,6 +56,8 @@ const CommonDetailsRows = ({
   return [
     volatilityType ? [{ value: 'Vol type' }, { value: volatilityType }] : null,
     productType ? [{ value: 'Product' }, { value: productType }] : null,
+    expiryType ? [{ value: 'Expiry' }, { value: expiryType }] : null,
+
     [
       { value: 'Exit Strategy' },
       { value: EXIT_STRATEGIES_DETAILS[exitStrategy].label }

components/trades/atmStraddle/TradeSetupForm.tsx

@@ -34,6 +34,7 @@ import ProductTypeComponent from '../../lib/ProductTypeComponent'
 import VolatilityTypeComponent from '../../lib/VolatilityTypeComponent'
 import RollbackComponent from '../../lib/RollbackComponent'
 import SlManagerComponent from '../../lib/SlManagerComponent'
+import ExpiryTypeComponent from '../../lib/ExpiryTypeComponent'
 
 interface ATMStraddleTradeSetupFormProps {
   formHeading?: string
@@ -113,6 +114,8 @@ const TradeSetupForm = ({
 
           <ProductTypeComponent state={state} onChange={onChange} />
 
+          <ExpiryTypeComponent state={state} onChange={onChange} />
+
           <Grid item xs={12}>
             <TextField
               fullWidth

components/trades/atmStrangle/TradeSetupForm.tsx

@@ -36,6 +36,7 @@ import ProductTypeComponent from '../../lib/ProductTypeComponent'
 import RollbackComponent from '../../lib/RollbackComponent'
 import DiscreteSlider from '../../lib/Slider'
 import SlManagerComponent from '../../lib/SlManagerComponent'
+import ExpiryTypeComponent from '../../lib/ExpiryTypeComponent'
 
 interface ATMStrangleTradeSetupFormProps {
   formHeading?: string
@@ -116,6 +117,8 @@ const TradeSetupForm = ({
 
           <ProductTypeComponent state={state} onChange={onChange} />
 
+          <ExpiryTypeComponent state={state} onChange={onChange} />
+
           <Grid item xs={12}>
             <FormControl component='fieldset'>
               <FormLabel component='legend'>Entry strategy</FormLabel>

components/trades/directionalOptionSelling/TradeSetupForm.tsx

@@ -38,6 +38,7 @@ import HedgeComponent from '../../lib/HedgeComponent'
 import RollbackComponent from '../../lib/RollbackComponent'
 import ProductTypeComponent from '../../lib/ProductTypeComponent'
 import SlManagerComponent from '../../lib/SlManagerComponent'
+import ExpiryTypeComponent from '../../lib/ExpiryTypeComponent'
 
 interface DOSTradeSetupFormProps {
   state: Partial<DIRECTIONAL_OPTION_SELLING_CONFIG>
@@ -114,6 +115,8 @@ const TradeSetupForm = ({
 
           <ProductTypeComponent state={state} onChange={onChange} />
 
+          <ExpiryTypeComponent state={state} onChange={onChange} />
+
           <Grid item xs={12}>
             <TextField
               fullWidth

lib/browserUtils.ts

@@ -133,7 +133,8 @@ export const formatFormDataForApi = ({
         martingaleIncrementSize,
         strikeByPrice,
         slmPercent,
-        exitStrategy
+        exitStrategy,
+        expiryType
       } = data as DIRECTIONAL_OPTION_SELLING_CONFIG
 
       const apiProps: DIRECTIONAL_OPTION_SELLING_TRADE = {
@@ -142,6 +143,7 @@ export const formatFormDataForApi = ({
         martingaleIncrementSize: Number(martingaleIncrementSize),
         slmPercent: Number(slmPercent),
         maxTrades: Number(maxTrades),
+        expiryType,
         strikeByPrice: strikeByPrice ? Number(strikeByPrice) : undefined,
         ...getSchedulingApiProps({
           isAutoSquareOffEnabled,
@@ -169,7 +171,8 @@ export const formatFormDataForApi = ({
         trailEveryPercentageChangeValue,
         trailingSlPercent,
         exitStrategy,
-        combinedExitStrategy
+        combinedExitStrategy,
+        expiryType
       } = data as ATM_STRADDLE_CONFIG
 
       const apiProps: ATM_STRADDLE_TRADE = {
@@ -184,6 +187,7 @@ export const formatFormDataForApi = ({
           exitStrategy,
           combinedExitStrategy
         }),
+        expiryType,
         maxSkewPercent: Number(maxSkewPercent),
         thresholdSkewPercent: Number(thresholdSkewPercent),
         ...getSchedulingApiProps({
@@ -212,7 +216,8 @@ export const formatFormDataForApi = ({
         trailingSlPercent,
         exitStrategy,
         expireIfUnsuccessfulInMins,
-        combinedExitStrategy
+        combinedExitStrategy,
+        expiryType
       } = data as ATM_STRANGLE_CONFIG
 
       const apiProps: ATM_STRANGLE_TRADE = {
@@ -227,6 +232,7 @@ export const formatFormDataForApi = ({
           exitStrategy,
           combinedExitStrategy
         }),
+        expiryType,
         inverted: Boolean(inverted),
         ...getSchedulingApiProps({
           isAutoSquareOffEnabled,

lib/constants.ts

@@ -22,6 +22,21 @@ export interface INSTRUMENT_PROPERTIES {
   freezeQty: number
 }
 
+export interface KITE_INSTRUMENT_INFO {
+  exchange: string // "NFO"
+  exchange_token: string // "36320"
+  expiry: string // "2021-10-28"
+  instrument_token: string // "9297922"
+  instrument_type: string // "CE"
+  last_price: string // "0"
+  lot_size: string // "50"
+  name: string // "NIFTY"
+  segment: string // "NFO-OPT"
+  strike: string // "17350"
+  tick_size: string // "0.05"
+  tradingsymbol: string // "NIFTY21OCT17350CE"
+}
+
 export const INSTRUMENT_DETAILS: Record<INSTRUMENTS, INSTRUMENT_PROPERTIES> = {
   [INSTRUMENTS.NIFTY]: {
     lotSize: 50,
@@ -86,6 +101,18 @@ export enum VOLATILITY_TYPE {
   SHORT = 'SHORT'
 }
 
+export enum EXPIRY_TYPE {
+  CURRENT = 'CURRENT',
+  NEXT = 'NEXT',
+  MONTHLY = 'MONTHLY'
+}
+
+export const EXPIRY_TYPE_HUMAN = {
+  [EXPIRY_TYPE.CURRENT]: 'Current weekly',
+  [EXPIRY_TYPE.NEXT]: 'Next weekly',
+  [EXPIRY_TYPE.MONTHLY]: 'Current Monthly'
+}
+
 export enum STRANGLE_ENTRY_STRATEGIES {
   DISTANCE_FROM_ATM = 'DISTANCE_FROM_ATM',
   DELTA_STIKES = 'DELTA_STIKES'
@@ -136,6 +163,7 @@ export const STRATEGIES_DETAILS = {
       trailingSlPercent: NEXT_PUBLIC_DEFAULT_SLM_PERCENT,
       productType: PRODUCT_TYPE.MIS,
       volatilityType: VOLATILITY_TYPE.SHORT,
+      expiryType: EXPIRY_TYPE.CURRENT,
       runNow: false,
       expireIfUnsuccessfulInMins: 10,
       exitStrategy: EXIT_STRATEGIES.INDIVIDUAL_LEG_SLM_1X,
@@ -173,6 +201,7 @@ export const STRATEGIES_DETAILS = {
       deltaStrikes: 20,
       productType: PRODUCT_TYPE.MIS,
       volatilityType: VOLATILITY_TYPE.SHORT,
+      expiryType: EXPIRY_TYPE.CURRENT,
       runNow: false,
       exitStrategy: EXIT_STRATEGIES.INDIVIDUAL_LEG_SLM_1X,
       slOrderType: SL_ORDER_TYPE.SLL,
@@ -214,6 +243,7 @@ export const STRATEGIES_DETAILS = {
       martingaleIncrementSize: 1,
       isHedgeEnabled: true,
       productType: PRODUCT_TYPE.MIS,
+      expiryType: EXPIRY_TYPE.CURRENT,
       hedgeDistance: 2000,
       entryStrategy: DOS_ENTRY_STRATEGIES.FIXED_TIME,
       exitStrategy: EXIT_STRATEGIES.MIN_XPERCENT_OR_SUPERTREND,

lib/queue-processor/optionSellerStrategy/optionEntryWatcher.js

@@ -18,7 +18,7 @@ const optionSellerEntryWatcher = async ({
   addHedge
 }) => {
   try {
-    const { user, orderTag, instrument } = initialJobData
+    const { user, orderTag, instrument, expiryType } = initialJobData
     const kite = syncGetKiteInstance(user)
     const orderHistory = await kite.getOrderHistory(limitOrderAckId)
     const revOrderHistory = orderHistory.reverse()
@@ -81,7 +81,8 @@ const optionSellerEntryWatcher = async ({
         completedOrder.tradingsymbol.length - 2,
         completedOrder.tradingsymbol.length - 1
       ),
-      user
+      user,
+      expiry: expiryType
     })
 
     const hedgeOrder = {

lib/queue-processor/tradingQueue.ts

@@ -21,6 +21,7 @@ async function processJob (job: Job) {
     data,
     data: { strategy }
   } = job
+  console.log(`[job processing] Beginning job processing for ${strategy}`)
   switch (strategy) {
     case STRATEGIES.ATM_STRADDLE: {
       return atmStraddle(data)

lib/session.ts

@@ -1,13 +1,70 @@
 // this file is a wrapper with defaults to be used in both API routes and `getServerSideProps` functions
 import { withIronSession } from 'next-iron-session'
+import { KiteConnect } from 'kiteconnect'
 // NB: not the best place to require these
 // ideally these should live in their own file that gets included as a middleware
 import './queue-processor'
 import './exit-strategies'
 import './watchers'
+import { setUserSession, premiumAuthCheck } from './utils'
+import { KiteProfile } from '../types/kite'
+
+const withAdminCheck = handler => {
+  return async function withAdminWrapper (req, res) {
+    const sxApiKey = req.headers['signalx-api-key']
+    const kiteApiKey = req.headers['signalx-kite-key']
+    const kiteAccessToken = req.headers['signalx-kite-token']
+
+    if (!sxApiKey && !kiteApiKey) {
+      // UI based flow
+      return handler(req, res)
+    }
+
+    // for premium and club users
+    let validSxUser = false
+    try {
+      validSxUser = await premiumAuthCheck()
+    } catch (e) {
+      validSxUser = false
+    }
+
+    // block intrusion
+    if (!validSxUser) {
+      return handler(req, res)
+    }
+
+    if (!kiteAccessToken) {
+      // sx admin work, no broker authorization
+      await setUserSession(req, {} as KiteProfile)
+    } else {
+      console.log(
+        'key and token found in headers. attempting to connect kite and save session'
+      )
+      try {
+        const kc = new KiteConnect({
+          api_key: kiteApiKey,
+          access_token: kiteAccessToken
+        })
+
+        const kiteProfile = await kc.getProfile()
+        await setUserSession(req, {
+          access_token: kiteAccessToken,
+          ...kiteProfile
+        })
+        console.log('session generated from headers')
+        return handler(req, res)
+      } catch (error) {
+        console.log(error)
+        return res
+          .status(403)
+          .send('Forbidden. Unauthorized key or token provided')
+      }
+    }
+  }
+}
 
 export default function withSession (handler) {
-  return withIronSession(handler, {
+  return withIronSession(withAdminCheck(handler), {
     password: process.env.SECRET_COOKIE_PASSWORD!,
     cookieName: 'khaching/kite/session',
     cookieOptions: {