feat: linkedin

.gitignore

@@ -13,3 +13,5 @@ go.work
 # LLM-related files
 .aider*
 GEMINI.md
+
+/examples/*json

args/args.go

@@ -3,13 +3,17 @@ package args
 import (
 	"encoding/json"
 	"fmt"
-	"strings"
+
+	"github.com/masa-finance/tee-types/args/linkedin"
+	teetypes "github.com/masa-finance/tee-types/types"
 )
 
+type LinkedInProfileArguments = linkedin.ProfileArguments
+
 // QueryTypeArgument provides a minimal structure to extract the QueryType (json "type")
 // This is used across different job types to determine the specific capability being requested
 type QueryTypeArgument struct {
-	QueryType string `json:"type"`
+	QueryType teetypes.Capability `json:"type"`
 }
 
 // UnmarshalJSON implements custom JSON unmarshaling with normalization
@@ -20,6 +24,6 @@ func (q *QueryTypeArgument) UnmarshalJSON(data []byte) error {
 	if err := json.Unmarshal(data, aux); err != nil {
 		return fmt.Errorf("failed to unmarshal QueryType arguments: %w", err)
 	}
-	q.QueryType = strings.ToLower(aux.QueryType)
+	q.QueryType = aux.QueryType
 	return nil
 }

args/linkedin/linkedin.go

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

args/linkedin/profile/profile.go

@@ -0,0 +1,132 @@
+package profile
+
+import (
+	"encoding/json"
+	"errors"
+	"fmt"
+
+	teetypes "github.com/masa-finance/tee-types/types"
+	"github.com/masa-finance/tee-types/types/linkedin/experiences"
+	"github.com/masa-finance/tee-types/types/linkedin/functions"
+	"github.com/masa-finance/tee-types/types/linkedin/industries"
+	"github.com/masa-finance/tee-types/types/linkedin/profile"
+	"github.com/masa-finance/tee-types/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")
+)
+
+const (
+	DefaultMaxItems    = 10
+	DefaultScraperMode = profile.ScraperModeShort
+	MaxItems           = 1000 // 2500 on the actor, but we will run over 1MB memory limit on responses
+)
+
+// Arguments defines args for LinkedIn profile operations
+type Arguments struct {
+	QueryType             teetypes.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 (a *Arguments) UnmarshalJSON(data []byte) error {
+	type Alias Arguments
+	aux := &struct {
+		*Alias
+	}{
+		Alias: (*Alias)(a),
+	}
+
+	if err := json.Unmarshal(data, aux); err != nil {
+		return fmt.Errorf("failed to unmarshal LinkedIn profile arguments: %w", err)
+	}
+
+	a.setDefaultValues()
+
+	return a.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)
+	}
+	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() teetypes.Capability {
+	return a.QueryType
+}
+
+func (a *Arguments) ValidateForJobType(jobType teetypes.JobType) error {
+	if err := a.Validate(); err != nil {
+		return err
+	}
+
+	return jobType.ValidateCapability(a.QueryType)
+}

args/linkedin/profile/profile_suite_test.go

@@ -0,0 +1,13 @@
+package profile_test
+
+import (
+	"testing"
+
+	. "github.com/onsi/ginkgo/v2"
+	. "github.com/onsi/gomega"
+)
+
+func TestArgs(t *testing.T) {
+	RegisterFailHandler(Fail)
+	RunSpecs(t, "Args Suite")
+}