import Ember from 'ember';

export default function () {
    Ember.Route.reopen({
        // `titleToken` can either be a static string or a function
        // that accepts a model object and returns a string (or array
        // of strings if there are multiple tokens).
        titleToken: null,

        // `title` can either be a static string or a function
        // that accepts an array of tokens and returns a string
        // that will be the document title. The `collectTitleTokens` action
        // stops bubbling once a route is encountered that has a `title`
        // defined.
        title: null,

        _actions: {
            collectTitleTokens: function (tokens) {
                var titleToken = this.titleToken,
                    finalTitle;

                if (typeof this.titleToken === 'function') {
                    titleToken = this.titleToken(this.currentModel);
                }

                if (Ember.isArray(titleToken)) {
                    tokens.unshift.apply(this, titleToken);
                } else if (titleToken) {
                    tokens.unshift(titleToken);
                }

                if (this.title) {
                    if (typeof this.title === 'function') {
                        finalTitle = this.title(tokens);
                    } else {
                        finalTitle = this.title;
                    }

                    this.router.setTitle(finalTitle);
                } else {
                    return true;
                }
            }
        }
    });

    Ember.Router.reopen({
        updateTitle: Ember.on('didTransition', function () {
            this.send('collectTitleTokens', []);
        }),

        setTitle: function (title) {
            if (Ember.testing) {
                this._title = title;
            } else {
                window.document.title = title;
            }
        }
    });
}