ledgerrb/public/app/js/controllers/DashboardCtrl.js

app.controller( 'DashboardCtrl',
		[ '$scope', '$filter', 'API',
		  function ( $scope, $filter, API ) {
		      $scope.xFunction = function () {
			  return function ( d ) {
			      return d.account;
			  };
		      };
		      $scope.yFunction = function () {
			  return function ( d ) {
			      return d.amount;
			  };
		      };
		      $scope.toolTipContentFunction = function () {
			  return function ( key, x, y, e, graph ) {
			      var details = $scope.balance.details[ key ];
			      return '<md-content><h3>' + key + '</h3>' + '<table>' + _( details ).map( function ( transaction ) {
				  return '<tr><td>' + transaction.date + '</td><td>' + transaction.payee + '</td><td style="text-align: right">' + $filter( 'number' )( transaction.amount, 2 ) + ' ' + transaction.currency + '</td></tr>';
			      } ).join( '' ) + '<tr><th></th><th>Total :</th><th>' + x + ' €</th></tr>' + '</table></md-content>';
			  };
		      };

		      // compute an account's score: from 1 (good) to 10 (bad), 0 is neutral/undecided
		      var score_account = function ( account ) {
			  if ( account.match( /^Income/ ) ) {
			      return -10;
			  } else if ( account.match( /^Expenses:(courses|Hang)$/ ) ) {
			      return 1;
			  } else if ( account.match( /^Expenses:Home/ ) ) {
			      return 1;
			  } else if ( account.match( /^Expenses:Health/ ) ) {
			      return 1;
			  } else if ( account.match( /^Expenses:Car/ ) ) {
			      return 4;
			  } else if ( account.match( /^Expenses:(Food|Transport)/ ) ) {
			      return 5;
			  } else if ( account.match( /^Expenses:(Shopping|Leisure)/ ) ) {
			      return 9;
			  } else if ( account.match( /^Expenses:Gadgets/ ) ) {
			      return 10;
			  } else if ( account.match( /^Liabilities/ ) ) {
			      return 0;
			  } else if ( account.match( /^Assets/ ) ) {
			      return -100;
			  } else {
			      return 0;
			  }
		      };

		      $scope.coloring_score = function ( score ) {
			  var adjusted_score = score;
			  var color_scale = [ '#99f', '#0f0', '#3f0', '#6f0', '#9f0', '#cf0', '#fc0', '#f90', '#f60', '#f30', '#f00' ];

			  if ( score <= -100 ) {
			      // Assets
			      adjusted_score = ( score * -1 ) - 100;
			      color_scale = [ '#f0f' ];
			  } else if ( score <= -10 ) {
			      // Income
			      adjusted_score = ( score * -1 ) - 10;
			      color_scale = [ '#360' ];
			  }

			  return color_scale[ adjusted_score ];
		      };

		      $scope.color = function () {
			  return function ( d, i ) {
			      return $scope.coloring_score( score_account( d.data.account ) );
			  };
		      };

		      $scope.filter_data = function() {
			  _($scope.balance.buckets).each( function( bucket ) {
			      bucket.data = [];

			      if ( _(bucket.accounts_selected).isEmpty() && bucket.score_threshold === 0 ) {
				  bucket.data = bucket.raw_data;
			      } else {
				  _(bucket.accounts_selected).each( function( account_selected ) {
				      bucket.data = bucket.data.concat( $filter('filter')( bucket.raw_data, account_selected, true ) );
				  } );
			      }

			      bucket.total_detailed = _.chain( bucket.data )
				  .groupBy( function( account ) {
				      return account.account.split(':')[0];
				  } )
				  .each( function( category ) {
				      category.total = _( category ).reduce( function ( memo, account ) {
					  return memo + account.amount;
				      }, 0 );
				  } )
				      .value();
			      bucket.total_detailed = _.chain(bucket.total_detailed)
				  .keys()
				  .map( function( key ) {
				      return { account: key,
					       amount: bucket.total_detailed[key].total };
				  } )
				  .value();

			  } );
		      };

		      var Bucket = function( categories, period ) {
			  var _this = this;
			  this.categories = categories;
			  this.period = period;
			  this.score_threshold = 0;
			  this.orderBy = 'amount';
			  this.orderDesc = false;
			  this.order_by = function( field ) {
			      if ( _this.orderBy == field ) {
				  _this.orderDesc = !_this.orderDesc;
			      } else {
				  _this.orderBy = field;
			      }
			  };
		      };

		      $scope.depth = 99;

		      var retrieve_period_detailed_data = function () {
			  $scope.balance = {
			      buckets: [ new Bucket( 'Expenses Liabilities Equity Income', $scope.period ),
					 new Bucket( 'Assets', null ) ],
			      details: {}
			  };

			  _($scope.balance.buckets).each( function( bucket ) {
			      API.balance( { period: bucket.period,
					     categories: bucket.categories,
					     depth: $scope.depth } )
				  .then( function ( response ) {
				      bucket.raw_data = _.chain( response.data )
					  .map( function( account ) {
					      account.amount = ( account.amount < 0 ) ? account.amount * -1 : account.amount;
					      account.score = score_account( account.account );
					      return account;
					  } )
					  .sortBy( function ( account ) {
					      return 1 / account.amount;
					  } )
					  .sortBy( function ( account ) {
					      return account.account.split(":")[0];
					  } )
					  .value()
					  .reverse();
				      bucket.raw_total = _( response.data ).reduce( function ( memo, account ) {
					  return memo + account.amount;
				      }, 0 );
				      bucket.accounts_selected = bucket.raw_data;

				      $scope.filter_data();
				  } );
			  } );
		      };

		      var retrieve_accounts = function() {
			  API.accounts()
			      .then( function ( response ) {
				  $scope.accounts = response.data.map( function( account_ary ) {
				      return account_ary.join( ':' );
				  } );
			      } );
		      };

		      var retrieve_graph_values = function( params ) {
			  API.graph_values( params ).then( function( response ) {
			      $scope.periods = [];
			      $scope.monthly_values = _.chain( response.data )
				  .keys()
				  .reverse()
				  .map( function( key ) {
				      var multiplicator = ( key == "Income" ) ? -1 : 1;
				      return { "key": key,
					       "values": _(response.data[ key ]).map( function( value ) {
						   var date = new Date( value.date );
						   var period = date.getFullYear() + '-' + ( date.getMonth() < 9 ? '0' : '' ) + ( date.getMonth() + 1 );
						   $scope.periods.push( period );
						   return [ period,
							    parseInt( value.amount ) * multiplicator ];
					       } ) };
				  } )
				  .value();
			      $scope.periods = _.chain($scope.periods).uniq().reverse().value();
			      $scope.period = _($scope.periods).first();
			  } );
		      };
		      $scope.barGraphToolTipContentFunction = function () {
			  return function ( key, x, y, e, graph ) {
			      return '<md-content><h3><em>' + key + '</em> during ' + x + '</h3><h2>' + y + ' €</h2></md-content>';
			  };
		      };
		      $scope.graphed_accounts = [ 'Expenses', 'Income' ];

		      $scope.$watch( 'period', function () {
			  retrieve_period_detailed_data();
		      } );

		      retrieve_accounts();
		      $scope.$watch( 'graphed_accounts', function () {
			  retrieve_graph_values( { period: '',
						   categories: $scope.graphed_accounts.join(' ') } );
		      } );

		      $scope.$on( 'elementClick.directive', function( angularEvent, event ) {
			  $scope.period = event.point[ 0 ];
			  retrieve_period_detailed_data();
		      } );
		  }
		] );