{-# LANGUAGE OverloadedStrings #-}

import C.Zcash (RawData(..), rustWrapperIsUA)
import qualified Data.ByteString as BS
import qualified Data.Text.Encoding as E
import Data.Word
import Test.Hspec
import Zcash

main :: IO ()
main = do
  hspec $ do
    describe "F4Jumble" $ do
      it "jumble a string" $ do
        let input =
              [ 0x5d
              , 0x7a
              , 0x8f
              , 0x73
              , 0x9a
              , 0x2d
              , 0x9e
              , 0x94
              , 0x5b
              , 0x0c
              , 0xe1
              , 0x52
              , 0xa8
              , 0x04
              , 0x9e
              , 0x29
              , 0x4c
              , 0x4d
              , 0x6e
              , 0x66
              , 0xb1
              , 0x64
              , 0x93
              , 0x9d
              , 0xaf
              , 0xfa
              , 0x2e
              , 0xf6
              , 0xee
              , 0x69
              , 0x21
              , 0x48
              , 0x1c
              , 0xdd
              , 0x86
              , 0xb3
              , 0xcc
              , 0x43
              , 0x18
              , 0xd9
              , 0x61
              , 0x4f
              , 0xc8
              , 0x20
              , 0x90
              , 0x5d
              , 0x04
              , 0x2b
              ] :: [Word8]
        let out =
              [ 0x03
              , 0x04
              , 0xd0
              , 0x29
              , 0x14
              , 0x1b
              , 0x99
              , 0x5d
              , 0xa5
              , 0x38
              , 0x7c
              , 0x12
              , 0x59
              , 0x70
              , 0x67
              , 0x35
              , 0x04
              , 0xd6
              , 0xc7
              , 0x64
              , 0xd9
              , 0x1e
              , 0xa6
              , 0xc0
              , 0x82
              , 0x12
              , 0x37
              , 0x70
              , 0xc7
              , 0x13
              , 0x9c
              , 0xcd
              , 0x88
              , 0xee
              , 0x27
              , 0x36
              , 0x8c
              , 0xd0
              , 0xc0
              , 0x92
              , 0x1a
              , 0x04
              , 0x44
              , 0xc8
              , 0xe5
              , 0x85
              , 0x8d
              , 0x22
              ] :: [Word8]
        BS.pack out `shouldBe` f4Jumble (BS.pack input)
      it "unjumble a string" $ do
        let input =
              [ 0x5d
              , 0x7a
              , 0x8f
              , 0x73
              , 0x9a
              , 0x2d
              , 0x9e
              , 0x94
              , 0x5b
              , 0x0c
              , 0xe1
              , 0x52
              , 0xa8
              , 0x04
              , 0x9e
              , 0x29
              , 0x4c
              , 0x4d
              , 0x6e
              , 0x66
              , 0xb1
              , 0x64
              , 0x93
              , 0x9d
              , 0xaf
              , 0xfa
              , 0x2e
              , 0xf6
              , 0xee
              , 0x69
              , 0x21
              , 0x48
              , 0x1c
              , 0xdd
              , 0x86
              , 0xb3
              , 0xcc
              , 0x43
              , 0x18
              , 0xd9
              , 0x61
              , 0x4f
              , 0xc8
              , 0x20
              , 0x90
              , 0x5d
              , 0x04
              , 0x2b
              ] :: [Word8]
        let out =
              [ 0x03
              , 0x04
              , 0xd0
              , 0x29
              , 0x14
              , 0x1b
              , 0x99
              , 0x5d
              , 0xa5
              , 0x38
              , 0x7c
              , 0x12
              , 0x59
              , 0x70
              , 0x67
              , 0x35
              , 0x04
              , 0xd6
              , 0xc7
              , 0x64
              , 0xd9
              , 0x1e
              , 0xa6
              , 0xc0
              , 0x82
              , 0x12
              , 0x37
              , 0x70
              , 0xc7
              , 0x13
              , 0x9c
              , 0xcd
              , 0x88
              , 0xee
              , 0x27
              , 0x36
              , 0x8c
              , 0xd0
              , 0xc0
              , 0x92
              , 0x1a
              , 0x04
              , 0x44
              , 0xc8
              , 0xe5
              , 0x85
              , 0x8d
              , 0x22
              ] :: [Word8]
        f4UnJumble (BS.pack out) `shouldBe` BS.pack input
    describe "Decode UVK from YWallet" $ do
      let uvk =
            "uview1u833rp8yykd7h4druwht6xp6k8krle45fx8hqsw6vzw63n24atxpcatws82z092kryazuu6d7rayyut8m36wm4wpjy2z8r9hj48fx5pf49gw4sjrq8503qpz3vqj5hg0vg9vsqeasg5qjuyh94uyfm7v76udqcm2m0wfc25hcyqswcn56xxduq3xkgxkr0l73cjy88fdvf90eq5fda9g6x7yv7d0uckpevxg6540wc76xrc4axxvlt03ptaa2a0rektglmdy68656f3uzcdgqqyu0t7wk5cvwghyyvgqc0rp3vgu5ye4nd236ml57rjh083a2755qemf6dk6pw0qrnfm7246s8eg2hhzkzpf9h73chhng7xhmyem2sjh8rs2m9nhfcslsgenm"
      let res = decodeBech32 uvk
      let rawBytes = f4UnJumble $ bytes res
      it "decodes Bech32" $ do
        print $ hrp res
        print $ BS.length $ bytes res
        hrp res `shouldBe` "uview"
      it "unjumble result" $ do
        BS.takeEnd 16 rawBytes `shouldBe` "uview00000000000"
    describe "Unified address" $ do
      it "succeeds with correct UA" $ do
        let ua =
              "u1salpdyefywvsg2dlmxg9589yznh0h9v6qjr478k80amtkqkws5pr408lxt2953dpprvu06mahxt99cv65fgsm7sw8hlchplfg5pl89ur"
        isValidUnifiedAddress ua `shouldBe` True
      it "fails with incorrect UA" $ do
        let ua =
              "u1salpdyefbreakingtheaddressh0h9v6qjr478k80amtkqkws5pr408lxt2953dpprvu06mahxt99cv65fgsm7sw8hlchplfg5pl89ur"
        isValidUnifiedAddress ua `shouldBe` False