forked from go-gitea/gitea
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Convert files to utf-8 for indexing (go-gitea#7814)
* Convert files to utf-8 for indexing * Move utf8 functions to modules/base * Bump repoIndexerLatestVersion to 3 * Add tests for base/encoding.go * Changes to pass gosimple * Move UTF8 funcs into new modules/charset package
- Loading branch information
Showing
13 changed files
with
371 additions
and
166 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,152 @@ | ||
| // Copyright 2014 The Gogs Authors. All rights reserved. | ||
| // Use of this source code is governed by a MIT-style | ||
| // license that can be found in the LICENSE file. | ||
|
|
||
| package charset | ||
|
|
||
| import ( | ||
| "bytes" | ||
| "fmt" | ||
| "unicode/utf8" | ||
|
|
||
| "code.gitea.io/gitea/modules/log" | ||
| "code.gitea.io/gitea/modules/setting" | ||
|
|
||
| "github.com/gogits/chardet" | ||
| "golang.org/x/net/html/charset" | ||
| "golang.org/x/text/transform" | ||
| ) | ||
|
|
||
| // UTF8BOM is the utf-8 byte-order marker | ||
| var UTF8BOM = []byte{'\xef', '\xbb', '\xbf'} | ||
|
|
||
| // ToUTF8WithErr converts content to UTF8 encoding | ||
| func ToUTF8WithErr(content []byte) (string, error) { | ||
| charsetLabel, err := DetectEncoding(content) | ||
| if err != nil { | ||
| return "", err | ||
| } else if charsetLabel == "UTF-8" { | ||
| return string(RemoveBOMIfPresent(content)), nil | ||
| } | ||
|
|
||
| encoding, _ := charset.Lookup(charsetLabel) | ||
| if encoding == nil { | ||
| return string(content), fmt.Errorf("Unknown encoding: %s", charsetLabel) | ||
| } | ||
|
|
||
| // If there is an error, we concatenate the nicely decoded part and the | ||
| // original left over. This way we won't lose data. | ||
| result, n, err := transform.Bytes(encoding.NewDecoder(), content) | ||
| if err != nil { | ||
| result = append(result, content[n:]...) | ||
| } | ||
|
|
||
| result = RemoveBOMIfPresent(result) | ||
|
|
||
| return string(result), err | ||
| } | ||
|
|
||
| // ToUTF8WithFallback detects the encoding of content and coverts to UTF-8 if possible | ||
| func ToUTF8WithFallback(content []byte) []byte { | ||
| charsetLabel, err := DetectEncoding(content) | ||
| if err != nil || charsetLabel == "UTF-8" { | ||
| return RemoveBOMIfPresent(content) | ||
| } | ||
|
|
||
| encoding, _ := charset.Lookup(charsetLabel) | ||
| if encoding == nil { | ||
| return content | ||
| } | ||
|
|
||
| // If there is an error, we concatenate the nicely decoded part and the | ||
| // original left over. This way we won't lose data. | ||
| result, n, err := transform.Bytes(encoding.NewDecoder(), content) | ||
| if err != nil { | ||
| return append(result, content[n:]...) | ||
| } | ||
|
|
||
| return RemoveBOMIfPresent(result) | ||
| } | ||
|
|
||
| // ToUTF8 converts content to UTF8 encoding and ignore error | ||
| func ToUTF8(content string) string { | ||
| res, _ := ToUTF8WithErr([]byte(content)) | ||
| return res | ||
| } | ||
|
|
||
| // ToUTF8DropErrors makes sure the return string is valid utf-8; attempts conversion if possible | ||
| func ToUTF8DropErrors(content []byte) []byte { | ||
| charsetLabel, err := DetectEncoding(content) | ||
| if err != nil || charsetLabel == "UTF-8" { | ||
| return RemoveBOMIfPresent(content) | ||
| } | ||
|
|
||
| encoding, _ := charset.Lookup(charsetLabel) | ||
| if encoding == nil { | ||
| return content | ||
| } | ||
|
|
||
| // We ignore any non-decodable parts from the file. | ||
| // Some parts might be lost | ||
| var decoded []byte | ||
| decoder := encoding.NewDecoder() | ||
| idx := 0 | ||
| for { | ||
| result, n, err := transform.Bytes(decoder, content[idx:]) | ||
| decoded = append(decoded, result...) | ||
| if err == nil { | ||
| break | ||
| } | ||
| decoded = append(decoded, ' ') | ||
| idx = idx + n + 1 | ||
| if idx >= len(content) { | ||
| break | ||
| } | ||
| } | ||
|
|
||
| return RemoveBOMIfPresent(decoded) | ||
| } | ||
|
|
||
| // RemoveBOMIfPresent removes a UTF-8 BOM from a []byte | ||
| func RemoveBOMIfPresent(content []byte) []byte { | ||
| if len(content) > 2 && bytes.Equal(content[0:3], UTF8BOM) { | ||
| return content[3:] | ||
| } | ||
| return content | ||
| } | ||
|
|
||
| // DetectEncoding detect the encoding of content | ||
| func DetectEncoding(content []byte) (string, error) { | ||
| if utf8.Valid(content) { | ||
| log.Debug("Detected encoding: utf-8 (fast)") | ||
| return "UTF-8", nil | ||
| } | ||
|
|
||
| textDetector := chardet.NewTextDetector() | ||
| var detectContent []byte | ||
| if len(content) < 1024 { | ||
| // Check if original content is valid | ||
| if _, err := textDetector.DetectBest(content); err != nil { | ||
| return "", err | ||
| } | ||
| times := 1024 / len(content) | ||
| detectContent = make([]byte, 0, times*len(content)) | ||
| for i := 0; i < times; i++ { | ||
| detectContent = append(detectContent, content...) | ||
| } | ||
| } else { | ||
| detectContent = content | ||
| } | ||
| result, err := textDetector.DetectBest(detectContent) | ||
| if err != nil { | ||
| return "", err | ||
| } | ||
| // FIXME: to properly decouple this function the fallback ANSI charset should be passed as an argument | ||
| if result.Charset != "UTF-8" && len(setting.Repository.AnsiCharset) > 0 { | ||
| log.Debug("Using default AnsiCharset: %s", setting.Repository.AnsiCharset) | ||
| return setting.Repository.AnsiCharset, err | ||
| } | ||
|
|
||
| log.Debug("Detected encoding: %s", result.Charset) | ||
| return result.Charset, err | ||
| } |
Oops, something went wrong.