/*
 * Copyright 2012 LinkedIn Corp.
 * 
 * Licensed under the Apache License, Version 2.0 (the "License"); you may not
 * use this file except in compliance with the License. You may obtain a copy of
 * the License at
 * 
 * http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
 * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and limitations under
 * the License.
 */

$.namespace('azkaban');

var statusStringMap = {
	"FAILED": "Failed",
	"SUCCEEDED": "Success",
	"FAILED_FINISHING": "Running w/Failure",
	"RUNNING": "Running",
	"WAITING": "Waiting",
	"KILLED": "Killed",
	"DISABLED": "Disabled",
	"READY": "Ready",
	"UNKNOWN": "Unknown",
	"QUEUED": "Queued"
};

var handleJobMenuClick = function(action, el, pos) {
	var jobid = el[0].jobid;
	var requestURL = contextURL + "/manager?project=" + projectName + "&flow=" + 
			flowId + "&job=" + jobid;
	if (action == "open") {
		window.location.href = requestURL;
	}
	else if (action == "openwindow") {
		window.open(requestURL);
	}
}

var flowTabView;
azkaban.FlowTabView = Backbone.View.extend({
	events: {
		"click #graphViewLink": "handleGraphLinkClick",
		"click #executionsViewLink": "handleExecutionLinkClick",
		"click #summaryViewLink": "handleSummaryLinkClick"
	},
	
	initialize: function(settings) {
		var selectedView = settings.selectedView;
		if (selectedView == "executions") {
			this.handleExecutionLinkClick();
		}
		else {
			this.handleGraphLinkClick();
		}
	},
	
	render: function() {
		console.log("render graph");
	},
	
	handleGraphLinkClick: function(){
		$("#executionsViewLink").removeClass("active");
		$("#graphViewLink").addClass("active");
		$('#summaryViewLink').removeClass('active');
		
		$("#executionsView").hide();
		$("#graphView").show();
		$('#summaryView').hide();
	},
	
	handleExecutionLinkClick: function() {
		$("#graphViewLink").removeClass("active");
		$("#executionsViewLink").addClass("active");
		$('#summaryViewLink').removeClass('active');
		
		$("#graphView").hide();
		$("#executionsView").show();
		$('#summaryView').hide();
		executionModel.trigger("change:view");
	},

  handleSummaryLinkClick: function() {
		$('#graphViewLink').removeClass('active');
		$('#executionsViewLink').removeClass('active');
		$('#summaryViewLink').addClass('active');

		$('#graphView').hide();
		$('#executionsView').hide();
		$('#summaryView').show();
	},
});

var jobListView;
var svgGraphView;
var executionsView;

azkaban.ExecutionsView = Backbone.View.extend({
	events: {
		"click #pageSelection li": "handleChangePageSelection"
	},
	
	initialize: function(settings) {
		this.model.bind('change:view', this.handleChangeView, this);
		this.model.bind('render', this.render, this);
		this.model.set({page: 1, pageSize: 16});
		this.model.bind('change:page', this.handlePageChange, this);
	},
	
	render: function(evt) {
		console.log("render");
		// Render page selections
		var tbody = $("#execTableBody");
		tbody.empty();
		
		var executions = this.model.get("executions");
		for (var i = 0; i < executions.length; ++i) {
			var row = document.createElement("tr");
			
			var tdId = document.createElement("td");
			var execA = document.createElement("a");
			$(execA).attr("href", contextURL + "/executor?execid=" + executions[i].execId);
			$(execA).text(executions[i].execId);
			tdId.appendChild(execA);
			row.appendChild(tdId);
			
			var tdUser = document.createElement("td");
			$(tdUser).text(executions[i].submitUser);
			row.appendChild(tdUser);
			
			var startTime = "-";
			if (executions[i].startTime != -1) {
				var startDateTime = new Date(executions[i].startTime);
				startTime = getDateFormat(startDateTime);
			}

			var tdStartTime = document.createElement("td");
			$(tdStartTime).text(startTime);
			row.appendChild(tdStartTime);
			
			var endTime = "-";
			var lastTime = executions[i].endTime;
			if (executions[i].endTime != -1) {
				var endDateTime = new Date(executions[i].endTime);
				endTime = getDateFormat(endDateTime);
			}
			else {
				lastTime = (new Date()).getTime();
			}

			var tdEndTime = document.createElement("td");
			$(tdEndTime).text(endTime);
			row.appendChild(tdEndTime);
			
			var tdElapsed = document.createElement("td");
			$(tdElapsed).text( getDuration(executions[i].startTime, lastTime));
			row.appendChild(tdElapsed);
			
			var tdStatus = document.createElement("td");
			var status = document.createElement("div");
			$(status).addClass("status");
			$(status).addClass(executions[i].status);
			$(status).text(statusStringMap[executions[i].status]);
			tdStatus.appendChild(status);
			row.appendChild(tdStatus);

			var tdAction = document.createElement("td");
			row.appendChild(tdAction);

			tbody.append(row);
		}
		
		this.renderPagination(evt);
	},
	
	renderPagination: function(evt) {
		var total = this.model.get("total");
		total = total? total : 1;
		var pageSize = this.model.get("pageSize");
		var numPages = Math.ceil(total / pageSize);
		
		this.model.set({"numPages": numPages});
		var page = this.model.get("page");
		
		//Start it off
		$("#pageSelection .selected").removeClass("selected");
		
		// Disable if less than 5
		console.log("Num pages " + numPages)
		var i = 1;
		for (; i <= numPages && i <= 5; ++i) {
			$("#page" + i).removeClass("disabled");
		}
		for (; i <= 5; ++i) {
			$("#page" + i).addClass("disabled");
		}
		
		// Disable prev/next if necessary.
		if (page > 1) {
			$("#previous").removeClass("disabled");
			$("#previous")[0].page = page - 1;
			$("#previous a").attr("href", "#page" + (page - 1));
		}
		else {
			$("#previous").addClass("disabled");
		}
		
		if (page < numPages) {
			$("#next")[0].page = page + 1;
			$("#next").removeClass("disabled");
			$("#next a").attr("href", "#page" + (page + 1));
		}
		else {
			$("#next")[0].page = page + 1;
			$("#next").addClass("disabled");
		}
		
		// Selection is always in middle unless at barrier.
		if (page < 3) {
			selectionPosition = page;
		}
		else if (page > numPages - 2) {
			selectionPosition = 5 - (numPages - page) - 1;
		}
		else {
			selectionPosition = 3;
		}

		$("#page"+selectionPosition).addClass("selected");
		$("#page"+selectionPosition)[0].page = page;
		var selecta = $("#page" + selectionPosition + " a");
		selecta.text(page);
		selecta.attr("href", "#page" + page);

		for (var j = 1, tpage = page - selectionPosition + 1; j < selectionPosition; ++j, ++tpage) {
			$("#page" + j)[0].page = tpage;
			var a = $("#page" + i + " a");
			a.text(tpage);
			a.attr("href", "#page" + tpage);
		}

		for (var i = selectionPosition + 1, tpage = page + 1; i <= numPages; ++i, ++tpage) {
			$("#page" + i)[0].page = tpage;
			var a = $("#page" + i + " a");
			a.text(tpage);
			a.attr("href", "#page" + tpage);
		}
	},
	
	handleChangePageSelection: function(evt) {
		if ($(evt.currentTarget).hasClass("disabled")) {
			return;
		}
		var page = evt.currentTarget.page;
		this.model.set({"page": page});
	},
	
	handleChangeView: function(evt) {
		if (this.init) {
			return;
		}
		console.log("init");
		this.handlePageChange(evt);
		this.init = true;
	},
	
	handlePageChange: function(evt) {
		var page = this.model.get("page") - 1;
		var pageSize = this.model.get("pageSize");
		var requestURL = contextURL + "/manager";
		
		var model = this.model;
		var requestData = {
			"project": projectName, 
			"flow": flowId, 
			"ajax": "fetchFlowExecutions", 
			"start": page * pageSize, 
			"length": pageSize
		};
		var successHandler = function(data) {
			model.set({
				"executions": data.executions, 
				"total": data.total
			});
			model.trigger("render");
		};
		$.get(requestURL, requestData, successHandler, "json");
	}
});

var summaryView;
azkaban.SummaryView = Backbone.View.extend({
	events: {
    'click #analyze-btn': 'fetchLastRun'
	},
	
	initialize: function(settings) {
		this.model.bind('change:view', this.handleChangeView, this);
		this.model.bind('render', this.render, this);
		
		this.fetchDetails();
    this.fetchSchedule();
		this.model.trigger('render');
	},

  fetchDetails: function() {
    var requestURL = contextURL + "/manager";
    var requestData = {
      'ajax': 'fetchflowdetails',
      'project': projectName,
      'flow': flowId
    };
		var model = this.model;
    var successHandler = function(data) {
      model.set({ 'jobTypes': data.jobTypes });
      model.trigger('render');
    };
    $.get(requestURL, requestData, successHandler, 'json');
  },

	fetchSchedule: function() {
		var requestURL = contextURL + "/schedule"
		var requestData = {
			'ajax': 'fetchSchedule',
			'projectId': projectId,
			'flowId': flowId
		};
		var model = this.model;
		var successHandler = function(data) {
      model.set({'schedule': data.schedule});
      model.trigger('render');
		};
		$.get(requestURL, requestData, successHandler, 'json');
	},

	fetchLastRun: function() {
		var requestURL = contextURL + "/manager";
    var requestData = {
      'ajax': 'fetchLastSuccessfulFlowExecution',
      'project': projectName,
      'flow': flowId
    };
    var view = this;
    var successHandler = function(data) {
      if (data.success == "false" || data.execId == null) {
        view.renderLastRun(data);
        return;
      }
      view.analyzeLastRun(data.execId);
    };
    $.get(requestURL, requestData, successHandler, 'json');
	},

  fetchJobs: function(execId) {
    var requestURL = contextURL + "/executor";
    var requestData = {"execid": execId, "ajax":"fetchexecflow"};
    var jobs = [];
    var successHandler = function(data) {
      for (var i = 0; i < data.nodes.length; ++i) {
        var node = data.nodes[i];
        jobs.push(node.id);
      }
    };
    $.ajax({
      url: requestURL,
      data: requestData,
      success: successHandler,
      dataType: "json",
      async: false
    });
    return jobs;
  },

  fetchJobStats: function(jobId, execId) {
    var requestURL = contextURL + "/executor";
    var requestData = {
      "execid": execId,
      "flowid": flowId,
      "jobid": jobId,
      "ajax": "fetchExecJobStats"
    };
    var stats = null;
    var successHandler = function(data) {
      stats = data;
    };
    $.ajax({
      url: requestURL,
      data: requestData,
      success: successHandler,
      dataType: "json",
      async: false
    });
    return stats;
  },

  updateStats: function(jobStats, data) {
    var aggregateStats = data.stats;
    var state = jobStats.state;
    var conf = jobStats.conf;
    var mappers = parseInt(state.totalMappers);
    var reducers = parseInt(state.totalReducers);
    if (mappers > aggregateStats.maxMapSlots) {
      aggregateStats.maxMapSlots = mappers;
    }
    if (reducers > aggregateStats.maxReduceSlots) {
      aggregateStats.maxReduceSlots = reducers;
    }
    aggregateStats.totalMapSlots += mappers;
    aggregateStats.totalReduceSlots += reducers;
  },

  analyzeLastRun: function(execId) {
    var jobs = this.fetchJobs(execId);
    if (jobs == null) {
      this.renderLastRun(null);
      return;
    }

    var data = {
      success: false,
      message: null,
      warnings: [],
      stats: {
        maxMapSlots: 0,
        maxReduceSlots: 0,
        totalMapSlots: 0,
        totalReduceSlots: 0,
        numJobs: jobs.length,
        longestTaskTime: 0
      }
    };

    for (var i = 0; i < jobs.length; ++i) {
      var job = jobs[i];
      var jobStats = this.fetchJobStats(job, execId);
      if (jobStats.jobStats == null) {
        data.warnings.push("No job stats available for job " + job.id);
        continue;
      }
      for (var j = 0; j < jobStats.jobStats.length; ++j) {
        this.updateStats(jobStats.jobStats[j], data);
      }
    }
    data.success = true;
    this.renderLastRun(data);
  },

  renderLastRun: function(data) {
    var view = this;
    if (data == null) {
      var msg = { message: "Error retrieving last run data."};
      dust.render("flowsummary-no-data", msg, function(err, out) {
        view.displayLastRun(out);
      });
    }
    else if (data.success == "false") {
      dust.render("flowsummary-no-data", data, function(err, out) {
        view.displayLastRun(out);
      });
    }
    else {
      dust.render("flowsummary-last-run", data, function(err, out) {
        view.displayLastRun(out);
      });
    }
  },

  displayLastRun: function(out) {
    $('#last-run-container').html(out);
  },

	handleChangeView: function(evt) {
	},

	render: function(evt) {
		var data = {
      projectName: projectName,
			flowName: flowId,
      jobTypes: this.model.get('jobTypes'),
			schedule: this.model.get('schedule'),
		};
		dust.render("flowsummary", data, function(err, out) {
			$('#summary-view-content').html(out);
		});
	},
});

var exNodeClickCallback = function(event) {
	console.log("Node clicked callback");
	var jobId = event.currentTarget.jobid;
	var requestURL = contextURL + "/manager?project=" + projectName + "&flow=" + 
			flowId + "&job=" + jobId;

	var menu = [	
		{title: "Open Job...", callback: function() {window.location.href=requestURL;}},
		{title: "Open Job in New Window...", callback: function() {window.open(requestURL);}}
	];

	contextMenuView.show(event, menu);
}

var exJobClickCallback = function(event) {
	console.log("Node clicked callback");
	var jobId = event.currentTarget.jobid;
	var requestURL = contextURL + "/manager?project=" + projectName + "&flow=" + 
			flowId + "&job=" + jobId;

	var menu = [	
		{title: "Open Job...", callback: function() {window.location.href=requestURL;}},
		{title: "Open Job in New Window...", callback: function() {window.open(requestURL);}}
	];

	contextMenuView.show(event, menu);
}

var exEdgeClickCallback = function(event) {
	console.log("Edge clicked callback");
}

var exGraphClickCallback = function(event) {
	console.log("Graph clicked callback");
	var requestURL = contextURL + "/manager?project=" + projectName + "&flow=" + flowId;

	var menu = [	
		{title: "Open Flow...", callback: function() {window.location.href=requestURL;}},
		{title: "Open Flow in New Window...", callback: function() {window.open(requestURL);}},
		{break: 1},
		{title: "Center Graph", callback: function() {graphModel.trigger("resetPanZoom");}}
	];
	
	contextMenuView.show(event, menu);
}

var graphModel;
azkaban.GraphModel = Backbone.Model.extend({});

var executionModel;
azkaban.ExecutionModel = Backbone.Model.extend({});

var summaryModel;
azkaban.SummaryModel = Backbone.Model.extend({});

var mainSvgGraphView;

$(function() {
	var selected;
	// Execution model has to be created before the window switches the tabs.
	executionModel = new azkaban.ExecutionModel();
	executionsView = new azkaban.ExecutionsView({
		el: $('#executionsView'), 
		model: executionModel
	});
	
  summaryModel = new azkaban.SummaryModel();
	summaryView = new azkaban.SummaryView({
		el: $('#summaryView'),
		model: summaryModel
	});
	
  flowTabView = new azkaban.FlowTabView({
		el: $('#headertabs'), 
		selectedView: selected 
	});

	graphModel = new azkaban.GraphModel();
	mainSvgGraphView = new azkaban.SvgGraphView({
		el: $('#svgDiv'), 
		model: graphModel, 
		rightClick: { 
			"node": exNodeClickCallback, 
			"edge": exEdgeClickCallback, 
			"graph": exGraphClickCallback 
		}
	});
	
  jobsListView = new azkaban.JobListView({
		el: $('#jobList'), 
		model: graphModel, 
		contextMenuCallback: exJobClickCallback
	});
	
	var requestURL = contextURL + "/manager";

	// Set up the Flow options view. Create a new one every time :p
	$('#executebtn').click(function() {
		var data = graphModel.get("data");
		var nodes = data.nodes;
		var executingData = {
			project: projectName,
			ajax: "executeFlow",
			flow: flowId
		};

		flowExecuteDialogView.show(executingData);
	});

	var requestData = {
		"project": projectName, 
		"ajax": "fetchflowgraph", 
		"flow": flowId
	};
	var successHandler = function(data) {
		// Create the nodes
		var nodes = {};
		for (var i = 0; i < data.nodes.length; ++i) {
			var node = data.nodes[i];
			nodes[node.id] = node;
		}
		for (var i = 0; i < data.edges.length; ++i) {
			var edge = data.edges[i];
			var fromNode = nodes[edge.from];
			var toNode = nodes[edge.target];
			
			if (!fromNode.outNodes) {
				fromNode.outNodes = {};
			}
			fromNode.outNodes[toNode.id] = toNode;
			
			if (!toNode.inNodes) {
				toNode.inNodes = {};
			}
			toNode.inNodes[fromNode.id] = fromNode;
		}
	
		console.log("data fetched");
		graphModel.set({data: data});
		graphModel.set({nodes: nodes});
		graphModel.set({disabled: {}});
		graphModel.trigger("change:graph");
		
		// Handle the hash changes here so the graph finishes rendering first.
		if (window.location.hash) {
			var hash = window.location.hash;
			if (hash == "#executions") {
				flowTabView.handleExecutionLinkClick();
			}
			if (hash == "#summary") {
				flowTabView.handleSummaryLinkClick();
			}
			else if (hash == "#graph") {
				// Redundant, but we may want to change the default. 
				selected = "graph";
			}
			else {
				if ("#page" == hash.substring(0, "#page".length)) {
					var page = hash.substring("#page".length, hash.length);
					console.log("page " + page);
					flowTabView.handleExecutionLinkClick();
					executionModel.set({"page": parseInt(page)});
				}
				else {
					selected = "graph";
				}
			}
		}
	};
	$.get(requestURL, requestData, successHandler, "json");
});
