Wsrtjtyk

import React, { Component } from "react";

import { connect } from "react-redux";

import { firestoreConnect, isLoaded } from "react-redux-firebase";

import { compose } from "redux";

import { Grid } from "semantic-ui-react";

import BudgetHeader from "../BudgetHeader/BudgetHeader";

import BudgetList from "../BudgetList/BudgetList";

import BudgetNav from "../BudgetNav/BudgetNav";

import { updateBudgetData, compareChooser } from "../budgetActions";

import LoadingComponent from "../../../app/layout/LoadingComponent";



const mapState = state => ({

  users: state.firestore.ordered.users

});



const actions = {

  updateBudgetData,

  compareChooser

};



class BudgetDashboard extends Component {

  updateBudgetData = (newData, fieldToUpdate) => {

    this.props.updateBudgetData(newData, fieldToUpdate);

  };



  onCompareChange = (event, data) => {

    if (data.checked) {

      this.props.compareChooser("net");

    } else {

      this.props.compareChooser("gross");

    }

  };



  render() {

    const { users } = this.props;

    if (!isLoaded(users)) return <LoadingComponent inverted={true} />;

    return (

      <Grid>

        <Grid.Column width={16}>

          <BudgetNav />

          <BudgetHeader

            general={users[0].budget[1]}

            onCompareChange={this.onCompareChange}

          />

          <BudgetList

            className="standardSeparation"

            data={users[0].budget[0]}

            unallocatedBudget={users[0].budget[1].unallocatedBudget}

            unallocatedSpending={users[0].budget[1].unallocatedSpent}

            updateData={this.updateBudgetData}

          />

        </Grid.Column>

      </Grid>

    );

  }

}



export default compose(

  connect(

    mapState,

    actions

  ),

  firestoreConnect([

    {

      collection: "users",

      doc: "asdfasdfF21871d",

      subcollections: [{ collection: "budget" }]

    }

  ])

)(BudgetDashboard);

export const updateBudgetData = (inputData, fieldToUpdate) => {

  return async (dispatch, getState, { getFirebase, getFirestore }) => {

    try {

      const firestore = getFirestore();



      // Get all firestore data nodes that will be used for updating after data change

      const userBudget = firestore

        .collection("users")

        .doc("asdfasdfas8128fsa")

        .collection("budget");

      const userDataRef = userBudget.doc("data");

      const userGeneralRef = userBudget.doc("general");



      // variables to be used in generating updated maps

      var grossIncome = 0;

      var netIncome = 0;

      var userDataMap = {};

      var userGeneralMap = {};



      // get grid data

      await userDataRef

        .get()

        .then(function(doc) {

          if (doc.exists) {

            userDataMap = doc.data();

          } else {

            console.log("No user data");

          }

        })

        .catch(function(error) {

          console.log("User Data Document Get Error: ", error);

        });

      // get header data

      await userGeneralRef

        .get()

        .then(function(doc) {

          if (doc.exists) {

            const data = doc.data();

            userGeneralMap = data;

            grossIncome = data.grossIncome;

            netIncome = data.netIncome;

          } else {

            console.log("No general data");

          }

        })

        .catch(function(error) {

          console.log("User General Document Get Error: ", error);

        });



      // generate new data and general document mappings

      const updatedMaps = generateUpdatedMaps(

        userDataMap,

        userGeneralMap,

        grossIncome,

        netIncome,

        inputData,

        fieldToUpdate

      );

      const newUserDataMap = updatedMaps.newDataMap;

      const newUserGeneralMap = updatedMaps.newGeneralMap;



      // update firestore data with new mappings

      userDataRef.set(newUserDataMap);

      userGeneralRef.set(newUserGeneralMap);



      toastr.success("Success!", "Budget data has been updated");

    } catch (error) {

      toastr.error("Oops", "Something went wrong");

    }

  };

};