package main

import (
    "github.com/opencvlzg/ciproxy"
)

func main() {
    // 使用链式调用创建服务器
    proxy := ciproxy.New().
        SetHost("127.0.0.1", "8080").
        SetMethod(ciproxy.HttpsProxy).
        Use(loggingMiddleware).
        Handle(customHandler)

    // 启动服务器
    proxy.Run()
}

func loggingMiddleware(ctx *ciproxy.Context) {
    ctx.Next()
}

func customHandler(ctx *ciproxy.Context) {
    // 自定义处理逻辑
    ctx.Next()
}

package main

import (
    "github.com/opencvlzg/ciproxy"
)

func main() {
    proxy := ciproxy.New(
        ciproxy.WithHost("127.0.0.1", "8080"),
        ciproxy.WithMethod(ciproxy.HttpInterceptProxy),
        ciproxy.WithMaxConnections(10000),
    )

    proxy.Use(loggingMiddleware)
    proxy.Run()
}

package main

import (
    "flag"
    "github.com/opencvlzg/ciproxy"
)

func main() {
    ip := flag.String("ip", "127.0.0.1", "Server Ip Address")
    port := flag.String("port", "8080", "Server Port")
    method := flag.String("method", ciproxy.HttpsProxy, "Proxy Method")
    flag.Parse()

    proxyServe := ciproxy.ProxyServe{
        Ip:       *ip,
        Port:     *port,
        Method:   *method,
        Protocol: "TCP",
    }
    proxyServe.AddMiddleware(loggingMiddleware)
    proxyServe.Start()
}

package main

import (
    "github.com/opencvlzg/ciproxy"
    "github.com/opencvlzg/ciproxy/pkg/mitm"
    "github.com/opencvlzg/ciproxy/pkg/middleware"
)

func main() {
    // 创建 MITM 拦截器
    interceptor := mitm.NewInterceptor(&mitm.InterceptorConfig{
        EnableTrafficCapture: true,
        EnableHTTP2:          true,
    })

    // 添加拦截中间件
    interceptor.UseFunc(func(ctx *ciproxy.Context) *http.Response {
        // 修改请求或返回自定义响应
        if ctx.GetRequestHeader("X-Custom") == "" {
            ctx.SetRequestHeader("X-Custom", "injected")
        }
        return nil // 返回 nil 放行，返回 response 则阻断请求
    })

    // 启动代理服务器
    proxy := ciproxy.New().SetMethod(ciproxy.HttpInterceptProxy)
    proxy.Run()
}

package main

import (
    "github.com/opencvlzg/ciproxy/pkg/transfer"
)

func main() {
    store := transfer.GetTrafficStore()

    // 捕获请求
    capturedReq := transfer.CaptureFromRequest(req)
    id := store.Capture(capturedReq)

    // 获取流量
    entry, ok := store.Get(id)

    // 列出所有流量
    entries := store.List()

    // 导出/导入
    data, _ := store.Export()
    store.Import(data)

    // 重放请求
    replayer := transfer.NewReplayer()
    resp, _ := replayer.Replay(entry)
}

import "github.com/opencvlzg/ciproxy/pkg/middleware/builtins"

// 日志中间件
proxy.Use(builtins.Logging())

// CORS 中间件
proxy.Use(builtins.CORS())

// 请求头修改
proxy.Use(builtins.SetHeader("X-Proxy", "CiProxy"))

// 请求体修改
proxy.Use(builtins.SetBody([]byte("modified body")))

// 请求阻断
proxy.Use(builtins.BlockWithStatus(403, "Forbidden"))

config, _ := ciproxy.LoadConfig("config.json")
proxy := ciproxy.New(ciproxy.WithConfig(config))
proxy.Run()