feat: migrate types

.cursor/rules/collaboration.mdc

@@ -0,0 +1,30 @@
+---
+description: 
+globs: 
+alwaysApply: true
+---
+# Collaboration Rules
+
+## Planning and Confirmation Rule
+
+**Before implementing any code changes, features, or modifications:**
+
+1. **Create a Plan**: Always develop a clear, detailed plan that outlines:
+   - What changes will be made
+   - Which files will be modified or created
+   - The approach and methodology
+   - Expected outcomes and impacts
+
+2. **Confirm with User**: Present the plan to the user and wait for explicit confirmation before:
+   - Making any file modifications
+   - Creating new files
+   - Running commands that modify the codebase
+   - Implementing any suggested changes
+
+3. **Get Approval**: Only proceed with implementation after receiving clear approval from the user.
+
+4. **No Assumptions**: Never assume the user wants changes implemented immediately, even if they seem obvious or beneficial.
+
+**Exception**: Read-only operations (viewing files, searching, analyzing) do not require prior confirmation.
+
+This rule ensures we maintain collaborative control over the codebase and prevents unwanted changes.

.cursor/rules/testing.mdc

@@ -0,0 +1,15 @@
+---
+alwaysApply: true
+---
+
+## Testing Rule
+
+**Before writing any test code:**
+
+1. **Create a Test Plan**: Always develop a clear, detailed plan that outlines:
+   - What tests will be written
+   - Which files will be modified or created
+   - The approach and methodology
+   - Expected outcomes and impacts
+
+2. **Prefer Ginkgo & Gomega**: When writing tests, use the Ginkgo and Gomega frameworks (BDD style) where possible to structure and assert tests in Go. Only use the built-in `testing` package for compatibility or legacy reasons, or if instructed.

.github/dependabot.yml

@@ -0,0 +1,11 @@
+# To get started with Dependabot version updates, you'll need to specify which
+# package ecosystems to update and where the package manifests are located.
+# Please see the documentation for all configuration options:
+# https://docs.github.com/code-security/dependabot/dependabot-version-updates/configuration-options-for-the-dependabot.yml-file
+
+version: 2
+updates:
+  - package-ecosystem: "gomod" # See documentation for possible values
+    directory: "/" # Location of package manifests
+    schedule:
+      interval: "weekly"

.gitignore

@@ -81,6 +81,9 @@ bp-todo.md
 # TEE
 tee/private.pem
 
-# LLML
+# LLM-related files
 .aider*
 GEMINI.md
+
+# Examples from tee-types
+/examples/*json

Makefile

@@ -86,7 +86,9 @@ test-tiktok: docker-build-test
 	@docker run --user root $(ENV_FILE_ARG) -v $(PWD)/.masa:/home/masa -v $(PWD)/coverage:/app/coverage --rm --workdir /app -e DATA_DIR=/home/masa $(TEST_IMAGE) go test -v ./internal/jobs/tiktok_test.go ./internal/jobs/jobs_suite_test.go
 
 test-reddit: docker-build-test
-	@docker run --user root $(ENV_FILE_ARG) -v $(PWD)/.masa:/home/masa -v $(PWD)/coverage:/app/coverage --rm --workdir /app -e DATA_DIR=/home/masa $(TEST_IMAGE) go test -v ./internal/jobs/reddit_test.go ./internal/jobs/redditapify/client_test.go ./api/types/reddit/reddit_suite_test.go
+	@docker run --user root $(ENV_FILE_ARG) -v $(PWD)/.masa:/home/masa -v $(PWD)/coverage:/app/coverage --rm --workdir /app -e DATA_DIR=/home/masa $(TEST_IMAGE) sh -c "cd /app && go test -v ./internal/jobs/reddit_test.go ./internal/jobs/jobs_suite_test.go"
+	@docker run --user root $(ENV_FILE_ARG) -v $(PWD)/.masa:/home/masa -v $(PWD)/coverage:/app/coverage --rm --workdir /app -e DATA_DIR=/home/masa $(TEST_IMAGE) sh -c "cd /app && go test -v ./internal/jobs/redditapify"
+	@docker run --user root $(ENV_FILE_ARG) -v $(PWD)/.masa:/home/masa -v $(PWD)/coverage:/app/coverage --rm --workdir /app -e DATA_DIR=/home/masa $(TEST_IMAGE) sh -c "cd /app && go test -v ./api/types/reddit_test.go"
 
 test-web: docker-build-test
 	@docker run --user root $(ENV_FILE_ARG) -v $(PWD)/.masa:/home/masa -v $(PWD)/coverage:/app/coverage --rm --workdir /app -e DATA_DIR=/home/masa $(TEST_IMAGE) sh -c "cd /app && go test -v ./internal/jobs/web_test.go ./internal/jobs/jobs_suite_test.go"

api/args/base/base.go

@@ -0,0 +1,49 @@
+package base
+
+import (
+	"encoding/json"
+	"fmt"
+
+	"github.com/masa-finance/tee-worker/api/types"
+)
+
+// JobArgument defines the interface that all job arguments must implement
+type JobArgument interface {
+	UnmarshalJSON([]byte) error
+	GetCapability() types.Capability
+	ValidateCapability(jobType types.JobType) error
+	SetDefaultValues()
+	Validate() error
+}
+
+// Verify interface implementation
+var _ JobArgument = (*Arguments)(nil)
+
+type Arguments struct {
+	Type types.Capability `json:"type"`
+}
+
+func (t *Arguments) UnmarshalJSON(data []byte) error {
+	type Alias Arguments
+	aux := &struct{ *Alias }{Alias: (*Alias)(t)}
+	if err := json.Unmarshal(data, aux); err != nil {
+		return fmt.Errorf("%v: %w", "failed to unmarshal arguments", err)
+	}
+	t.SetDefaultValues()
+	return t.Validate()
+}
+
+func (a *Arguments) GetCapability() types.Capability {
+	return a.Type
+}
+
+func (a *Arguments) ValidateCapability(jobType types.JobType) error {
+	return jobType.ValidateCapability(&a.Type)
+}
+
+func (a *Arguments) SetDefaultValues() {
+}
+
+func (a *Arguments) Validate() error {
+	return nil
+}

api/args/linkedin/linkedin.go

@@ -0,0 +1,7 @@
+package linkedin
+
+import (
+	"github.com/masa-finance/tee-worker/api/args/linkedin/profile"
+)
+
+type Profile = profile.Arguments

api/args/linkedin/profile/profile.go

@@ -0,0 +1,141 @@
+package profile
+
+import (
+	"encoding/json"
+	"errors"
+	"fmt"
+
+	"github.com/masa-finance/tee-worker/api/args/base"
+	"github.com/masa-finance/tee-worker/api/types"
+	"github.com/masa-finance/tee-worker/api/types/linkedin/experiences"
+	"github.com/masa-finance/tee-worker/api/types/linkedin/functions"
+	"github.com/masa-finance/tee-worker/api/types/linkedin/industries"
+	"github.com/masa-finance/tee-worker/api/types/linkedin/profile"
+	"github.com/masa-finance/tee-worker/api/types/linkedin/seniorities"
+)
+
+var (
+	ErrScraperModeNotSupported = errors.New("scraper mode not supported")
+	ErrMaxItemsTooLarge        = errors.New("max items must be less than or equal to 100")
+	ErrExperienceNotSupported  = errors.New("years of experience not supported")
+	ErrSeniorityNotSupported   = errors.New("seniority level not supported")
+	ErrFunctionNotSupported    = errors.New("function not supported")
+	ErrIndustryNotSupported    = errors.New("industry not supported")
+	ErrUnmarshalling           = errors.New("failed to unmarshal LinkedIn profile arguments")
+)
+
+const (
+	DefaultMaxItems    = 10
+	DefaultScraperMode = profile.ScraperModeShort
+	MaxItems           = 1000 // 2500 on the actor, but we will run over 1MB memory limit on responses
+)
+
+// Verify interface implementation
+var _ base.JobArgument = (*Arguments)(nil)
+
+// Arguments defines args for LinkedIn profile operations
+type Arguments struct {
+	Type                  types.Capability    `json:"type"`
+	ScraperMode           profile.ScraperMode `json:"profileScraperMode"`
+	Query                 string              `json:"searchQuery"`
+	MaxItems              uint                `json:"maxItems"`
+	Locations             []string            `json:"locations,omitempty"`
+	CurrentCompanies      []string            `json:"currentCompanies,omitempty"`
+	PastCompanies         []string            `json:"pastCompanies,omitempty"`
+	CurrentJobTitles      []string            `json:"currentJobTitles,omitempty"`
+	PastJobTitles         []string            `json:"pastJobTitles,omitempty"`
+	Schools               []string            `json:"schools,omitempty"`
+	YearsOfExperience     []experiences.Id    `json:"yearsOfExperienceIds,omitempty"`
+	YearsAtCurrentCompany []experiences.Id    `json:"yearsAtCurrentCompanyIds,omitempty"`
+	SeniorityLevels       []seniorities.Id    `json:"seniorityLevelIds,omitempty"`
+	Functions             []functions.Id      `json:"functionIds,omitempty"`
+	Industries            []industries.Id     `json:"industryIds,omitempty"`
+	FirstNames            []string            `json:"firstNames,omitempty"`
+	LastNames             []string            `json:"lastNames,omitempty"`
+	RecentlyChangedJobs   bool                `json:"recentlyChangedJobs,omitempty"`
+	StartPage             uint                `json:"startPage,omitempty"`
+}
+
+func (t *Arguments) UnmarshalJSON(data []byte) error {
+	type Alias Arguments
+	aux := &struct{ *Alias }{Alias: (*Alias)(t)}
+	if err := json.Unmarshal(data, aux); err != nil {
+		return fmt.Errorf("%w: %w", ErrUnmarshalling, err)
+	}
+	t.SetDefaultValues()
+	return t.Validate()
+}
+
+func (a *Arguments) SetDefaultValues() {
+	if a.MaxItems == 0 {
+		a.MaxItems = DefaultMaxItems
+	}
+	if a.ScraperMode == "" {
+		a.ScraperMode = DefaultScraperMode
+	}
+}
+
+func (a *Arguments) Validate() error {
+	var errs []error
+
+	if a.MaxItems > MaxItems {
+		errs = append(errs, ErrMaxItemsTooLarge)
+	}
+
+	err := a.ValidateCapability(types.LinkedInJob)
+	if err != nil {
+		errs = append(errs, err)
+	}
+
+	if !profile.AllScraperModes.Contains(a.ScraperMode) {
+		errs = append(errs, ErrScraperModeNotSupported)
+	}
+
+	for _, yoe := range a.YearsOfExperience {
+		if !experiences.All.Contains(yoe) {
+			errs = append(errs, fmt.Errorf("%w: %v", ErrExperienceNotSupported, yoe))
+		}
+	}
+	for _, yac := range a.YearsAtCurrentCompany {
+		if !experiences.All.Contains(yac) {
+			errs = append(errs, fmt.Errorf("%w: %v", ErrExperienceNotSupported, yac))
+		}
+	}
+	for _, sl := range a.SeniorityLevels {
+		if !seniorities.All.Contains(sl) {
+			errs = append(errs, fmt.Errorf("%w: %v", ErrSeniorityNotSupported, sl))
+		}
+	}
+	for _, f := range a.Functions {
+		if !functions.All.Contains(f) {
+			errs = append(errs, fmt.Errorf("%w: %v", ErrFunctionNotSupported, f))
+		}
+	}
+	for _, i := range a.Industries {
+		if !industries.All.Contains(i) {
+			errs = append(errs, fmt.Errorf("%w: %v", ErrIndustryNotSupported, i))
+		}
+	}
+
+	if len(errs) > 0 {
+		return errors.Join(errs...)
+	}
+
+	return nil
+}
+
+func (a *Arguments) GetCapability() types.Capability {
+	return a.Type
+}
+
+func (a *Arguments) ValidateCapability(jobType types.JobType) error {
+	return jobType.ValidateCapability(&a.Type)
+}
+
+// NewArguments creates a new Arguments instance and applies default values immediately
+func NewArguments() Arguments {
+	args := Arguments{}
+	args.SetDefaultValues()
+	args.Validate() // This will set the default capability via ValidateCapability
+	return args
+}

api/types/reddit/reddit_suite_test.go → api/args/linkedin/profile/profile_suite_test.go

@@ -1,4 +1,4 @@
-package reddit_test
+package profile_test
 
 import (
 	"testing"
@@ -7,7 +7,7 @@ import (
 	. "github.com/onsi/gomega"
 )
 
-func TestReddit(t *testing.T) {
+func TestArgs(t *testing.T) {
 	RegisterFailHandler(Fail)
-	RunSpecs(t, "Reddit Suite")
-}
+	RunSpecs(t, "Args Suite")
+}