#!/usr/bin/env python
# encoding: utf-8
"""
taskpaper.py

Created by Emil on 2009-06-02.
Copyright (c) 2009 Emil Erlandsson

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.  
"""

import sys
import os  

class TaskPaperProject(object):
  """Representing a project in a TaskPaper file"""
  def __init__(self, line):
    self.line = line
    self.name = ""
    self.tasks = []  
    self.notes = []
    self.__parse()
  
  def __parse(self):
    """Parses the line and sets up variables"""
    self.name = self.line[:-1]
   
  def add_note(self, note):
    """Adds a note to this project"""
    self.notes.append(note)
    
  def add_task(self, task):
    """Adds a task to this project"""
    self.tasks.append(task)
                          
                          
class TaskPaperTask(object):
  """Represents a task in a taskpaper"""
  def __init__(self, line):
    self.line = line       
    self.tags = []
    self.project = None 
    self.name = ""
    self.notes = []
    self.__parse() 
  
  def __parse(self):
    """Parses the line and sets up variables"""
    tokens = self.line.replace("- ", "").split("@")
    self.name = tokens[0]
    self.tags = tokens[1:]
  
  def set_project(self, project):
    self.project = project

  def add_note(self, note):
    """Adds a note to this project"""
    self.notes.append(note)
    
  def add_tag(self, tag):
    self.tags.append(tag)
  

class TaskPaper(object):
  """A wrapper class for TaskPaper files"""
  def __init__(self, filename):
    self.filename = filename
    self.projects = []
    self.tasks = []       

  def add_task(self, task):
    """Adds a no-parent task to this task paper"""
    self.tasks.append(task)
  
  def add_project(self, project):
    """Adds a project to the taskpaper"""
    self.projects.append(project)

  def __str__(self):
    return "TaskPaper with %d headless tasks and %d projects" \
    % (len(self.tasks), len(self.projects)) 
    
  def print_entries(self):
    """Prints all entries in the task paper"""
    
    print "Taskpaper: %s" % self.filename
    
    for project in self.projects:
      print "\n%s:" % project.name
      for note in project.notes:
        print note
        
      for task in project.tasks:
        rep = "- %s" % task.name.strip()
        for tag in task.tags:
          rep += " @%s" % tag.strip()
        print rep
        for note in task.notes:
          print "%s" % note
    print "\n\n"
    
  def print_stats(self): 
    tasks = len(self.tasks)
    notes = 0
    
    for project in self.projects:
      tasks += len(project.tasks)
      notes += len(project.notes)
      for task in project.tasks:
        notes += len(task.notes)
    
    print "%s has: %d projects with %d tasks and %d notes." \
    % (self.filename, len(self.projects), tasks, notes)
    


def parse_task_paper(filename):
  """Parsing of a TaskPaper file"""
  handle = file(filename)
  
  taskpaper = TaskPaper(filename)
  project = None
  last = None
  for line in handle.readlines():
    line = unicode(line.strip(), "utf-8")
    
    if line.startswith("- "): # It is a task
      task = TaskPaperTask(line)
      last = task
      if project is not None:
        project.add_task(task)
        task.project = project         
        
    elif line.endswith(":"): # It is a project
      project = TaskPaperProject(line)
      taskpaper.add_project(project)  
      last = project
      
    else:                    # It is a note
      last.add_note(line)
  
  return taskpaper
    

if __name__ == '__main__':  
  tp = parse_task_paper(sys.argv[1]) 
  tp.print_entries()