sigit.io/src/pages/create/index.tsx

import { Clear, DragHandle } from '@mui/icons-material'
import {
  Box,
  Button,
  FormControl,
  IconButton,
  InputLabel,
  MenuItem,
  Paper,
  Select,
  Table,
  TableBody,
  TableCell,
  TableContainer,
  TableHead,
  TableRow,
  TextField,
  Tooltip,
  Typography
} from '@mui/material'
import JSZip from 'jszip'
import { MuiFileInput } from 'mui-file-input'
import { useEffect, useRef, useState } from 'react'
import { useSelector } from 'react-redux'
import { useNavigate } from 'react-router-dom'
import { toast } from 'react-toastify'
import { LoadingSpinner } from '../../components/LoadingSpinner'
import { UserComponent } from '../../components/username'
import { MetadataController, NostrController } from '../../controllers'
import { appPrivateRoutes } from '../../routes'
import { State } from '../../store/rootReducer'
import { Meta, ProfileMetadata, User, UserRole } from '../../types'
import {
  encryptArrayBuffer,
  generateEncryptionKey,
  generateKeysFile,
  getHash,
  hexToNpub,
  isOnline,
  npubToHex,
  queryNip05,
  sendDM,
  shorten,
  signEventForMetaFile,
  uploadToFileStorage
} from '../../utils'
import styles from './style.module.scss'
import { DndProvider } from 'react-dnd'
import { HTML5Backend } from 'react-dnd-html5-backend'
import type { Identifier, XYCoord } from 'dnd-core'
import { useDrag, useDrop } from 'react-dnd'
import saveAs from 'file-saver'
import { Event, kinds } from 'nostr-tools'

export const CreatePage = () => {
  const navigate = useNavigate()
  const [isLoading, setIsLoading] = useState(false)
  const [loadingSpinnerDesc, setLoadingSpinnerDesc] = useState('')

  const [authUrl, setAuthUrl] = useState<string>()

  const [title, setTitle] = useState('')
  const [selectedFiles, setSelectedFiles] = useState<File[]>([])

  const [userInput, setUserInput] = useState('')
  const [userRole, setUserRole] = useState<UserRole>(UserRole.signer)
  const [error, setError] = useState<string>()

  const [users, setUsers] = useState<User[]>([])

  const usersPubkey = useSelector((state: State) => state.auth.usersPubkey)

  const nostrController = NostrController.getInstance()

  useEffect(() => {
    if (usersPubkey) {
      setUsers((prev) => {
        const existingUserIndex = prev.findIndex(
          (user) => user.pubkey === usersPubkey
        )

        // make logged in user the first signer by default
        if (existingUserIndex === -1)
          return [{ pubkey: usersPubkey, role: UserRole.signer }, ...prev]

        return prev
      })
    }
  }, [usersPubkey])

  const handleAddUser = async () => {
    setError(undefined)

    const addUser = (pubkey: string) => {
      setUsers((prev) => {
        const signers = prev.filter((user) => user.role === UserRole.signer)
        const viewers = prev.filter((user) => user.role === UserRole.viewer)

        const existingUserIndex = prev.findIndex(
          (user) => user.pubkey === pubkey
        )

        // add new
        if (existingUserIndex === -1) {
          if (userRole === UserRole.signer) {
            return [...signers, { pubkey, role: userRole }, ...viewers]
          } else {
            return [...signers, ...viewers, { pubkey, role: userRole }]
          }
        }

        const existingUser = prev[existingUserIndex]

        // return existing
        if (existingUser.role === userRole) return prev

        // change user role
        const updatedUsers = [...prev]
        const updatedUser = { ...updatedUsers[existingUserIndex] }
        updatedUser.role = userRole
        updatedUsers[existingUserIndex] = updatedUser

        // signers should be placed at the start of the  array
        return [
          ...updatedUsers.filter((user) => user.role === UserRole.signer),
          ...updatedUsers.filter((user) => user.role === UserRole.viewer)
        ]
      })
    }

    const input = userInput.toLowerCase()

    if (input.startsWith('npub')) {
      const pubkey = npubToHex(input)
      if (pubkey) {
        addUser(pubkey)
        setUserInput('')
      } else {
        setError('Provided npub is not valid. Please enter correct npub.')
      }
      return
    }

    if (input.includes('@')) {
      setIsLoading(true)
      setLoadingSpinnerDesc('Querying for nip05')
      const nip05Profile = await queryNip05(input)
        .catch((err) => {
          console.error(`error occurred in querying nip05: ${input}`, err)
          return null
        })
        .finally(() => {
          setIsLoading(false)
          setLoadingSpinnerDesc('')
        })

      if (nip05Profile && nip05Profile.pubkey) {
        const pubkey = nip05Profile.pubkey
        addUser(pubkey)
        setUserInput('')
      } else {
        setError('Provided nip05 is not valid. Please enter correct nip05.')
      }
      return
    }

    setError('Invalid input! Make sure to provide correct npub or nip05.')
  }

  const handleUserRoleChange = (role: UserRole, pubkey: string) => {
    setUsers((prevUsers) =>
      prevUsers.map((user) => {
        if (user.pubkey === pubkey) {
          return {
            ...user,
            role
          }
        }

        return user
      })
    )
  }

  const handleRemoveUser = (pubkey: string) => {
    setUsers((prev) => prev.filter((user) => user.pubkey !== pubkey))
  }

  /**
   * changes the position of signer in the signers list
   *
   * @param dragIndex represents the current position of user
   * @param hoverIndex represents the target position of user
   */
  const moveSigner = (dragIndex: number, hoverIndex: number) => {
    setUsers((prevUsers) => {
      const updatedUsers = [...prevUsers]
      const [draggedUser] = updatedUsers.splice(dragIndex, 1)
      updatedUsers.splice(hoverIndex, 0, draggedUser)
      return updatedUsers
    })
  }

  const handleSelectFiles = (files: File[]) => {
    setSelectedFiles((prev) => {
      const prevFileNames = prev.map((file) => file.name)

      const newFiles = files.filter(
        (file) => !prevFileNames.includes(file.name)
      )

      return [...prev, ...newFiles]
    })
  }

  const handleRemoveFile = (fileToRemove: File) => {
    setSelectedFiles((prevFiles) =>
      prevFiles.filter((file) => file.name !== fileToRemove.name)
    )
  }

  // Validate inputs before proceeding
  const validateInputs = (): boolean => {
    if (!title.trim()) {
      toast.error('Title can not be empty')
      return false
    }

    if (users.length === 0) {
      toast.error(
        'No signer/viewer is provided. At least add one signer or viewer.'
      )
      return false
    }

    if (selectedFiles.length === 0) {
      toast.error('No file is selected. Select at least 1 file')
      return false
    }

    return true
  }

  // Handle errors during file arrayBuffer conversion
  const handleFileError = (file: File) => (err: any) => {
    console.log(
      `Error while getting arrayBuffer of file ${file.name} :>> `,
      err
    )
    toast.error(
      err.message || `Error while getting arrayBuffer of file ${file.name}`
    )
    return null
  }

  // Generate hash for each selected file
  const generateFileHashes = async (): Promise<{
    [key: string]: string
  } | null> => {
    const fileHashes: { [key: string]: string } = {}

    for (const file of selectedFiles) {
      const arraybuffer = await file.arrayBuffer().catch(handleFileError(file))
      if (!arraybuffer) return null

      const hash = await getHash(arraybuffer)
      if (!hash) {
        return null
      }

      fileHashes[file.name] = hash
    }

    return fileHashes
  }

  // Create a zip file with the selected files and sign the event
  const createZipFile = async (fileHashes: {
    [key: string]: string
  }): Promise<{ zip: JSZip; createSignature: string } | null> => {
    const zip = new JSZip()

    selectedFiles.forEach((file) => {
      zip.file(`files/${file.name}`, file)
    })

    const signers = users.filter((user) => user.role === UserRole.signer)
    const viewers = users.filter((user) => user.role === UserRole.viewer)

    setLoadingSpinnerDesc('Signing nostr event')

    const createSignature = await signEventForMetaFile(
      JSON.stringify({
        signers: signers.map((signer) => hexToNpub(signer.pubkey)),
        viewers: viewers.map((viewer) => hexToNpub(viewer.pubkey)),
        fileHashes
      }),
      nostrController,
      setIsLoading
    )

    if (!createSignature) return null

    try {
      return {
        zip,
        createSignature: JSON.stringify(createSignature, null, 2)
      }
    } catch (error) {
      return null
    }
  }

  // Add metadata and file hashes to the zip file
  const addMetaToZip = async (
    zip: JSZip,
    createSignature: string
  ): Promise<string | null> => {
    // create content for meta file
    const meta: Meta = {
      title,
      createSignature,
      docSignatures: {}
    }

    try {
      const stringifiedMeta = JSON.stringify(meta, null, 2)
      zip.file('meta.json', stringifiedMeta)

      const metaHash = await getHash(stringifiedMeta)
      if (!metaHash) return null

      const metaHashJson = {
        [usersPubkey!]: metaHash
      }

      zip.file('hashes.json', JSON.stringify(metaHashJson, null, 2))
      return metaHash
    } catch (err) {
      console.error(err)
      toast.error('An error occurred in converting meta json to string')
      return null
    }
  }

  // Handle errors during zip file generation
  const handleZipError = (err: any) => {
    console.log('Error in zip:>> ', err)
    setIsLoading(false)
    toast.error(err.message || 'Error occurred in generating zip file')
    return null
  }

  // Generate the zip file
  const generateZipFile = async (zip: JSZip): Promise<ArrayBuffer | null> => {
    setLoadingSpinnerDesc('Generating zip file')

    const arraybuffer = await zip
      .generateAsync({
        type: 'arraybuffer',
        compression: 'DEFLATE',
        compressionOptions: { level: 6 }
      })
      .catch(handleZipError)

    return arraybuffer
  }

  // Encrypt the zip file with the generated encryption key
  const encryptZipFile = async (
    arraybuffer: ArrayBuffer,
    encryptionKey: string
  ): Promise<ArrayBuffer> => {
    setLoadingSpinnerDesc('Encrypting zip file')
    return encryptArrayBuffer(arraybuffer, encryptionKey)
  }

  // create final zip file
  const createFinalZipFile = async (
    encryptedArrayBuffer: ArrayBuffer,
    encryptionKey: string
  ): Promise<File | null> => {
    // Get the current timestamp in seconds
    const unixNow = Math.floor(Date.now() / 1000)
    const blob = new Blob([encryptedArrayBuffer])
    // Create a File object with the Blob data
    const file = new File([blob], `compressed.sigit`, {
      type: 'application/sigit'
    })

    const firstSigner = users.filter((user) => user.role === UserRole.signer)[0]

    const keysFileContent = await generateKeysFile(
      [firstSigner.pubkey],
      encryptionKey
    )
    if (!keysFileContent) return null

    const zip = new JSZip()
    zip.file(`compressed.sigit`, file)
    zip.file('keys.json', keysFileContent)

    const arraybuffer = await zip
      .generateAsync({
        type: 'arraybuffer',
        compression: 'DEFLATE',
        compressionOptions: { level: 6 }
      })
      .catch(handleZipError)

    if (!arraybuffer) return null

    const finalZipFile = new File(
      [new Blob([arraybuffer])],
      `${unixNow}.sigit.zip`,
      {
        type: 'application/zip'
      }
    )

    return finalZipFile
  }

  // Handle file upload and further actions based on online/offline status
  const handleFileUpload = async (file: File, arrayBuffer: ArrayBuffer) => {
    if (await isOnline()) {
      const fileUrl = await uploadFile(file)

      if (!fileUrl) return

      await sendDMs(fileUrl)
      setIsLoading(false)

      navigate(appPrivateRoutes.sign, { state: { arrayBuffer } })
    } else {
      handleOffline(file, arrayBuffer)
    }
  }

  // Handle errors during file upload
  const handleUploadError = (err: any) => {
    console.log('Error in upload:>> ', err)
    setIsLoading(false)
    toast.error(err.message || 'Error occurred in uploading zip file')
    return null
  }

  // Upload the file to the storage and send DMs to signers/viewers
  const uploadFile = async (file: File): Promise<string | null> => {
    setIsLoading(true)
    setLoadingSpinnerDesc('Uploading zip file to file storage.')

    const fileUrl = await uploadToFileStorage(file, nostrController)
      .then((url) => {
        toast.success('zip file uploaded to file storage')
        return url
      })
      .catch(handleUploadError)

    return fileUrl
  }

  // Send DMs to signers and viewers with the file URL
  const sendDMs = async (fileUrl: string) => {
    setLoadingSpinnerDesc('Sending DM to signers/viewers')

    const signers = users.filter((user) => user.role === UserRole.signer)
    const viewers = users.filter((user) => user.role === UserRole.viewer)

    if (signers.length > 0) {
      await sendDM(
        fileUrl,
        signers[0].pubkey,
        nostrController,
        true,
        setAuthUrl
      )
    } else {
      for (const viewer of viewers) {
        await sendDM(fileUrl, viewer.pubkey, nostrController, false, setAuthUrl)
      }
    }
  }

  // Manage offline scenarios for signing or viewing the file
  const handleOffline = (file: File, arrayBuffer: ArrayBuffer) => {
    saveAs(file, 'request.sigit.zip')
    navigate(appPrivateRoutes.sign, { state: { arrayBuffer } })
  }

  const handleCreate = async () => {
    if (!validateInputs()) return

    setIsLoading(true)
    setLoadingSpinnerDesc('Generating hashes for files')

    const fileHashes = await generateFileHashes()
    if (!fileHashes) return

    const createZipResponse = await createZipFile(fileHashes)
    if (!createZipResponse) return

    const { zip, createSignature } = createZipResponse

    const metaHash = await addMetaToZip(zip, createSignature)
    if (!metaHash) return

    setLoadingSpinnerDesc('Generating zip file')

    const arraybuffer = await generateZipFile(zip)
    if (!arraybuffer) return

    const encryptionKey = await generateEncryptionKey()

    setLoadingSpinnerDesc('Encrypting zip file')
    const encryptedArrayBuffer = await encryptZipFile(
      arraybuffer,
      encryptionKey
    )

    const finalZipFile = await createFinalZipFile(
      encryptedArrayBuffer,
      encryptionKey
    )

    if (!finalZipFile) return

    return await handleFileUpload(finalZipFile, arraybuffer)
  }

  if (authUrl) {
    return (
      <iframe
        title="Nsecbunker auth"
        src={authUrl}
        width="100%"
        height="500px"
      />
    )
  }

  return (
    <>
      {isLoading && <LoadingSpinner desc={loadingSpinnerDesc} />}
      <Box className={styles.container}>
        <TextField
          label="Title"
          value={title}
          onChange={(e) => setTitle(e.target.value)}
          variant="outlined"
        />

        <Box>
          <MuiFileInput
            fullWidth
            multiple
            placeholder="Choose Files"
            value={selectedFiles}
            onChange={(value) => handleSelectFiles(value)}
          />

          {selectedFiles.length > 0 && (
            <ul>
              {selectedFiles.map((file, index) => (
                <li key={index}>
                  <Typography component="label">{file.name}</Typography>
                  <IconButton onClick={() => handleRemoveFile(file)}>
                    <Clear style={{ color: 'red' }} />{' '}
                  </IconButton>
                </li>
              ))}
            </ul>
          )}
        </Box>

        <Typography component="label" variant="h6">
          Add Counterparts
        </Typography>
        <Box className={styles.inputBlock}>
          <Box className={styles.inputBlock}>
            <TextField
              label="nip05 / npub"
              value={userInput}
              onChange={(e) => setUserInput(e.target.value)}
              helperText={error}
              error={!!error}
            />
            <FormControl fullWidth>
              <InputLabel id="select-role-label">Role</InputLabel>
              <Select
                labelId="select-role-label"
                id="demo-simple-select"
                value={userRole}
                label="Role"
                onChange={(e) => setUserRole(e.target.value as UserRole)}
              >
                <MenuItem value={UserRole.signer}>{UserRole.signer}</MenuItem>
                <MenuItem value={UserRole.viewer}>{UserRole.viewer}</MenuItem>
              </Select>
            </FormControl>

            <Box sx={{ mt: 1, display: 'flex', justifyContent: 'center' }}>
              <Button
                disabled={!userInput}
                onClick={handleAddUser}
                variant="contained"
              >
                Add
              </Button>
            </Box>
          </Box>
        </Box>
        <DisplayUser
          users={users}
          handleUserRoleChange={handleUserRoleChange}
          handleRemoveUser={handleRemoveUser}
          moveSigner={moveSigner}
        />
        <Box sx={{ mt: 1, display: 'flex', justifyContent: 'center' }}>
          <Button onClick={handleCreate} variant="contained">
            Create
          </Button>
        </Box>
      </Box>
    </>
  )
}

type DisplayUsersProps = {
  users: User[]
  handleUserRoleChange: (role: UserRole, pubkey: string) => void
  handleRemoveUser: (pubkey: string) => void
  moveSigner: (dragIndex: number, hoverIndex: number) => void
}

const DisplayUser = ({
  users,
  handleUserRoleChange,
  handleRemoveUser,
  moveSigner
}: DisplayUsersProps) => {
  const [metadata, setMetadata] = useState<{ [key: string]: ProfileMetadata }>(
    {}
  )

  useEffect(() => {
    users.forEach((user) => {
      if (!(user.pubkey in metadata)) {
        const metadataController = new MetadataController()

        const handleMetadataEvent = (event: Event) => {
          const metadataContent =
            metadataController.extractProfileMetadataContent(event)
          if (metadataContent)
            setMetadata((prev) => ({
              ...prev,
              [user.pubkey]: metadataContent
            }))
        }

        metadataController.on(user.pubkey, (kind: number, event: Event) => {
          if (kind === kinds.Metadata) {
            handleMetadataEvent(event)
          }
        })

        metadataController
          .findMetadata(user.pubkey)
          .then((metadataEvent) => {
            if (metadataEvent) handleMetadataEvent(metadataEvent)
          })
          .catch((err) => {
            console.error(
              `error occurred in finding metadata for: ${user.pubkey}`,
              err
            )
          })
      }
    })
  }, [users])

  return (
    <TableContainer component={Paper} elevation={3} sx={{ marginTop: '20px' }}>
      <Table>
        <TableHead>
          <TableRow>
            <TableCell className={styles.tableHeaderCell}>User</TableCell>
            <TableCell className={styles.tableHeaderCell}>Role</TableCell>
            <TableCell>Action</TableCell>
          </TableRow>
        </TableHead>
        <TableBody>
          <DndProvider backend={HTML5Backend}>
            {users
              .filter((user) => user.role === UserRole.signer)
              .map((user, index) => (
                <SignerRow
                  key={`signer-${index}`}
                  userMeta={metadata[user.pubkey]}
                  user={user}
                  index={index}
                  moveSigner={moveSigner}
                  handleUserRoleChange={handleUserRoleChange}
                  handleRemoveUser={handleRemoveUser}
                />
              ))}
          </DndProvider>
          {users
            .filter((user) => user.role === UserRole.viewer)
            .map((user, index) => {
              const userMeta = metadata[user.pubkey]
              return (
                <TableRow key={index}>
                  <TableCell className={styles.tableCell}>
                    <UserComponent
                      pubkey={user.pubkey}
                      name={
                        userMeta?.display_name ||
                        userMeta?.name ||
                        shorten(hexToNpub(user.pubkey))
                      }
                      image={userMeta?.picture}
                    />
                  </TableCell>
                  <TableCell className={styles.tableCell}>
                    <Select
                      fullWidth
                      value={user.role}
                      onChange={(e) =>
                        handleUserRoleChange(
                          e.target.value as UserRole,
                          user.pubkey
                        )
                      }
                    >
                      <MenuItem value={UserRole.signer}>
                        {UserRole.signer}
                      </MenuItem>
                      <MenuItem value={UserRole.viewer}>
                        {UserRole.viewer}
                      </MenuItem>
                    </Select>
                  </TableCell>
                  <TableCell>
                    <Tooltip title="Remove User" arrow>
                      <IconButton onClick={() => handleRemoveUser(user.pubkey)}>
                        <Clear style={{ color: 'red' }} />
                      </IconButton>
                    </Tooltip>
                  </TableCell>
                </TableRow>
              )
            })}
        </TableBody>
      </Table>
    </TableContainer>
  )
}

interface DragItem {
  index: number
  id: string
  type: string
}

type SignerRowProps = {
  userMeta: ProfileMetadata
  user: User
  index: number
  moveSigner: (dragIndex: number, hoverIndex: number) => void
  handleUserRoleChange: (role: UserRole, pubkey: string) => void
  handleRemoveUser: (pubkey: string) => void
}

const SignerRow = ({
  userMeta,
  user,
  index,
  moveSigner,
  handleUserRoleChange,
  handleRemoveUser
}: SignerRowProps) => {
  const ref = useRef<HTMLTableRowElement>(null)

  const [{ handlerId }, drop] = useDrop<
    DragItem,
    void,
    { handlerId: Identifier | null }
  >({
    accept: 'row',
    collect(monitor) {
      return {
        handlerId: monitor.getHandlerId()
      }
    },
    hover(item: DragItem, monitor) {
      if (!ref.current) {
        return
      }
      const dragIndex = item.index
      const hoverIndex = index

      // Don't replace items with themselves
      if (dragIndex === hoverIndex) {
        return
      }

      // Determine rectangle on screen
      const hoverBoundingRect = ref.current?.getBoundingClientRect()

      // Get vertical middle
      const hoverMiddleY =
        (hoverBoundingRect.bottom - hoverBoundingRect.top) / 2

      // Determine mouse position
      const clientOffset = monitor.getClientOffset()

      // Get pixels to the top
      const hoverClientY = (clientOffset as XYCoord).y - hoverBoundingRect.top

      // Only perform the move when the mouse has crossed half of the items height
      // When dragging downwards, only move when the cursor is below 50%
      // When dragging upwards, only move when the cursor is above 50%

      // Dragging downwards
      if (dragIndex < hoverIndex && hoverClientY < hoverMiddleY) {
        return
      }

      // Dragging upwards
      if (dragIndex > hoverIndex && hoverClientY > hoverMiddleY) {
        return
      }

      // Time to actually perform the action
      moveSigner(dragIndex, hoverIndex)

      // Note: we're mutating the monitor item here!
      // Generally it's better to avoid mutations,
      // but it's good here for the sake of performance
      // to avoid expensive index searches.
      item.index = hoverIndex
    }
  })

  const [{ isDragging }, drag] = useDrag({
    type: 'row',
    item: () => {
      return { id: user.pubkey, index }
    },
    collect: (monitor: any) => ({
      isDragging: monitor.isDragging()
    })
  })

  const opacity = isDragging ? 0 : 1
  drag(drop(ref))

  return (
    <TableRow
      sx={{ cursor: 'move', opacity }}
      data-handler-id={handlerId}
      ref={ref}
    >
      <TableCell
        className={styles.tableCell}
        sx={{ display: 'flex', alignItems: 'center', gap: '10px' }}
      >
        <DragHandle />
        <UserComponent
          pubkey={user.pubkey}
          name={
            userMeta?.display_name ||
            userMeta?.name ||
            shorten(hexToNpub(user.pubkey))
          }
          image={userMeta?.picture}
        />
      </TableCell>
      <TableCell className={styles.tableCell}>
        <Select
          fullWidth
          value={user.role}
          onChange={(e) =>
            handleUserRoleChange(e.target.value as UserRole, user.pubkey)
          }
        >
          <MenuItem value={UserRole.signer}>{UserRole.signer}</MenuItem>
          <MenuItem value={UserRole.viewer}>{UserRole.viewer}</MenuItem>
        </Select>
      </TableCell>
      <TableCell>
        <Tooltip title="Remove User" arrow>
          <IconButton onClick={() => handleRemoveUser(user.pubkey)}>
            <Clear style={{ color: 'red' }} />
          </IconButton>
        </Tooltip>
      </TableCell>
    </TableRow>
  )
}