timetracking now works for projects as well

client/components/projects/Project.js

@@ -9,6 +9,7 @@ import NotLoggedIn from '../NotLoggedIn';
 import NotFound from '../NotFound';
 import TasksList from '../tasks/TasksList';
 import AddForm from './AddForm';
+import { formatTime } from '../../helpers';
 
 class Project extends React.Component {
   constructor(props) {
@@ -100,11 +101,15 @@ class Project extends React.Component {
       return <NotFound />;
     }
     const addLinkText = this.state.showForm ? 'Hide The Form' : 'New Task';
+    const projectTime = this.state.currentProject.timeSpent
+      ? formatTime(this.state.currentProject.timeSpent)
+      : '';
     return (
       <Layout>
         <h1 className="page-title">
           {this.state.currentProject.name || '...'}
         </h1>
+        <h3 className="page-title">Time spent: {projectTime}</h3>
         <a
           onClick={this.showAddForm}
           href="#"

client/reducers/projects.js

@@ -4,6 +4,7 @@ import {
   RENAME_PROJECT,
   CLEAR_PROJECTS,
   DELETE_PROJECT,
+  ADD_TIMELOG,
 } from '../actions/actionTypes';
 
 function projects(state = [], action) {
@@ -28,6 +29,15 @@ function projects(state = [], action) {
       const projectsList = state.filter(item => item._id !== action.id);
       return projectsList;
     }
+    case ADD_TIMELOG: {
+      const projectList = state.map(item => {
+        if (item._id === action.data.project._id) {
+          item.timeSpent += action.seconds; // eslint-disable-line no-param-reassign
+        }
+        return item;
+      });
+      return projectList;
+    }
     default:
       return state;
   }

controllers/timelogController.js

@@ -1,6 +1,7 @@
 const mongoose = require('mongoose');
 
 const Timelog = mongoose.model('Timelog');
+const Project = mongoose.model('Project');
 const Task = mongoose.model('Task');
 
 exports.addTime = async (req, res) => {
@@ -10,8 +11,17 @@ exports.addTime = async (req, res) => {
     task.timeSpent += req.body.seconds; // eslint-disable-line no-param-reassign
     task.save();
   });
-  const [timelog, task] = await Promise.all([timelogPromise, taskPromise]);
-  res.json({ timelog, task });
+  const projectPromise = Project.findById(req.body.project, (err, project) => {
+    project.timeSpent += req.body.seconds; // eslint-disable-line no-param-reassign
+    project.save();
+  });
+
+  const [timelog, task, project] = await Promise.all([
+    timelogPromise,
+    taskPromise,
+    projectPromise,
+  ]);
+  res.json({ timelog, task, project });
 };
 
 exports.getLogs = async (req, res) => {

models/Project.js

@@ -13,6 +13,10 @@ const projectSchema = new mongoose.Schema({
     ref: 'User',
     required: 'Project should belong to an author',
   },
+  timeSpent: {
+    type: Number,
+    default: 0,
+  },
 });
 
 module.exports = mongoose.model('Project', projectSchema);