// main.go
package main

import (
  "encoding/json"
  "github.com/lhside/chrome-go"
  "io/ioutil"
  "log"
  "os"
)

// NMHが受け取るJSON形式のメッセージに対応する構造体の定義。
type Request struct {
  Path string `json:"path"`
}

// NMHが返すJSON形式のメッセージに対応する構造体の定義。
type Response struct {
  Contents string `json:"contents"`
  Error    string `json:"error"`
}

func main() {
  // ログ類を標準エラー出力に出力するよう設定する。
  log.SetOutput(os.Stderr)

  // 拡張機能からのメッセージを標準入力で受け取る。
  rawRequest, err := chrome.Receive(os.Stdin)
  if err != nil {
    log.Fatal(err)
  }

  // 入力を構造体の定義を元にJSONとしてパースする。
  request := &Request{
    Path: "",
  }
  if err := json.Unmarshal(rawRequest, request); err != nil {
    log.Fatal(err)
  }

  // パース結果を使って処理を行う。
  contents, errorMessage := FetchFile(request.Path)

  // 処理結果を元にJSON文字列を作る。
  response := &Response{
    Contents: contents,
    Error:    errorMessage,
  }
  body, err := json.Marshal(response)
  if err != nil {
    log.Fatal(err)
  }

  // JSON文字列を標準出力で拡張機能に返す。
  err = chrome.Post(body, os.Stdout)
  if err != nil {
    log.Fatal(err)
  }
}

func FetchFile(path string) (contents string, errorMessage string) {
  buffer, err := ioutil.ReadFile(path)
  if err != nil {
    return "", "Failed to fetch the file "+path+": "+err.Error()
  }
  return string(buffer), ""
}

// main.go
package main

import (
  "encoding/json"
  "github.com/lhside/chrome-go"
  "io" // 追加。
  "io/ioutil"
  "log"
  "os"
)

type Request struct {
  Path string `json:"path"`
}

type Response struct {
  Contents string `json:"contents"`
  Error    string `json:"error"`
}

// 追加：入出力をまとめた、情報受け渡し用の構造体の定義。
type Context struct {
  Input    io.Reader
  Output   io.Writer
  ErrorOut io.Writer
}

func main() {
  // 通常の実行時は、構造体を標準入出力で初期化する。
  context := &Context{
    Input:    os.Stdin,
    Output:   os.Stdout,
    ErrorOut: os.Stderr,
  }

  // 入出力を添えて、main関数の主要な処理を分離した関数を実行する。
  if err := ProcessRequest(context); err != nil {
    log.Fatal(err)
  }
}

// 追加：main関数の主要な処理を分離した関数。
func ProcessRequest(context *Context) error {
  // ログ類の出力先を設定する。
  log.SetOutput(context.ErrorOut) // 引数で受け取ったio.Writerを使う。

  // 拡張機能からのメッセージを入力用のio.Reader経由で受け取る。
  rawRequest, err := chrome.Receive(context.Input) // 引数で受け取ったio.Readerを使う。
  if err != nil {
    return err // エラーは自分でlog.Fatalに渡さずに返す。
  }

  // 入力を構造体の定義を元にJSONとしてパースする。
  request := &Request{
    Path: "",
  }
  if err := json.Unmarshal(rawRequest, request); err != nil {
    return err
  }

  // パース結果を使って処理を行う。
  contents, errorMessage := FetchFile(request.Path)

  // 処理結果を元にJSON文字列を作る。
  response := &Response{
    Contents: contents,
    Error:    errorMessage,
  }
  body, err := json.Marshal(response)
  if err != nil {
    return err
  }

  // JSON文字列を出力用のio.Writer経由で拡張機能に返す。
  err = chrome.Post(body, context.Output) // 引数で受け取ったio.Writerを使う。
  if err != nil {
    return err
  }

  return nil
}

// main_test.go
package main

import (
  "bytes"
  "encoding/binary"
  "github.com/stretchr/testify/assert"
  "io"
  "io/ioutil"
  "os"
  "path/filepath"
  "strings"
  "testing"
)

// 与えられた文字列から標準入力代わりのio.Readerを用意するユーティリティ。
func CreateInput(input string) io.Reader {
  // NMHが受け取るメッセージとして妥当な形式になるよう、
  // 先頭にメッセージの長さの情報を付与したバイナリーに変換する。
  buffer := new(bytes.Buffer)
  err := binary.Write(buffer, binary.LittleEndian, []byte(input))
  if err != nil {
    return nil
  }
  header := make([]byte, 4)
  binary.LittleEndian.PutUint32(header, uint32(buffer.Len()))
  message := append(header, buffer.Bytes()...)
  reader := bytes.NewReader(message)
  return reader
}

// io.Writerに出力された内容を読み取って文字列として返すユーティリティ。
func ReadOutput(output io.Reader) string {
  // NMHが出力したメッセージには先頭にメッセージの長さの情報が付与されているため、
  // それを取り除いた残りだけを文字列として取り出す。
  var length uint32
  if err := binary.Read(output, binary.LittleEndian, &length); err != nil {
    return ""
  }
  if length == 0 {
    return ""
  }
  message := make([]byte, length)
  if n, err := output.Read(message); err != nil || n != len(message) {
    return ""
  }
  return string(message)
}

// テスト関数。
func TestNativeMessagingCall_Success(t *testing.T) {
  // リポジトリー内の適当なファイルとして、親ディレクトリーにあるファイルを参照してみる。
  wd, _ := os.Getwd()
  path := filepath.Join(wd, "..", "manifest.json")
  inputMessage := `{`+
    `  "Path":"`+path+`"` +
    `}`
  // NMHに渡されるメッセージを伴った標準入出力をモックする。
  input := CreateInput(inputMessage)
  var output bytes.Buffer
  var errorOut bytes.Buffer
  context := &Context{
    Input:    input,
    Output:   &output,
    ErrorOut: &errorOut,
  }

  // main関数に代わって主要な処理を実行する。
  err := ProcessRequest(context)

  // 内部でエラーが発生しなかったことを確認する。
  assert.NoError(t, err)
  assert.Equal(t, "", errorOut.String())

  // 標準出力のモックに出力された内容が期待通りかを、参照したファイルの実際の内容と比較して確認する。
  fileContents, err := ioutil.ReadFile(path)
  assert.NoError(t, err)
  escapedContents := strings.ReplaceAll(
    strings.ReplaceAll(string(fileContents), `"`, `\"`),
    "\n", `\n`)
  assert.Equal(t, `{"contents":"`+escapedContents+`","error":""}`, ReadOutput(&output))
}

// main_windows.go：Windows専用の実装

//go:build windows

package main

import (
  "golang.org/x/sys/windows/registry"
  "io"
  ...
)

func ReadIntegerRegValue(key registry.Key, valueName string) (data uint64, errorMessage string) {
  ...
}

func ReadStringsRegValue(key registry.Key, valueName string) (data []string, errorMessage string) {
  ...
}

func FetchEnterpriseConfigsAndResponseFromGPO(base registry.Key, keyPath string, configs *TbStyleConfigs) error {
  ...
}

// 実際にmain.goから呼ばれる関数はこれのみ。
func FetchEnterpriseConfigsAndResponse(output io.Writer) error {
  ...
}

// main_other.go：未対応のプラットフォーム向けのスタブ実装

//go:build !windows
// ↑これで「Windowsでない環境向けの実装」であることを示す。
//   macOS用の実装も存在している場合は、「!windows && !darwin」として
//   「WindowsでもmacOSでもない環境向けの実装」であることを示す。

package main

import (
  "io"
)

// 実際にmain.goから呼ばれる関数。
func FetchEnterpriseConfigsAndResponse(output io.Writer) error {
  return nil
}