mirror of
				https://github.com/taigrr/wtf
				synced 2025-01-18 04:03:14 -08:00 
			
		
		
		
	* Create generalized todo module Makes existing modules "backends" This gives us a consistent interface * Fix check issues * Fix lint errors * Make changes to stay on wtfutil todoist fork
		
			
				
	
	
		
			109 lines
		
	
	
		
			2.2 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			109 lines
		
	
	
		
			2.2 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| package backend
 | |
| 
 | |
| import (
 | |
| 	"strconv"
 | |
| 
 | |
| 	"github.com/olebedev/config"
 | |
| 	"github.com/wtfutil/todoist"
 | |
| )
 | |
| 
 | |
| type Todoist struct {
 | |
| 	projects []interface{}
 | |
| }
 | |
| 
 | |
| func (todo *Todoist) Title() string {
 | |
| 	return "Todoist"
 | |
| }
 | |
| 
 | |
| func (todo *Todoist) Setup(config *config.Config) {
 | |
| 	todoist.Token = config.UString("apiKey")
 | |
| 	todo.projects = config.UList("projects")
 | |
| }
 | |
| 
 | |
| func (todo *Todoist) BuildProjects() []*Project {
 | |
| 	projects := []*Project{}
 | |
| 
 | |
| 	for _, id := range todo.projects {
 | |
| 		i := strconv.Itoa(id.(int))
 | |
| 		proj := todo.GetProject(i)
 | |
| 		projects = append(projects, proj)
 | |
| 	}
 | |
| 	return projects
 | |
| }
 | |
| 
 | |
| func (todo *Todoist) GetProject(id string) *Project {
 | |
| 	// Todoist seems to experience a lot of network issues on their side
 | |
| 	// If we can't connect, handle it with an empty project until we can
 | |
| 	proj := &Project{
 | |
| 		Index:   -1,
 | |
| 		backend: todo,
 | |
| 	}
 | |
| 	i64, _ := strconv.ParseUint(id, 10, 32)
 | |
| 	i := uint(i64)
 | |
| 	project, err := todoist.GetProject(i)
 | |
| 	if err != nil {
 | |
| 		proj.Err = err
 | |
| 		return proj
 | |
| 	}
 | |
| 
 | |
| 	proj.ID = strconv.FormatUint(uint64(project.ID), 10)
 | |
| 	proj.Name = project.Name
 | |
| 
 | |
| 	tasks, err := todo.LoadTasks(proj.ID)
 | |
| 	proj.Err = err
 | |
| 	proj.Tasks = tasks
 | |
| 
 | |
| 	return proj
 | |
| }
 | |
| 
 | |
| func toTask(task todoist.Task) Task {
 | |
| 	id := strconv.FormatUint(uint64(task.ID), 10)
 | |
| 	return Task{
 | |
| 		ID:        id,
 | |
| 		Completed: task.Completed,
 | |
| 		Name:      task.Content,
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func (todo *Todoist) LoadTasks(id string) ([]Task, error) {
 | |
| 	tasks, err := todoist.ListTask(todoist.QueryParam{"project_id": id})
 | |
| 
 | |
| 	if err != nil {
 | |
| 		return nil, err
 | |
| 	}
 | |
| 	var finalTasks []Task
 | |
| 	for _, item := range tasks {
 | |
| 		finalTasks = append(finalTasks, toTask(item))
 | |
| 	}
 | |
| 	return finalTasks, nil
 | |
| }
 | |
| 
 | |
| func (todo *Todoist) CloseTask(task *Task) error {
 | |
| 	if task != nil {
 | |
| 		i64, _ := strconv.ParseUint(task.ID, 10, 32)
 | |
| 		i := uint(i64)
 | |
| 		internal := todoist.Task{ID: i}
 | |
| 		return internal.Close()
 | |
| 	}
 | |
| 	return nil
 | |
| }
 | |
| 
 | |
| func (todo *Todoist) DeleteTask(task *Task) error {
 | |
| 	if task != nil {
 | |
| 		i64, _ := strconv.ParseUint(task.ID, 10, 32)
 | |
| 		i := uint(i64)
 | |
| 		internal := todoist.Task{ID: i}
 | |
| 		return internal.Delete()
 | |
| 	}
 | |
| 	return nil
 | |
| }
 | |
| 
 | |
| func (todo *Todoist) Sources() []string {
 | |
| 	var result []string
 | |
| 	for _, id := range todo.projects {
 | |
| 		i := strconv.Itoa(id.(int))
 | |
| 		result = append(result, i)
 | |
| 	}
 | |
| 	return result
 | |
| }
 |