Merge: Nitdoc: more cleaning and uniformize method names

[nit.git] / contrib / nitrpg / src / statistics.nit

diff --git a/contrib/nitrpg/src/statistics.nit b/contrib/nitrpg/src/statistics.nit
index 061722f..7b493c6 100644 (file)
--- a/contrib/nitrpg/src/statistics.nit
+++ b/contrib/nitrpg/src/statistics.nit
@@ -26,6 +26,9 @@ import counter
 
 redef class GameEntity
 
+       # Statistics for this entity.
+       fun stats: GameStats is abstract
+
        # Load statistics for this `MEntity` if any.
        fun load_statistics: nullable GameStats do
                var key = self.key / "statistics"
@@ -33,23 +36,37 @@ redef class GameEntity
                var json = game.store.load_object(key)
                return new GameStats.from_json(game, json)
        end
+end
+
+redef class Game
+
+       redef var stats is lazy do
+               return load_statistics or else new GameStats(game)
+       end
 
-       # Save statistics under this `MEntity`.
-       fun save_statistics(stats: GameStats) do
-               game.store.store_object(key / stats.key, stats.to_json)
+       redef fun save do
+               super
+               stats.save_in(self)
+       end
+
+       redef fun pretty do
+               var res = new FlatBuffer
+               res.append super
+               res.append "# stats:\n"
+               res.append stats.pretty
+               return res.write_to_string
        end
 end
 
-redef class Game
+redef class Player
 
-       # Statistics for this game instance.
-       var stats: GameStats is lazy do
+       redef var stats is lazy do
                return load_statistics or else new GameStats(game)
        end
 
        redef fun save do
                super
-               save_statistics(stats)
+               stats.save_in(self)
        end
 
        redef fun pretty do
@@ -104,11 +121,7 @@ end
 class StatisticsReactor
        super GameReactor
 
-       redef fun react_event(game, e) do
-               super # log events
-               e.react_stats_event(game)
-               game.save
-       end
+       redef fun react_event(game, e) do e.react_stats_event(game)
 end
 
 redef class GithubEvent
@@ -123,13 +136,24 @@ redef class IssuesEvent
 
        # Count opened and closed issues.
        redef fun react_stats_event(game) do
+               var player = issue.user.player(game)
                if action == "opened" then
                        game.stats.inc("issues")
                        game.stats.inc("issues_open")
+                       game.save
+                       player.stats.inc("issues")
+                       player.stats.inc("issues_open")
+                       player.save
                else if action == "reopened" then
                        game.stats.inc("issues_open")
+                       game.save
+                       player.stats.inc("issues_open")
+                       player.save
                else if action == "closed" then
                        game.stats.dec("issues_open")
+                       game.save
+                       player.stats.dec("issues_open")
+                       player.save
                end
        end
 end
@@ -138,16 +162,52 @@ redef class PullRequestEvent
 
        # Count opened and closed pull requests.
        redef fun react_stats_event(game) do
+               var player = pull.user.player(game)
                if action == "opened" then
                        game.stats.inc("pulls")
                        game.stats.inc("pulls_open")
+                       game.save
+                       player.stats.inc("pulls")
+                       player.stats.inc("pulls_open")
+                       player.save
                else if action == "reopened" then
                        game.stats.inc("pulls_open")
+                       game.save
+                       player.stats.inc("pulls_open")
+                       player.save
                else if action == "closed" then
                        game.stats.dec("pulls_open")
+                       player.stats.dec("pulls_open")
                        if pull.merged then
                                game.stats["commits"] += pull.commits
+                               player.stats["commits"] += pull.commits
+                       end
+                       game.save
+                       player.save
+               end
+       end
+end
+
+redef class IssueCommentEvent
+
+       # Count posted comments
+       redef fun react_stats_event(game) do
+               if action == "created" then
+                       var player = comment.user.player(game)
+                       game.stats.inc("comments")
+                       player.stats.inc("comments")
+                       # FIXME use a more precise way to locate reviews
+                       if comment.has_ok_review then
+                               game.stats.inc("reviews")
+                               player.stats.inc("reviews")
                        end
+                       game.save
+                       player.save
                end
        end
 end
+
+redef class IssueComment
+       # Does this comment contain a "+1"?
+       fun has_ok_review: Bool do return body.has("\\+1\\b".to_re)
+end