Production-Ready Driver's License OCR for Go

package main

import (
	"bytes"
	"encoding/base64"
	"encoding/json"
	"fmt"
	"io"
	"net/http"
	"os"
)

// Get 200 free credits:
// 👉 https://structocr.com/register

// RequestPayload defines the structure for the API request body.
type RequestPayload struct {
	Image string `json:"img"`
}

// ExtraDetailsData handles regional specifics and deep security codes.
type ExtraDetailsData struct {
	CardSecurityNumber *string `json:"card_security_number"`
	FathersName        *string `json:"fathers_name"`
	MothersName        *string `json:"mothers_name"`
	RgNumber           *string `json:"rg_number"`
}

// DriverLicenseData defines the structure of the extracted data.
type DriverLicenseData struct {
	Type           string            `json:"type"`
	CountryCode    string            `json:"country_code"`
	Region         string            `json:"region"`
	DocumentNumber string            `json:"document_number"`
	PersonalNumber *string           `json:"personal_number"`
	Surname        string            `json:"surname"`
	GivenNames     string            `json:"given_names"`
	DateOfBirth    string            `json:"date_of_birth"`
	DateOfExpiry   string            `json:"date_of_expiry"`
	DateOfIssue    string            `json:"date_of_issue"`
	Sex            string            `json:"sex"`
	Address        string            `json:"address"`
	VehicleClass   string            `json:"vehicle_class"`
	ExtraDetails   *ExtraDetailsData `json:"extra_details"`
}

// APIResponse defines the structure of the full API response.
type APIResponse struct {
	Success bool              `json:"success"`
	Data    DriverLicenseData `json:"data"`
	Error   string            `json:"error,omitempty"` // Capture error codes if any
}

func main() {
	// 1. Setup Configuration
	apiURL := "https://api.structocr.com/v1/driver-license"
	apiKey := "YOUR_API_KEY_HERE" // Replace with your actual key
	imagePath := "license.jpg"      // Ensure this file exists

	// 2. Read Image and Encode to Base64
	imageBytes, err := os.ReadFile(imagePath)
	if err != nil {
		fmt.Printf("Error reading image: %v\n", err)
		return
	}

	base64Image := base64.StdEncoding.EncodeToString(imageBytes)

	// 3. Prepare Request Payload
	payload := RequestPayload{
		Image: base64Image,
	}
	payloadBytes, err := json.Marshal(payload)
	if err != nil {
		fmt.Printf("Error marshalling payload: %v\n", err)
		return
	}

	// 4. Create HTTP Request
	req, err := http.NewRequest("POST", apiURL, bytes.NewBuffer(payloadBytes))
	if err != nil {
		fmt.Printf("Error creating request: %v\n", err)
		return
	}

	// Important: Set the correct headers
	req.Header.Set("Content-Type", "application/json")
	req.Header.Set("x-api-key", apiKey)

	// 5. Execute Request
	client := &http.Client{}
	resp, err := client.Do(req)
	if err != nil {
		fmt.Printf("Network error: %v\n", err)
		return
	}
	defer resp.Body.Close()

	// 6. Parse Response
	body, _ := io.ReadAll(resp.Body)

	if resp.StatusCode != http.StatusOK {
		fmt.Printf("API Error (%d): %s\n", resp.StatusCode, string(body))
		return
	}

	var apiResponse APIResponse
	if err := json.Unmarshal(body, &apiResponse); err != nil {
		fmt.Printf("JSON Parse Error: %v\n", err)
		return
	}

	if apiResponse.Success {
		fmt.Println("✅ Extraction Successful!")
		fmt.Printf("Type: %s\n", apiResponse.Data.Type)
		fmt.Printf("Name: %s %s\n", apiResponse.Data.GivenNames, apiResponse.Data.Surname)
		fmt.Printf("Doc Number: %s\n", apiResponse.Data.DocumentNumber)
		fmt.Printf("Expiry: %s\n", apiResponse.Data.DateOfExpiry)
		fmt.Printf("Vehicle Class: %s\n", apiResponse.Data.VehicleClass)

		// Highlighting the advanced security feature extraction
		if apiResponse.Data.ExtraDetails != nil && apiResponse.Data.ExtraDetails.CardSecurityNumber != nil {
			fmt.Printf("Security Code (DD): %s\n", *apiResponse.Data.ExtraDetails.CardSecurityNumber)
		}
	} else {
		fmt.Printf("API indicated failure: %s\n", apiResponse.Error)
	}
}