redash/rd_ui/app/scripts/controllers/query_view.js

(function () {
    'use strict';

    function QueryViewCtrl($scope, $window, $route, $http, $location, growl, notifications, Query, Visualization) {
        var DEFAULT_TAB = 'table';
        var route = $route.current;

        $scope.query = route.locals.query;
        $scope.queryResult = $scope.query.getQueryResult();
        $scope.queryExecuting = false;

        $scope.isOwner = currentUser.canEdit($scope.query);
        $scope.canEdit = $scope.isSourceVisible && $scope.isOwner;
        $scope.canFork = true;

        $scope.sourceHref = getQuerySourceUrl();

        function getQuerySourceUrl(queryId) {
            return '/queries/' + $scope.query.id + '/source#' + $location.hash();
        };

        $scope.lockButton = function (lock) {
            $scope.queryExecuting = lock;
        };



        $scope.duplicateQuery = function () {
            var oldId = $scope.query.id;
            $scope.query.id = null;
            $scope.query.ttl = -1;

            $scope.saveQuery(true, oldId);
        };

        $scope.executeQuery = function () {
            $scope.queryResult = $scope.query.getQueryResult(0);
            $scope.lockButton(true);
            $scope.cancelling = false;
        };

        $scope.cancelExecution = function () {
            $scope.cancelling = true;
            $scope.queryResult.cancelExecution();
        };

        $scope.$watch('queryResult && queryResult.getError()', function (newError, oldError) {
            if (newError == undefined) {
                return;
            }

            if (oldError == undefined && newError != undefined) {
                $scope.lockButton(false);
            }
        });

        $scope.$watch('queryResult && queryResult.getData()', function (data, oldData) {
            if (!data) {
                return;
            }

            $scope.filters = $scope.queryResult.getFilters();

            if ($scope.queryResult.getId() == null) {
                $scope.dataUri = "";
            } else {
                $scope.dataUri = '/api/queries/' + $scope.query.id + '/results/' + $scope.queryResult.getId() + '.csv';
                $scope.dataFilename = $scope.query.name.replace(" ", "_") + moment($scope.queryResult.getUpdatedAt()).format("_YYYY_MM_DD") + ".csv";
            }
        });

        $scope.$watch("queryResult && queryResult.getStatus()", function (status) {
            if (!status) {
                return;
            }

            if (status == "done") {
                if ($scope.query.id && $scope.query.latest_query_data_id != $scope.queryResult.getId() &&
                    $scope.query.query_hash == $scope.queryResult.query_result.query_hash) {
                    Query.save({
                        'id': $scope.query.id,
                        'latest_query_data_id': $scope.queryResult.getId()
                    })
                }
                $scope.query.latest_query_data_id = $scope.queryResult.getId();

                notifications.showNotification("re:dash", $scope.query.name + " updated.");

                $scope.lockButton(false);
            }
        });

        $scope.$watch('query.name', function () {
            $scope.$parent.pageTitle = $scope.query.name;
        });
        $scope.$watch(function () {
            return $location.hash()
        }, function (hash) {
            $scope.selectedTab = hash || DEFAULT_TAB;
            $scope.sourceHref = getQuerySourceUrl();
        });
    };

    angular.module('redash.controllers')
    .controller('QueryViewCtrl', ['$scope', '$window', '$route', '$http', '$location', 'growl', 'notifications', 'Query', 'Visualization', QueryViewCtrl]);
})();