import { useState } from 'react';
import { View, Button, Text } from 'react-native';
import { useStytchB2BClient } from '@stytch/react-native/b2b';
import { SessionTokens } from '@stytch/react-native/b2b';

export const GetSessionTokens = () => {
  const stytch = useStytchB2BClient();
  const [sessionTokens, setSessionTokens] = useState<SessionTokens | null>(null);

  // Callback to retrieve session tokens on demand
  const handleGetTokens = () => {
    const tokens = stytch.session.getTokens();
    setSessionTokens(tokens);
  };

  return (
    <View>
      <Button title="Get Session Tokens" onPress={handleGetTokens} />
      {sessionTokens && (
        <View>
          <Text>Session Token: {sessionTokens.session_token}</Text>
          <Text>Session JWT: {sessionTokens.session_jwt}</Text>
        </View>
      )}
    </View>
  );
};