class WalkingTask extends React.Component {
  constructor(props) {
      super(props);
      this.state = {
          rightSide: "instruction",
          rightSideHeader: "Your instruction at the pulsing ball",
          imagePathOutcome: '/static/img/coin_questionmark.png',
          imagePathGuess: '/static/img/coin.png',
          winningSide: "",
          payment: [],
          keyNum: 0,
          modal: false,
      }
      this.testCurrentNode = this.testCurrentNode.bind(this);
      this.ballInstructions = this.ballInstructions.bind(this);
      this.submitCoin = this.submitCoin.bind(this);
      this.nextBall = this.nextBall.bind(this);
      this.winningSideUpdate = this.winningSideUpdate.bind(this);
      this.calculateEarnings = this.calculateEarnings.bind(this);
      this.openModal = this.openModal.bind(this);
      this.closeModal = this.closeModal.bind(this);
      this.arrayEquals = this.arrayEquals.bind(this);
  }

  componentDidMount() {
    // Disable ball clicks
    var balls = document.getElementsByClassName("ball")
    for(var i=0;i<balls.length;i++) {
        balls[i].disabled = true;
    }

    if (this.props.practiceRound) {
        document.getElementById("accumulatedEarningsDiv").style.visibility = "hidden";
    }

    // Determine component view
    if (this.props.continuationDec[this.props.roundNumber] == "Stop" &&
        this.props.roundNumber !== 5
    ) {
        this.openModal();
        this.setState({rightSide:"coin"})
    }
    
    if (this.props.continuationDec.slice(0,this.props.roundNumber).includes("Stop")
    ) { 
        document.getElementById("risksToTake").classList.remove("d-none")
        if (this.props.practiceRound) {
            $('#moreRisks').collapse('show')
            document.getElementById("task").classList.add("d-none")
        } else {
            document.getElementById("riskInfo").classList.add("d-none")
        }
        document.getElementById("accumulatedEarningsDiv").style.visibility = "hidden"

        this.setState({
            rightSideHeader: "Your instruction"
        })
    }
    
    if (this.props.roundNumber == 5) {
        // Disable further input
        document.getElementById("submitInstructionBtn").disabled = true
        $("input[name=ballDecision]").prop('disabled', true);

        document.getElementById("input-widget").style.opacity = .4
        document.getElementById("input-widget").style.cursor = "not-allowed"
        
        
        // Continue to next page
        if (this.props.continuationDec.slice(0,this.props.roundNumber).includes("Stop")
            && !this.props.practiceRound) {
                document.getElementById("confirm_btn").classList.remove("d-none")
                setTimeout( () => {
                    document.getElementById("confirm_btn").disabled = false;
                }, 1000)
            } else if (!this.props.practiceRound){
                this.openModal();
                // Hide risk info
                document.getElementById("riskInfo").classList.add("d-none")

            } else {
                document.getElementById("confirm_btn").classList.remove("d-none")
                document.getElementById("remainingRounds").classList.add("d-none")
                document.getElementById("nextScreen").classList.remove("d-none")
                document.getElementById("task").classList.add("d-none")

                if ( ! this.props.continuationDec.slice(0,this.props.roundNumber).includes("Stop")){
                    // Only stop in last round
                    // Include information about five risks for all Walking Tasks
                    document.getElementById("fiveRisks").classList.remove("d-none")
                }
            }
    } else if (this.props.type == "coin"){
        if ((this.props.continuationDec.slice(0,5).includes("Stop"))) {
            this.setState({
                rightSide: "coin",
                imagePathGuess: this.props.imagePathGuess,
                imagePathOutcome: '/static/img/coin_questionmark.png',
            }, () => {
                document.getElementById("placeholderFooter").classList.add("d-none");
                document.getElementById("noFeedback").classList.remove("d-none");
                // Grey background for selection section
                document.getElementById("pickWinningSide").classList.remove("alert-warning");
                document.getElementById("pickWinningSide").classList.add("alert-secondary");
    
                var ele = document.getElementsByName("winningSideInput");
            
                for(var i=0;i<ele.length;i++) {
                    ele[i].disabled = true;
                }
            })
        }
        else 
        {
            this.setState({
                rightSide: "coin",
                imagePathGuess: this.props.imagePathGuess,
                imagePathOutcome: this.props.imagePathOutcome,
            },
            () => {
                
                if (this.props.imagePathGuess == this.props.imagePathOutcome) {
                    document.getElementById("coinTossWon").classList.remove("d-none");
                    document.getElementById("moveUp").classList.remove("d-none");
                    document.getElementById("placeholderFooter").classList.add("d-none");
                } 
                else {
                    document.getElementById("coinTossLost").classList.remove("d-none");
                    document.getElementById("moveDown").classList.remove("d-none");
                    document.getElementById("placeholderFooter").classList.add("d-none");
                }
                
                // Grey background for selection section
                document.getElementById("pickWinningSide").classList.remove("alert-warning");
                document.getElementById("pickWinningSide").classList.add("alert-secondary");

                var ele = document.getElementsByName("winningSideInput");
                for(var i=0;i<ele.length;i++) {
                    ele[i].disabled = true;
                }
            })
            
            //setTimeout(this.nextBall, 2000)
        }        
    } else if (this.props.type == "nodeChoice"){
        // Node choice has been submitted
        this.setState({
            rightSide: "coin"
        }, () => {
            document.getElementById("pickWinningSide").classList.remove("d-none")})
    } else {
        this.testCurrentNode()
    }

    this.calculateEarnings();

    }

openModal() {
    this.setState({ 
        modal: true,
        keyNum: 4
    })
    document.getElementById("lineSVG").id="lineSVG_main"
} 

calculateEarnings = () => {
    const indArray = [
        [0],
        [1, 2],
        [3, 4, 5],
        [6, 7, 8, 9],
        [10, 11, 12, 13, 14],
        [15, 16, 17, 18, 19, 20]
        ]

    let upTick = js_vars.upTick
    let payment = []
    let j = 0

    for (let i in indArray) {
        for (let k in indArray[i]) {
            payment[j] = upTick * i - 2 * k * upTick;
            j++ 
        }
    }
    this.setState({payment})

    return payment
    }

// Submit coin toss
submitCoin() {
    liveSend({
        'type': "coin", 
        'value': this.state.winningSide
        })
}

nextBall() {
    //document.getElementById("nextCoinBtn").classList.add("d-none")
    document.getElementById("moveUp").classList.add("d-none")
    document.getElementById("moveDown").classList.add("d-none")
    document.getElementById("noFeedback").classList.add("d-none")
    document.getElementById("placeholderFooter").classList.remove("d-none")

    this.setState({
        rightSide: "instruction",
        imagePathOutcome: '/static/img/coin_questionmark.png',
        imagePathGuess: '/static/img/coin.png',
        winningSide: "",
    }, this.testCurrentNode)
    
}

arrayEquals(a, b) {
    return Array.isArray(a) &&
        Array.isArray(b) &&
        a.length === b.length &&
        a.every((val, index) => val === b[index]);
}

// Test Current Button
testCurrentNode() {
    const drawn = this.props.drawn || [0]
    const roundNumber = this.props.roundNumber || [0]
    const currentNode = drawn[roundNumber]
    const stoppingStrategy = js_vars.stopStrategy.continuation
    const pdStop = js_vars.stopStrategy.pdStopDecision
    let radioValue =""

    if (stoppingStrategy[currentNode] == 100) {
        // White Ball
        radioValue = "Continue"
    } else if  (stoppingStrategy[currentNode] == 0){
        // Black Ball
        radioValue = "Stop"
    } else if (stoppingStrategy[currentNode] == -100) {
        // White-black-gradient
        const realizedPath = drawn.slice(0,roundNumber+1)
        const relevantPaths = js_vars.allPaths[currentNode]

        let pathName = ""
        for (const [key, value] of Object.entries(relevantPaths)) {
            if (this.arrayEquals(value[1], realizedPath)) {
                pathName = value[0]
                break;
            }
          }

        radioValue = pdStop[pathName]
        
    }

    $("input[name=ballDecision][value=" + radioValue + "]").prop('checked', true);
}


ballInstructions () {
    // Submit changed choice
    let decision = document.querySelector('input[name="ballDecision"]:checked').value;

    if (decision == "Stop") {
        liveSend({
            'type': "nodeChoice", 
            'value': "Stop",
            'ballContinue': 0,
            })
    } else if (decision == "Continue") {
        liveSend({
            'type': "nodeChoice", 
            'value': "Continue",
            'ballContinue': 100,
            })
    }

    this.setState({
        rightSide: "coin"
    }, () => {
        document.getElementById("pickWinningSide").classList.remove("d-none")
    })

  }

  winningSideUpdate(chosenSide) {
    // Disable winning side buttons
    var ele = document.getElementsByName("winningSideInput");
    for(var i=0;i<ele.length;i++)
        ele[i].disabled = true;
        
    if (chosenSide == "H") {
        this.setState({
            winningSide: chosenSide,
            imagePathGuess: '/static/img/coin_h.png'
        },() => {this.submitCoin()})
    } else if (chosenSide == "T") {
        this.setState({
            winningSide: chosenSide,
            imagePathGuess: '/static/img/coin_t.png'
        }, () => {this.submitCoin()})
    }

    // Grey background for selection section
    document.getElementById("pickWinningSide").classList.remove("alert-warning");
    document.getElementById("pickWinningSide").classList.add("alert-secondary");

    var ele = document.getElementsByName("winningSideInput");
    for(var i=0;i<ele.length;i++) {
        ele[i].disabled = true;
    } 
}

    voidFunction() {
        return
    }

    closeModal(){
        document.getElementById("risksToTake").classList.remove("d-none")
        if (this.props.practiceRound) {
            $('#moreRisks').collapse('show')
            document.getElementById("task").classList.add("d-none")

            // Disable radio buttons
            for (let radio of document.querySelectorAll('input[type=radio]')) {
                radio.disabled = true;
            }
            document.getElementById("pickWinningSide").classList.remove("alert-warning")
            document.getElementById("pickWinningSide").classList.add("alert-secondary")
            
            setTimeout( () => {
                for (let radio of document.querySelectorAll('input[type=radio]')) {
                    radio.disabled = false;
                }
                document.getElementById("pickWinningSide").classList.remove("alert-secondary")
                document.getElementById("pickWinningSide").classList.add("alert-warning")
            }, 5000)

        } else {
            document.getElementById("riskInfo").classList.add("d-none")
        }
        document.getElementById("accumulatedEarningsDiv").style.visibility = "hidden"
        this.setState({ 
            //keyNum: 1,
            modal:false}) 
    }

  render() {
    // Find stop decision
    let submittedContinuation = this.props.continuationDec.slice(0,this.props.roundNumber + 1)
    let submittedStop = submittedContinuation.findIndex(element => element == "Stop")
    
    // Calculate accumulated earning to show
    let payment = 0
    if (submittedStop >= 0) {
        payment = parseFloat(this.state.payment[this.props.drawn[submittedStop]])
    } else {
        payment = parseFloat(this.state.payment[this.props.drawn[this.props.roundNumber]])
    }

    // Write payment to main page
    if (! this.props.practiceRound) {
        document.getElementById("finPay1").innerHTML = payment.toFixed(2)
    }

    return(
        <>
        <div className="row">
            {/* Left column showing stopping strategy */}
            <div className="col-md-6 pl-3 pr-1">
                <div className="card" style={{minHeight: "400px"}}>
                    <h5 className="card-header  bg-dark text-white text-center">Your decoloring</h5>
                    <div id="accumulatedEarningsDiv" className="text-center">
                        Your current accumulated earnings are £{payment.toFixed(2)}.
                    </div>
                    <div className="card-body" id="treeStructure" style={{
                            height: "250px",
                            width: "300px",
                            position: "relative",
                            margin: "0 auto"
                    }}>
                        { (! this.state.modal) ?
                        <>
                            <svg id="lineSVG" 
                                style={{ 
                                    position: "absolute",
                                    left: "0",
                                    top:"0",
                                    height: "100%",
                                    width: "100%"
                            }}/>
                            <DecisionTree
                                upTick = {js_vars.upTick} 
                                key = {this.state.keyNum}
                                isClicked={js_vars.stopStrategy.isClicked} 
                                drawn={this.props.drawn || [0]}
                                continuation={js_vars.stopStrategy.continuation}
                                initialColor = {this.props.initialColor} 
                                currentRound = {this.props.roundNumber}
                                continuationDec = {this.props.continuationDec}
                                ClickHandler = {this.voidFunction}
                                mouseEnter = {this.voidFunction}
                                mouseLeave = {this.voidFunction}
                                pulse = {true}
                            />
                        </>
                        :
                        <>
                        </>
                        }
                    </div>
                    
                    {/* Footer */}
                    <div id="placeholderFooter" 
                        style={{
                            height: "60px",
                            marginBottom:"0px",
                        }}>
                    </div>

                    <div id="noFeedback" 
                        className="card-footer alert 
                                    d-inline-flex justify-content-center align-items-center d-none" 
                        style={{
                            height: "60px",
                            marginBottom:"0px",
                            }}>

                        <div className="col-5 text-center d-inline-flex justify-content-center align-items-center ">
                            <button type="button" 
                                className="btn btn-primary px-3" 
                                onClick={this.nextBall}>
                                    Next instruction
                            </button>
                        </div>
                    </div>

                    <div id="moveUp" 
                        className="card-footer alert alert-success 
                                    d-inline-flex justify-content-center align-items-center d-none" 
                        style={{
                            height: "60px",
                            marginBottom:"0px",
                            }}>

                        { !(this.props.continuationDec.slice(0,5).includes("Stop")) ? 
                            <div className="col-5 text-center d-inline-flex justify-content-center align-items-center" >
                                You moved up.
                            </div>
                        : <></>}

                        <div className="col-5 text-center d-inline-flex justify-content-center align-items-center ">
                            <button type="button" 
                                className="btn btn-primary px-3" 
                                onClick={this.nextBall}>
                                    Next instruction
                            </button>
                        </div>
                    </div>

                    <div id="moveDown" 
                        className="card-footer alert alert-danger 
                                    d-inline-flex justify-content-center align-items-center d-none" 
                        style={{
                            height: "60px",
                            marginBottom:"0px",
                            }}>

                        { !(this.props.continuationDec.slice(0,5).includes("Stop")) ?
                        <div className="col-5 text-center d-inline-flex justify-content-center align-items-center" >
                            You moved down.
                        </div>
                        : <></>}

                        <div className="col-5 text-center d-inline-flex justify-content-center align-items-center ">
                            <button type="button" 
                                className="btn btn-primary px-3" 
                                onClick={this.nextBall}>
                                    Next instruction
                            </button>
                        </div>
                    </div>
                </div>
            </div>
            {/* Right column */}
            <div className="col-md-6 pr-3 pl-1">
                {this.state.rightSide == "instruction" ? 
                
                <div className="card" id="input-widget" style={{
                    minHeight: "400px",
                    }}>
                        <h5 className="card-header  bg-dark text-white text-center">{this.state.rightSideHeader}</h5>
                        <div className="card-body align-items-center d-flex justify-content-center" style={{
                            height: "250px",
                            width: "90%",
                            position: "relative",
                            margin: "0 auto"
                        }}>
                        <div id="nodeChoiceChange" style={{width:"100%"}}>
                            <table className="center table table-sm table-light table-borderless" style={{width:"100%", textAlign: "center"}}>
                            <thead>
                            <tr>
                                <th colSpan="2" style={{fontWeight:"normal"}}>
                                    At this ball, please
                                </th>
                            </tr>
                            <tr>
                                <th style={{width: "50%"}}> Stop</th>
                                <th style={{width: "50%"}}> Continue</th>
                            </tr>
                            </thead>
                            <tbody>
                            <tr>
                                <td>
                                <input 
                                    className="form-check-input largerRadio" 
                                    type="radio" 
                                    name="ballDecision" 
                                    value="Stop"
                                /> 
                                </td>
                                <td>
                                <input 
                                    className="form-check-input largerRadio" 
                                    type="radio" 
                                    name="ballDecision" 
                                    value="Continue" 
                                /> 
                                </td>
                            </tr>
                            <tr>
                                <td colSpan="2">taking risks.</td>
                            </tr>
                            </tbody>
                            </table>
                            <br></br>
                            <div className="col">
                                <button type="button" id="submitInstructionBtn" className="btn btn-primary btn-block" 
                                    onClick={this.ballInstructions}>Confirm instruction</button> 
                            </div>
                        </div>

                        </div>
                    </div>
                :
                (this.state.rightSide == "coin") ?
                    <CoinToss
                        imagePathGuess={this.state.imagePathGuess}
                        imagePathOutcome={this.state.imagePathOutcome}
                        winningSideUpdate = {this.winningSideUpdate}
                    />
                    :
                    <></>
                }

            </div>
        </div>

        <BonusModal 
            key = {this.state.keyNum} 
            show = {this.state.modal}
            keyNum = {this.state.keyNum}
            payment = {payment}
            endowment = {this.props.endowment || "£1.50"}
            drawn={this.props.drawn || [0]}
            currentRound = {this.props.roundNumber}
            continuationDec = {this.props.continuationDec}
            closeModal = {this.closeModal}
            practiceRound = {this.props.practiceRound}
        />
        </>
    )
  }
}


class BonusModal extends React.Component {
    constructor(props) {
        super(props);
        this.state = {
            keyForTriangle: 0,
        }
    }

    componentDidMount() {
        this.setState({keyForTriangle: 5})
    }

    render() {
        if(!this.props.show){
            return null;
        } else {
            $('#staticBackdrop').modal("show")
        }
      return (
      <>
        <div className="modal" id="staticBackdrop" 
         data-bs-backdrop="static" 
         data-bs-keyboard="false" 
         tabIndex="-1" 
         aria-labelledby="staticBackdropLabel"
         aria-hidden="true">
         <div className="modal-dialog modal-lg modal-dialog-centered">
            <div className="modal-content">
                <div className="modal-header">
                { this.props.practiceRound ? 
                    <h5 className="modal-title" id="bonusModalLongTitle">Your bonus for a Walking Task</h5>
                :
                  <h5 className="modal-title" id="bonusModalLongTitle">Your bonus for Walking Task {js_vars.taskNumber}</h5>
                }
                </div>
              <div className="modal-body">
                <div className="row">
                {this.props.practiceRound ? 
                    <div className="col">
                        If a Walking Task is your bonus task, your bonus equals the accumulated
                        earnings at the bonus ball, plus your endowment.
                    </div>    
                    :    
                    <div className="col">
                        Your bonus equals the accumulated
                        earnings of £{this.props.payment.toFixed(2)} plus 
                        the endowment of {this.props.endowment} if Walking Task {js_vars.taskNumber} is 
                        your bonus task.
                    </div>
                }
                </div>

                <div className="row mb-3">
                    <div className="col-md-6 offset-md-3 pl-3 pr-1">
                        <div className="card" style={{minHeight: "400px"}}>
                            <h5 className="card-header  bg-dark text-white text-center">Your decoloring</h5>
                            <div className="card-body" id="cardBodyModal" style={{
                                    height: "250px",
                                    width: "300px",
                                    position: "relative",
                                    margin: "0 auto"
                            }}>
                                <svg id="lineSVG" 
                                    style={{ 
                                        position: "absolute",
                                        left: "0",
                                        top:"0",
                                        height: "100%",
                                        width: "100%"
                                }}/>
                                <DecisionTree
                                    upTick = {js_vars.upTick} 
                                    key = {this.state.keyForTriangle}
                                    isClicked={js_vars.stopStrategy.isClicked} 
                                    drawn={this.props.drawn || [0]}
                                    continuation={js_vars.stopStrategy.continuation}
                                    currentRound = {this.props.currentRound}
                                    continuationDec = {this.props.continuationDec}
                                    pulse = {false}
                                    initialColor = {this.props.initialColor} 
                                />

                            </div>

                        </div>
                    </div>
                </div>
              </div>
              <div className="modal-footer justify-content-center">
              { this.props.currentRound < 5 ? 
                <button type="button" className="btn btn-primary" 
                    data-bs-dismiss="modal"
                    onClick = {this.props.closeModal}
                    >Okay
                </button>
                :
                <button className="btn btn-primary" 
                    >
                        {js_vars.taskNumber == 10 ?
                        <>
                            Continue
                        </>
                        :
                        <>
                            Next task
                        </>
                    }
                </button>
                }
              </div>
            </div>
          </div>
        </div>
      </>
    )}
  }