package main

import (
	"fmt"
    "log"

	"github.com/pquerna/otp/totp"
)

func main() {
    key, err := totp.Generate(totp.GenerateOpts{})
    if err != nil {
        log.Fatal(err)
    }
    fmt.Printf("key: %s\n", key.Secret())
}

package authcode

import (
	"encoding/json"
	"net/http"
	"net/url"
	"regexp"
	"strings"
	"testing"
)

// json2goで作成した、MailSlurperのメールアドレス一覧のレスポンス
type MailSlurperResponse struct {
	MailItems    []MailItem `json:"mailItems"`
	TotalPages   int        `json:"totalPages"`
	TotalRecords int        `json:"totalRecords"`
}

type MailItem struct {
	ID          string   `json:"id"`
	DateSent    string   `json:"dateSent"`
	FromAddress string   `json:"fromAddress"`
	ToAddresses []string `json:"toAddresses"`
	Subject     string   `json:"subject"`
	Body        string   `json:"body"`
}

// toアドレスでフィルタリングしてのメールの取り出し
func ReceiveMail(t *testing.T, host, to string) []MailItem {
	t.Helper()
	u, err := url.Parse(host)
	if err != nil {
		panic(err)
	}
	u.Path = "/mail"
	q := url.Values{}
	q.Set("to", to)
	u.RawQuery = q.Encode()

	res, err := http.Get(u.String())
	if err != nil {
		panic(err)
	}
	defer res.Body.Close()
	d := json.NewDecoder(res.Body)
	r := MailSlurperResponse{}
	err = d.Decode(&r)
	if err != nil {
		panic(err)
	}
	return r.MailItems
}

// 6桁のコードを取り出す（裏でメールサーバーから情報取得）
func ReceivePassCode(t *testing.T, host, to string) string {
	t.Helper()
	mails := ReceiveMail(t, host, to)

	p := regexp.MustCompile(`\d{6}`)

	for _, m := range mails {
		return p.FindString(m.Body)
	}
	return ""
}

package authcode

import (
	"os"
	"testing"
)

func TestValidate(t *testing.T) {
	secret := "LB6BHGYD63JCWM4BBPHCSRBXGZYKGDI3" // 事前に作成しておいたシークレット
    // これから作成する、パスコード送信処理
	err := SendPassCode("localhost:2500", "test user", "test@example.com", secret)
	if err != nil {
		t.Errorf("error should be nil: %v", err)
		return
	}

	code := ReceivePassCode(t, "http://localhost:8085", "test@example.com")
    // これから実装するバリデーション
	if !Validate(code, secret) {
		t.Error("validation failed")
	}
}

package authcode

import (
	"bytes"
	"encoding/base64"
	"net/mail"
	"net/smtp"
	"strings"
	"time"

	"github.com/pquerna/otp/totp"
)

// 上記のエントリーから、add76crlf, utf8Split, encodeSubjectをコピーしておくこと

// メールの作成
func GenerateMessage(toUserName, toAddress, secret string) ([]byte, error) {
	from := mail.Address{"Myサービス", "noreply@my-service.com"}
	to := mail.Address{toUserName, toAddress}

	var msg bytes.Buffer
	msg.WriteString("From: " + from.String() + "\r\n")
	msg.WriteString("To: " + to.String() + "\r\n")
	msg.WriteString(encodeSubject("Myサービスの認証コード"))
	msg.WriteString("MIME-Version: 1.0\r\n")
	msg.WriteString("Content-Type: text/plain; charset=\"utf-8\"\r\n")
	msg.WriteString("Content-Transfer-Encoding: base64\r\n")

	code, err := totp.GenerateCode(secret, time.Now())
	if err != nil {
		return nil, err
	}

	var body bytes.Buffer
	body.WriteString("認証コードはこちらです\n\n" + code + "\n\nMyサービス")
	msg.WriteString("\r\n")
	msg.WriteString(add76crlf(base64.StdEncoding.EncodeToString(body.Bytes())))

	return msg.Bytes(), nil
}

// メール送信
func SendPassCode(host, toUserName, toAddress, secret string) error {
	msg, err := GenerateMessage(toUserName, toAddress, secret)
	if err != nil {
		return err
	}
	err = smtp.SendMail(
		host,
		nil,
		"noreply@my-service.com",
		[]string{toAddress},
		msg,
	)
	return err
}

// クライアントに送信されたパスコードのバリデーションを行う
func Validate(passcode, secret string) bool {
	return totp.Validate(passcode, secret)
}

func ResetMailSlumper(host string) {
	u, err := url.Parse(host)
	if err != nil {
		panic(err)
	}
	u.Path = "/mail"

	req, _ := http.NewRequest("DELETE", u.String(), strings.NewReader(`{"pruneCode": "all"}`))
	req.Header.Set("Content-Type", "application/json")
	http.DefaultClient.Do(req)
}

func TestMain(m *testing.M) {
	ResetMailSlumper("http://localhost:8085")
	code := m.Run()
	os.Exit(code)
}