Shuo/lib/models.js

var _ = require("lodash");
var Backbone = require("backbone");
var moment   = require("moment");

var id = 1;
var models =
	module.exports =
		{};

models.User = Backbone.Model.extend({
	defaults: {
		mode: "",
		name: "",
	}
});

models.Users = Backbone.Collection.extend({
	model: models.User,
	sort: function(options) {
		this.models = _.sortBy(
			this.models,
			function(user) {
				return user.get("name").toLowerCase();
			}
		);

		// Move users with these modes to the top.
		var modes = ["+", "@"];
		for (var i in modes) {
			this.models = _.remove(this.models, function(user) {
				if (user.get("mode") == modes[i]) {
					return true;
				}
			}).concat(this.models);
		}
	}
});

models.Message = Backbone.Model.extend({
	defaults: {
		type: "",
		time: "",
		from: "",
		message: "",
	},
	initialize: function() {
		this.set("time", moment().format("HH:mm"));
	}
});

models.Messages = Backbone.Collection.extend({
	model: models.Message
});

models.Channel = Backbone.Model.extend({
	defaults: {
		type: "channel",
		name: "",
	},
	addUser: function(models) {
		return this.get("users").add(models);
	},
	addMessage: function(models) {
		return this.get("messages").add(models);
	},
	initialize: function() {
		this.set({
			id:       id++,
			messages: new models.Messages,
			users:    new models.Users,
		});

		this.get("messages").on("all", function(action, data) {
			this.trigger("message", {
				target: this.get("id"),
				type:   "message",
				data:   data,
				action: action,
			});
		}, this);

		var users = this.get("users");
		users.on("all", function(action, data) {
			this.trigger("user", {
				target: this.get("id"),
				type:   "user",
				data:   users,
				action: action,
			});
		}, this);
	}
});

models.Channels = Backbone.Collection.extend({
	model: models.Channel
});

models.Network = Backbone.Model.extend({
	defaults: {
		host: "",
		client: null,
	},
	addChannel: function(models) {
		return this.get("channels").add(models);
	},
	initialize: function() {
		this.set({
			id:       id++,
			channels: new models.Channels,
		});

		this.get("channels").on("all", function(action, data) {
			if (action == "message"
				|| action == "user") {
				return this.trigger(action, data);
			}
			this.trigger("channel", {
				target: this.get("id"),
				type:   "channel",
				data:   data,
				action: action,
			});
		}, this);

		this.addChannel({
			type: "network",
			name: this.get("host")
		});
	},
	toJSON: function() {
		return _.omit(this.attributes, "client");
	}
});

models.Networks = Backbone.Collection.extend({
	model: models.Network,
	initialize: function() {
		this.add({host: "Status"});
	},
	find: function(id) {
		var i = this.models.length;
		while (i--) {
			var find = this.models[i].get("channels").findWhere({id: id});
			if (find) {
				return {
					network: this.models[i],
					channel: find
				};
			}
		}
	}
});