path: root/internal/api/todoist.go

package api

import (
	"bytes"
	"context"
	"encoding/json"
	"fmt"
	"io"
	"net/http"
	"time"

	"task-dashboard/internal/models"
)

const (
	todoistBaseURL = "https://api.todoist.com/rest/v2"
)

// TodoistClient handles interactions with the Todoist API
type TodoistClient struct {
	apiKey     string
	baseURL    string
	httpClient *http.Client
}

// NewTodoistClient creates a new Todoist API client
func NewTodoistClient(apiKey string) *TodoistClient {
	return &TodoistClient{
		apiKey:  apiKey,
		baseURL: todoistBaseURL,
		httpClient: &http.Client{
			Timeout: 30 * time.Second,
		},
	}
}

// todoistTaskResponse represents the API response structure
type todoistTaskResponse struct {
	ID          string   `json:"id"`
	Content     string   `json:"content"`
	Description string   `json:"description"`
	ProjectID   string   `json:"project_id"`
	Priority    int      `json:"priority"`
	Labels      []string `json:"labels"`
	Due         *struct {
		Date     string `json:"date"`
		Datetime string `json:"datetime"`
	} `json:"due"`
	URL       string `json:"url"`
	CreatedAt string `json:"created_at"`
}

// todoistProjectResponse represents the project API response
type todoistProjectResponse struct {
	ID   string `json:"id"`
	Name string `json:"name"`
}

// GetTasks fetches all active tasks from Todoist
func (c *TodoistClient) GetTasks(ctx context.Context) ([]models.Task, error) {
	req, err := http.NewRequestWithContext(ctx, "GET", c.baseURL+"/tasks", nil)
	if err != nil {
		return nil, fmt.Errorf("failed to create request: %w", err)
	}

	req.Header.Set("Authorization", "Bearer "+c.apiKey)

	resp, err := c.httpClient.Do(req)
	if err != nil {
		return nil, fmt.Errorf("failed to fetch tasks: %w", err)
	}
	defer resp.Body.Close()

	if resp.StatusCode != http.StatusOK {
		body, _ := io.ReadAll(resp.Body)
		return nil, fmt.Errorf("todoist API error (status %d): %s", resp.StatusCode, string(body))
	}

	var apiTasks []todoistTaskResponse
	if err := json.NewDecoder(resp.Body).Decode(&apiTasks); err != nil {
		return nil, fmt.Errorf("failed to decode response: %w", err)
	}

	// Fetch projects to get project names
	projects, err := c.GetProjects(ctx)
	projectMap := make(map[string]string)
	if err == nil {
		// Build map of project ID to name
		for _, proj := range projects {
			projectMap[proj.ID] = proj.Name
		}
	}

	// Convert to our model
	tasks := make([]models.Task, 0, len(apiTasks))
	for _, apiTask := range apiTasks {
		task := models.Task{
			ID:          apiTask.ID,
			Content:     apiTask.Content,
			Description: apiTask.Description,
			ProjectID:   apiTask.ProjectID,
			ProjectName: projectMap[apiTask.ProjectID],
			Priority:    apiTask.Priority,
			Completed:   false,
			Labels:      apiTask.Labels,
			URL:         apiTask.URL,
		}

		// Parse created_at
		if createdAt, err := time.Parse(time.RFC3339, apiTask.CreatedAt); err == nil {
			task.CreatedAt = createdAt
		}

		// Parse due date
		if apiTask.Due != nil {
			var dueDate time.Time
			if apiTask.Due.Datetime != "" {
				dueDate, err = time.Parse(time.RFC3339, apiTask.Due.Datetime)
			} else if apiTask.Due.Date != "" {
				dueDate, err = time.Parse("2006-01-02", apiTask.Due.Date)
			}
			if err == nil {
				task.DueDate = &dueDate
			}
		}

		tasks = append(tasks, task)
	}

	return tasks, nil
}

// GetProjects fetches all projects
func (c *TodoistClient) GetProjects(ctx context.Context) ([]models.Project, error) {
	req, err := http.NewRequestWithContext(ctx, "GET", c.baseURL+"/projects", nil)
	if err != nil {
		return nil, fmt.Errorf("failed to create request: %w", err)
	}

	req.Header.Set("Authorization", "Bearer "+c.apiKey)

	resp, err := c.httpClient.Do(req)
	if err != nil {
		return nil, fmt.Errorf("failed to fetch projects: %w", err)
	}
	defer resp.Body.Close()

	if resp.StatusCode != http.StatusOK {
		body, _ := io.ReadAll(resp.Body)
		return nil, fmt.Errorf("todoist API error (status %d): %s", resp.StatusCode, string(body))
	}

	var apiProjects []todoistProjectResponse
	if err := json.NewDecoder(resp.Body).Decode(&apiProjects); err != nil {
		return nil, fmt.Errorf("failed to decode response: %w", err)
	}

	// Convert to model
	projects := make([]models.Project, 0, len(apiProjects))
	for _, apiProj := range apiProjects {
		projects = append(projects, models.Project{
			ID:   apiProj.ID,
			Name: apiProj.Name,
		})
	}

	return projects, nil
}

// CreateTask creates a new task in Todoist
func (c *TodoistClient) CreateTask(ctx context.Context, content, projectID string, dueDate *time.Time, priority int) (*models.Task, error) {
	// Prepare request payload
	payload := map[string]interface{}{
		"content": content,
	}

	if projectID != "" {
		payload["project_id"] = projectID
	}

	if dueDate != nil {
		payload["due_date"] = dueDate.Format("2006-01-02")
	}

	if priority > 0 {
		payload["priority"] = priority
	}

	jsonData, err := json.Marshal(payload)
	if err != nil {
		return nil, fmt.Errorf("failed to marshal request: %w", err)
	}

	// Create POST request
	req, err := http.NewRequestWithContext(ctx, "POST", c.baseURL+"/tasks", bytes.NewBuffer(jsonData))
	if err != nil {
		return nil, fmt.Errorf("failed to create request: %w", err)
	}
	req.Header.Set("Authorization", "Bearer "+c.apiKey)
	req.Header.Set("Content-Type", "application/json")

	// Execute request
	resp, err := c.httpClient.Do(req)
	if err != nil {
		return nil, fmt.Errorf("failed to create task: %w", err)
	}
	defer resp.Body.Close()

	if resp.StatusCode != http.StatusOK {
		body, _ := io.ReadAll(resp.Body)
		return nil, fmt.Errorf("todoist API error (status %d): %s", resp.StatusCode, string(body))
	}

	// Decode response
	var apiTask todoistTaskResponse
	if err := json.NewDecoder(resp.Body).Decode(&apiTask); err != nil {
		return nil, fmt.Errorf("failed to decode response: %w", err)
	}

	// Convert to our model
	task := &models.Task{
		ID:          apiTask.ID,
		Content:     apiTask.Content,
		Description: apiTask.Description,
		ProjectID:   apiTask.ProjectID,
		Priority:    apiTask.Priority,
		Completed:   false,
		Labels:      apiTask.Labels,
		URL:         apiTask.URL,
	}

	// Parse created_at
	if createdAt, err := time.Parse(time.RFC3339, apiTask.CreatedAt); err == nil {
		task.CreatedAt = createdAt
	}

	// Parse due date
	if apiTask.Due != nil {
		var taskDueDate time.Time
		if apiTask.Due.Datetime != "" {
			taskDueDate, err = time.Parse(time.RFC3339, apiTask.Due.Datetime)
		} else if apiTask.Due.Date != "" {
			taskDueDate, err = time.Parse("2006-01-02", apiTask.Due.Date)
		}
		if err == nil {
			task.DueDate = &taskDueDate
		}
	}

	return task, nil
}

// CompleteTask marks a task as complete in Todoist
func (c *TodoistClient) CompleteTask(ctx context.Context, taskID string) error {
	// Create POST request to close endpoint
	url := fmt.Sprintf("%s/tasks/%s/close", c.baseURL, taskID)
	req, err := http.NewRequestWithContext(ctx, "POST", url, nil)
	if err != nil {
		return fmt.Errorf("failed to create request: %w", err)
	}
	req.Header.Set("Authorization", "Bearer "+c.apiKey)

	// Execute request
	resp, err := c.httpClient.Do(req)
	if err != nil {
		return fmt.Errorf("failed to complete task: %w", err)
	}
	defer resp.Body.Close()

	if resp.StatusCode != http.StatusNoContent && resp.StatusCode != http.StatusOK {
		body, _ := io.ReadAll(resp.Body)
		return fmt.Errorf("todoist API error (status %d): %s", resp.StatusCode, string(body))
	}

	return nil
}